东欢坨电--代码提交

ruoyi-framework/src/main/java/com/ruoyi/framework/config/SecurityConfig.java

@@ -114,6 +114,10 @@ public class SecurityConfig extends WebSecurityConfigurerAdapter
                 .antMatchers("/*/api-docs").anonymous()
                 .antMatchers("/druid/**").anonymous()
 
+                .antMatchers("/powerDisplay/save").anonymous()
+                .antMatchers("/power/source").anonymous()
+                .antMatchers("/test/source").anonymous()
+                .antMatchers("/powerDisplay/peek").anonymous()
                 .antMatchers("/current/rule/salary").anonymous()
                 .antMatchers("/expense/source").anonymous()
                 // activiti modeler 放行

ruoyi-wages/src/main/java/com/ruoyi/system/service/impl/ImportExpenseSourceServiceImpl.java

 package com.ruoyi.system.service.impl;
 
 import com.alibaba.excel.EasyExcelFactory;
+import com.alibaba.fastjson.JSON;
 import com.ruoyi.common.core.domain.AjaxResult;
 import com.ruoyi.system.listener.ExcelListener;
 import com.ruoyi.system.mapper.ImportExpenseSourceMapper;
@@ -35,11 +36,12 @@ public class ImportExpenseSourceServiceImpl implements ImportExpenseSourceServic
     @Override
     @SneakyThrows
     public AjaxResult save(MultipartFile multipartFile) {
-        List<PowerImportSourceModel> powerImportSourceModelList = EasyExcelFactory.read(multipartFile.getInputStream(), PowerImportSourceModel.class, new ExcelListener<>()).sheet("Sheet2").headRowNumber(1).doReadSync();
+        List<PowerImportSourceModel> powerImportSourceModelList = EasyExcelFactory.read(multipartFile.getInputStream(), PowerImportSourceModel.class, new ExcelListener<>()).sheet("Sheet3").headRowNumber(1).doReadSync();
         powerImportSourceModelList.forEach(v->{
             v.setClassificationName(v.getClassificationName().replaceAll(StringUtils.SPACE,StringUtils.EMPTY));
             v.setLevelOneClassification(v.getLevelOneClassification().replaceAll(StringUtils.SPACE,StringUtils.EMPTY));
         });
+        System.err.println(JSON.toJSONString(powerImportSourceModelList));
         importExpenseSourceMapper.saveSource(powerImportSourceModelList);
         return AjaxResult.success();
     }

ruoyi-wages/src/main/java/com/ruoyi/system/service/impl/PowerDisplayServiceImpl.java

 package com.ruoyi.system.service.impl;
 
 
+import com.alibaba.fastjson.JSON;
 import com.ruoyi.common.core.domain.AjaxResult;
 import com.ruoyi.common.core.domain.entity.SysUser;
 import com.ruoyi.common.utils.SecurityUtils;
@@ -272,12 +273,83 @@ public class PowerDisplayServiceImpl implements PowerDisplayService {
             if (StringUtils.equals("011701", mineid)) {
                 saveFGZPowerData(newRuleRequestModelList);
             }
+            if (StringUtils.equals("010101", mineid)) {
+                saveTsPowerData(newRuleRequestModelList);
+            }
+            if (StringUtils.equals("010102", mineid)) {
+                saveDHTPowerData(newRuleRequestModelList);
+            }
+            if (StringUtils.equals("011706", mineid)) {
+                saveZRPowerData(newRuleRequestModelList);
+            }
         } else {
             return AjaxResult.error();
         }
         return AjaxResult.success();
     }
 
