龚焕茏
2024-03-12 dbe2fda2dbf7e44be430791cb9b323c42c674738
dujy-modules/dujy-demo/src/main/java/org/dromara/demo/service/impl/RsSceneryOperationDataServiceImpl.java
@@ -127,21 +127,25 @@
    @Override
    public String importExcel(List<RsSceneryOperationData> list) {
        List<String> sceneryInfoNames = new ArrayList<>();
        Set<String> sceneryInfoNames = new HashSet<>();
        // 查询数据库是否存在导入的景区
        Map<String, RsSceneryInfo> sceneryInfoMap = sceneryInfoMapper.selectList().stream()
            .collect(Collectors.toMap(RsSceneryInfo::getSceneryName, Function.identity()));
        for (RsSceneryOperationData operationData : list) {
            String sceneryName = operationData.getSceneryInfoName();
            RsSceneryInfo sceneryInfo = sceneryInfoMap.get(sceneryName);
            // 景区已存在,正常导入,默认禁用
            if (sceneryInfo != null) {
                operationData.setSceneryInfoId(sceneryInfo.getId());
                operationData.setPeriod(operationData.getPeriodDate().length() > 4 ? 1L : 2L);
                operationData.setCreateTime(new Date());
                operationData.setStatus(2L);
            } else {
                sceneryInfoNames.add(operationData.getSceneryInfoName());
            }
        }
        if (!sceneryInfoNames.isEmpty()) {
            return "很抱歉,导入失败!" + String.join("、", sceneryInfoNames) + "景区不存在,请前往【景区配置】添加后重试!";
            return "很抱歉,导入失败!" + String.join("、", "<" + sceneryInfoNames + ">") + "景区不存在,请前往【景区配置】添加后重试!";
        }
        baseMapper.insertBatch(list);
        return "恭喜您,数据已全部导入成功!共"  + list.size() + "条!";