From 064d686a81d8537cdcb462b2cd0743fbdfa9c663 Mon Sep 17 00:00:00 2001
From: lyh <925863403@qq.com>
Date: 星期一, 01 九月 2025 16:46:54 +0800
Subject: [PATCH] 修改

---
 lxzn-module-eam/src/main/java/org/jeecg/modules/eam/service/impl/EamThirdMaintenanceOrderServiceImpl.java |  163 +++++++++++++++++++++++++++---------------------------
 1 files changed, 82 insertions(+), 81 deletions(-)

diff --git a/lxzn-module-eam/src/main/java/org/jeecg/modules/eam/service/impl/EamThirdMaintenanceOrderServiceImpl.java b/lxzn-module-eam/src/main/java/org/jeecg/modules/eam/service/impl/EamThirdMaintenanceOrderServiceImpl.java
index 682dfd7..229d553 100644
--- a/lxzn-module-eam/src/main/java/org/jeecg/modules/eam/service/impl/EamThirdMaintenanceOrderServiceImpl.java
+++ b/lxzn-module-eam/src/main/java/org/jeecg/modules/eam/service/impl/EamThirdMaintenanceOrderServiceImpl.java
@@ -58,6 +58,7 @@
 import java.time.LocalDate;
 import java.time.format.DateTimeFormatter;
 import java.util.*;