+    /**
+     * 中润
+     *
+     * @param newRuleRequestModelList
+     */
+    private void saveZRPowerData(List<NewRuleRequestModel> newRuleRequestModelList) {
+        String month = newRuleRequestModelList.get(0).getWageMonth();
+        String minId = newRuleRequestModelList.get(0).getMineid();
+        ReplaceBlank(newRuleRequestModelList);
+        List<PowerDisplayDto> displayDtoList = this.getPowerDisplayDtoList(newRuleRequestModelList);
+        List<PowerDisplayDto> powerDisplayDtoList = groupData(displayDtoList);
+        powerDisplayDtoList.forEach(v -> {
+            v.setPowerTotal(String.valueOf(new BigDecimal(v.getPowerTotal())));
+            v.setElectricityTotal(String.valueOf(new BigDecimal(v.getElectricityTotal())));
+        });
+        getPowerSource(month, minId, powerDisplayDtoList);
+    }
+
+    /**
+     * 东欢坨
+     */
+    private void saveDHTPowerData(List<NewRuleRequestModel> newRuleRequestModelList) {
+        String month = newRuleRequestModelList.get(0).getWageMonth();
+        String minId = newRuleRequestModelList.get(0).getMineid();
+        ReplaceBlank(newRuleRequestModelList);
+        List<PowerDisplayDto> displayDtoList = this.getPowerDisplayDtoList(newRuleRequestModelList);
+        List<PowerSourceModel> powerSourceModelList = powerDisplayMapper.selectSource(minId);
+        List<PowerDisplayDto> collect = displayDtoList.stream().filter(v -> powerSourceModelList.stream().anyMatch(source -> {
+            if (!StringUtils.contains(v.getClassificationName(), "开滦")) {
+                return StringUtils.contains(v.getClassificationName().replaceAll(".*）", StringUtils.EMPTY), source.getClassificationName());
+            }
+            return StringUtils.contains(v.getClassificationName(), source.getClassificationName());
+        })).collect(Collectors.toList());
+        collect.forEach(v -> {
+            if (!StringUtils.contains(v.getClassificationName(), "开滦")) {
+                v.setClassificationName(v.getClassificationName().replaceAll(".*\\)", StringUtils.EMPTY).replaceAll(".*）", StringUtils.EMPTY).replaceAll("\\u3000",StringUtils.EMPTY));
+            } else {
+                v.setClassificationName(StringUtils.substringAfterLast(v.getClassificationName(), "."));
+            }
+            v.setPowerTotal(new BigDecimal(v.getPowerTotal()).toPlainString());
+            v.setElectricityTotal(new BigDecimal(v.getElectricityTotal().toLowerCase()).toPlainString());
+        });
+        List<PowerDisplayDto> powerDisplayDtoList = groupData(collect);
+        getPowerSource(month, minId, powerDisplayDtoList);
+    }
+
+
+    /**
+     * 唐山矿
+     *
+     * @param newRuleRequestModelList
+     */
+    private void saveTsPowerData(List<NewRuleRequestModel> newRuleRequestModelList) {
+        String month = newRuleRequestModelList.get(0).getWageMonth();
+        String minId = newRuleRequestModelList.get(0).getMineid();
+        ReplaceBlank(newRuleRequestModelList);
+        List<PowerDisplayDto> displayDtoList = this.getPowerDisplayDtoList(newRuleRequestModelList);
+        List<PowerDisplayDto> powerDisplayDtoList = groupData(displayDtoList);
+        getPowerSource(month, minId, powerDisplayDtoList);
+    }
+
+
     /**
      * 范各庄
      *
@@ -304,12 +376,7 @@ public class PowerDisplayServiceImpl implements PowerDisplayService {
         ReplaceBlank(powerDisplayDtoList);
         List<PowerDisplayDto> displayDtoList = getPowerDisplayDtoList(powerDisplayDtoList);
         disposeParamter(displayDtoList);
-        Map<String, List<PowerDisplayDto>> map = displayDtoList.stream().collect(Collectors.groupingBy(PowerDisplayDto::getClassificationName));
-        List<PowerDisplayDto> finalList = new ArrayList<>();
-        map.forEach((k, v) -> {
-            PowerDisplayDto pd = meagreData(v);
-            finalList.add(pd);
-        });
+        List<PowerDisplayDto> finalList = groupData(displayDtoList);
         List<PowerDisplayDto> collect = finalList.stream().filter(v -> !StringUtils.contains(v.getClassificationName(), "合计")).collect(Collectors.toList());
         List<PowerSourceModel> powerSourceModelList = powerDisplayMapper.selectSource("011702");
         collect = collect.stream().filter(v -> powerSourceModelList.stream().noneMatch(powerSourceModel -> StringUtils.equals(v.getClassificationName(), powerSourceModel.getLevelOneClassification()))).collect(Collectors.toList());
@@ -326,6 +393,27 @@ public class PowerDisplayServiceImpl implements PowerDisplayService {
         powerDisplayMapper.insert(collect);
     }
 
+    private void getPowerSource(String month, String minId, List<PowerDisplayDto> powerDisplayDtoList) {
+        List<PowerSourceModel> powerSourceModelList = powerDisplayMapper.selectSource(minId);
+        powerDisplayDtoList.forEach(v -> powerSourceModelList.forEach(source -> {
+            if (StringUtils.equals(source.getClassificationName(), v.getClassificationName())) {
+                v.setLevelOneClassification(source.getLevelOneClassification());
+            }
+        }));
+        powerDisplayMapper.deleteRepeat(month, minId);
+        powerDisplayMapper.insert(powerDisplayDtoList);
+    }
+
+    private List<PowerDisplayDto> groupData(List<PowerDisplayDto> displayDtoList) {
+        Map<String, List<PowerDisplayDto>> collect = displayDtoList.stream().collect(Collectors.groupingBy(PowerDisplayDto::getClassificationName));
+        List<PowerDisplayDto> finalList = new ArrayList<>();
+        collect.forEach((k, v) -> {
+            PowerDisplayDto pd = mergeData(v);
+            finalList.add(pd);
+        });
+        return finalList;
+    }
+
     private void calculateEleFeeByDept(List<NewRuleRequestModel> sourceDataList, List<PowerDisplayDto> displayDtoList, List<PowerConvertSourceModel> sourceModelList) {
         String month = sourceDataList.get(0).getWageMonth();
         String mindId = sourceDataList.get(0).getMineid();
@@ -385,7 +473,7 @@ public class PowerDisplayServiceImpl implements PowerDisplayService {
         powerDisplayMapper.insert(displayDtoList);
     }
 
-    private PowerDisplayDto meagreData(List<PowerDisplayDto> powerDisplayDtoList) {
+    private PowerDisplayDto mergeData(List<PowerDisplayDto> powerDisplayDtoList) {
         return powerDisplayDtoList.stream().collect(Collectors.collectingAndThen(Collectors.toMap(data -> data.getMindId() + data.getClassificationName() + data.getDateColumn(), data -> data, (existingData, newData) -> {
             existingData.setPowerTotal(StringUtils.isEmpty(existingData.getPowerTotal()) ? newData.getPowerTotal() : existingData.getPowerTotal());
             existingData.setPowerTip(StringUtils.isEmpty(existingData.getPowerTip()) ? newData.getPowerTip() : existingData.getPowerTip());
@@ -462,5 +550,4 @@ public class PowerDisplayServiceImpl implements PowerDisplayService {
             powerDisplayDto.setClassificationName(powerDisplayDto.getClassificationName().replaceAll("[^\\u4e00-\\u9fa5]", StringUtils.EMPTY));
         }
     }
-
 }