+import java.util.function.Supplier;
 import java.util.stream.Collectors;
 
 /**
@@ -708,41 +709,62 @@
      * */
     @Override
     public List<EchartsDto> selectList(String productionCode) {
-        List<EchartsDto> echartsDtos = new ArrayList<>();
+        // 鎻愬墠鍒濆鍖栧洓涓粺璁¢」瀵硅薄锛岃缃粯璁ゅ��0
+        Map<String, EchartsDto> echartsMap = new LinkedHashMap<>();
+        echartsMap.put("bysbzs", new EchartsDto("bysbzs", "鏈湀涓変繚璁″垝鎬绘暟", "0", null, null));
+        echartsMap.put("bwc", new EchartsDto("bwc", "鏈湀瀹屾垚", "0", null, null));
+        echartsMap.put("xysb", new EchartsDto("xysb", "涓嬫湀涓変繚璁″垝", "0", null, null));
+        echartsMap.put("xxysb", new EchartsDto("xxysb", "涓嬩笅鏈堜笁淇濊鍒�", "0", null, null));
+
         QueryWrapper<EamThirdMaintenanceOrder> queryWrapper = new QueryWrapper<>();
+        boolean hasValidCondition = false;
+
         if (StrUtil.isEmpty(productionCode)) {
-            // 鐢ㄦ埛鏁版嵁鏉冮檺妫�鏌�
             LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
-            if (sysUser == null) {
-                return echartsDtos; // 杩斿洖绌哄垪琛ㄨ�岄潪null
-            }
-            // 璁惧杩囨护閫昏緫
-            if (StringUtils.isNotBlank(sysUser.getEamEquipmentIds())) {
-                List<String> equipArr = Arrays.asList(sysUser.getEamEquipmentIds().split(","));
-                queryWrapper.in("e.equipment_code", equipArr);
-            } else {
-                List<BaseFactoryUser> baseFactoryUserList = baseFactoryUserService.list(new LambdaQueryWrapper<BaseFactoryUser>().eq(BaseFactoryUser::getUserId, sysUser.getId()));
-                if (CollectionUtils.isEmpty(baseFactoryUserList)) {
-                    return echartsDtos; // 杩斿洖绌哄垪琛ㄨ�岄潪null
+            if (sysUser != null) {
+                // 璁惧杩囨护閫昏緫
+                if (StringUtils.isNotBlank(sysUser.getEamEquipmentIds())) {
+                    List<String> equipArr = Arrays.asList(sysUser.getEamEquipmentIds().split(","));
+                    queryWrapper.in("e.equipment_code", equipArr);
+                    hasValidCondition = true;
+                } else {
+                    List<BaseFactoryUser> baseFactoryUserList = baseFactoryUserService.list(
+                            new LambdaQueryWrapper<BaseFactoryUser>().eq(BaseFactoryUser::getUserId, sysUser.getId())
+                    );
+                    if (!CollectionUtils.isEmpty(baseFactoryUserList)) {
+                        Set<String> factoryIds = baseFactoryUserList.stream()
+                                .map(BaseFactoryUser::getFactoryId)
+                                .collect(Collectors.toSet());
+                        Set<String> factoryCodes = baseFactoryService.listByIds(factoryIds).stream()
+                                .map(BaseFactory::getOrgCode)
+                                .collect(Collectors.toSet());
+                        queryWrapper.in("e.factory_org_code", factoryCodes);
+                        hasValidCondition = true;
+                    }
                 }
-                Set<String> factoryIds = baseFactoryUserList.stream().map(BaseFactoryUser::getFactoryId).collect(Collectors.toSet());
-                Set<String> factoryCodes = baseFactoryService.listByIds(factoryIds).stream().map(BaseFactory::getOrgCode).collect(Collectors.toSet());
+            }
+        } else {
+            List<BaseFactory> baseFactoryList = baseFactoryService.list(
+                    new LambdaQueryWrapper<BaseFactory>()
+                            .like(BaseFactory::getRemark, productionCode)
+                            .eq(BaseFactory::getDelFlag, CommonConstant.DEL_FLAG_0.toString())
+            );
+            if (!CollectionUtils.isEmpty(baseFactoryList)) {
+                Set<String> factoryCodes = new HashSet<>();
+                for (BaseFactory baseFactory : baseFactoryList) {
+                    List<BaseFactory> factoryList = baseFactoryService.queryFactoryByPid(baseFactory.getId());
+                    factoryCodes.addAll(factoryList.stream()
+                            .map(BaseFactory::getOrgCode)
+                            .collect(Collectors.toSet()));
+                }
                 queryWrapper.in("e.factory_org_code", factoryCodes);
+                hasValidCondition = true;
             }
-        }else {
-            //閫氳繃閫夋嫨鐨勮溅闂磋繃婊ゅ搴斾腑蹇�
-            List<BaseFactory> baseFactoryList = baseFactoryService.list(new LambdaQueryWrapper<BaseFactory>()
-                    .like(BaseFactory::getRemark, productionCode).eq(BaseFactory::getDelFlag, CommonConstant.DEL_FLAG_0.toString()));
-            if (CollectionUtils.isEmpty(baseFactoryList)) {
-                return echartsDtos;
-            }
-            //缁勮EAM鏁版嵁
-            Set<String> factoryCodes= new HashSet<>();
-            baseFactoryList.forEach(baseFactory -> {
-                List<BaseFactory> factoryList=baseFactoryService.queryFactoryByPid(baseFactory.getId());
-                factoryCodes.addAll(factoryList.stream().map(BaseFactory::getOrgCode).collect(Collectors.toSet()));
-            });
-            queryWrapper.in("e.factory_org_code", factoryCodes);
+        }
+
+        // 濡傛灉娌℃湁鏈夋晥鏌ヨ鏉′欢锛岀洿鎺ヨ繑鍥為粯璁ゅ�煎垪琛�
+        if (!hasValidCondition) {
+            return new ArrayList<>(echartsMap.values());
         }
 
         // 鑾峰彇褰撳墠骞存湀鍜屼笅涓や釜鏈堢殑骞存湀瀛楃涓�
@@ -751,61 +773,40 @@
         String nextNextMonth = LocalDate.now().plusMonths(2).format(DateTimeFormatter.ofPattern("yyyy-MM"));
 
         // 浠庡瓧鍏歌〃鑾峰彇閰嶇疆鍊�
-        String bysbjh = sysDictService.queryEnableDictItemsByCode("eam_home_third_number_bysbzs").get(0).getValue();
-        String bywc = sysDictService.queryEnableDictItemsByCode("eam_home_third_number_bwc").get(0).getValue();
-        String xysb = sysDictService.queryEnableDictItemsByCode("eam_home_third_number_xysb").get(0).getValue();
-        String xxysb = sysDictService.queryEnableDictItemsByCode("eam_home_third_number_xxysb").get(0).getValue();
+        Map<String, String> dictValues = new HashMap<>();
+        dictValues.put("bysbzs", sysDictService.queryEnableDictItemsByCode("eam_home_third_number_bysbzs").get(0).getValue());
+        dictValues.put("bwc", sysDictService.queryEnableDictItemsByCode("eam_home_third_number_bwc").get(0).getValue());
+        dictValues.put("xysb", sysDictService.queryEnableDictItemsByCode("eam_home_third_number_xysb").get(0).getValue());
+        dictValues.put("xxysb", sysDictService.queryEnableDictItemsByCode("eam_home_third_number_xxysb").get(0).getValue());
 
-        // 鏈湀涓変繚璁″垝鎬绘暟
-        if ("0".equals(bysbjh)) {
-            List<EamThirdMaintenanceOrder> currentMonthOrders = this.baseMapper.queryAsList(
-                    queryWrapper.clone().like("maintenance_date", currentYearMonth)
-            );
-            echartsDtos.add(new EchartsDto("bysbzs", "鏈湀涓変繚璁″垝鎬绘暟",
-                    String.valueOf(currentMonthOrders.size()), null, null));
+        // 鏇存柊缁熻椤癸紙浠呭湪閰嶇疆鍊间负"0"鏃舵煡璇㈠疄闄呮暟鎹級
+        updateEchartsValue(echartsMap, "bysbzs", dictValues,
+                () -> this.baseMapper.queryAsList(queryWrapper.clone().like("maintenance_date", currentYearMonth)).size());
+
+        updateEchartsValue(echartsMap, "bwc", dictValues,
+                () -> this.baseMapper.queryAsList(queryWrapper.clone()
+                        .like("maintenance_date", currentYearMonth)
+                        .eq("maintenance_status", ThirdMaintenanceStatusEnum.COMPLETE.name())
+                ).size());
+
+        updateEchartsValue(echartsMap, "xysb", dictValues,
+                () -> this.baseMapper.queryAsList(queryWrapper.clone().like("maintenance_date", nextMonth)).size());
+
+        updateEchartsValue(echartsMap, "xxysb", dictValues,
+                () -> this.baseMapper.queryAsList(queryWrapper.clone().like("maintenance_date", nextNextMonth)).size());
+
+        return new ArrayList<>(echartsMap.values());
+    }
+
+    // 杈呭姪鏂规硶锛氭洿鏂扮粺璁¢」鐨勫��
+    private void updateEchartsValue(Map<String, EchartsDto> echartsMap, String key,
+                                    Map<String, String> dictValues, Supplier<Integer> valueSupplier) {
+        if ("0".equals(dictValues.get(key))) {
+            int count = valueSupplier.get();
+            echartsMap.get(key).setValue(String.valueOf(count));
         } else {
-            echartsDtos.add(new EchartsDto("bysbzs", "鏈湀涓変繚璁″垝鎬绘暟",
-                    bysbjh, null, null));
+            echartsMap.get(key).setValue(dictValues.get(key));
         }
-
-        // 鏈湀瀹屾垚
-        if ("0".equals(bywc)) {
-            List<EamThirdMaintenanceOrder> currentMonthCompleted = this.baseMapper.queryAsList(
-                    queryWrapper.clone().like("maintenance_date", currentYearMonth)
-                            .eq("maintenance_status", ThirdMaintenanceStatusEnum.COMPLETE.name())
-            );
-            echartsDtos.add(new EchartsDto("bwc", "鏈湀瀹屾垚",
-                    String.valueOf(currentMonthCompleted.size()), null, null));
-        } else {
-            echartsDtos.add(new EchartsDto("bwc", "鏈湀瀹屾垚",
-                    bywc, null, null));
-        }
-
-        // 涓嬫湀涓変繚璁″垝
-        if ("0".equals(xysb)) {
-            List<EamThirdMaintenanceOrder> nextMonthOrders = this.baseMapper.queryAsList(
-                    queryWrapper.clone().like("maintenance_date", nextMonth)
-            );
-            echartsDtos.add(new EchartsDto("xysb", "涓嬫湀涓変繚璁″垝",
-                    String.valueOf(nextMonthOrders.size()), null, null));
-        } else {
-            echartsDtos.add(new EchartsDto("xysb", "涓嬫湀涓変繚璁″垝",
-                    xysb, null, null));
-        }
-
-        // 涓嬩笅鏈堜笁淇濊鍒�
-        if ("0".equals(xxysb)) {
-            List<EamThirdMaintenanceOrder> nextNextMonthOrders = this.baseMapper.queryAsList(
-                    queryWrapper.clone().like("maintenance_date", nextNextMonth)
-            );
-            echartsDtos.add(new EchartsDto("xxysb", "涓嬩笅鏈堜笁淇濊鍒�",
-                    String.valueOf(nextNextMonthOrders.size()), null, null));
-        } else {
-            echartsDtos.add(new EchartsDto("xxysb", "涓嬩笅鏈堜笁淇濊鍒�",
-                    xxysb, null, null));
-        }
-
-        return echartsDtos;
     }
 
     /**

--
Gitblit v1.9.3