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 |  216 ++++++++++++++++++++++++++++++++++++++++-------------
 1 files changed, 161 insertions(+), 55 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 ffdd2f4..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
@@ -19,6 +19,7 @@
 import org.jeecg.common.constant.CommonConstant;
 import org.jeecg.common.constant.DataBaseConstant;
 import org.jeecg.common.exception.JeecgBootException;
+import org.jeecg.common.system.vo.DictModel;
 import org.jeecg.common.system.vo.LoginUser;
 import org.jeecg.common.util.StrUtils;
 import org.jeecg.common.util.oConvertUtils;
@@ -57,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;
 
 /**
@@ -704,42 +706,65 @@
     /**
      * 璁惧绠$悊棣栭〉-涓変繚璁″垝
      * @return
-     */
+     * */
     @Override
-    public List<EchartsDto> selectList() {
-        List<EchartsDto> echartsDtos = new ArrayList<>();
-
-        // 鐢ㄦ埛鏁版嵁鏉冮檺妫�鏌�
-        LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
-        if (sysUser == null) {
-            return echartsDtos; // 杩斿洖绌哄垪琛ㄨ�岄潪null
-        }
+    public List<EchartsDto> selectList(String productionCode) {
+        // 鎻愬墠鍒濆鍖栧洓涓粺璁¢」瀵硅薄锛岃缃粯璁ゅ��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 (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 (StrUtil.isEmpty(productionCode)) {
+            LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
+            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;
+                    }
+                }
             }
+        } 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;
+            }
+        }
 
-            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);
+        // 濡傛灉娌℃湁鏈夋晥鏌ヨ鏉′欢锛岀洿鎺ヨ繑鍥為粯璁ゅ�煎垪琛�
+        if (!hasValidCondition) {
+            return new ArrayList<>(echartsMap.values());
         }
 
         // 鑾峰彇褰撳墠骞存湀鍜屼笅涓や釜鏈堢殑骞存湀瀛楃涓�
@@ -747,36 +772,117 @@
         String nextMonth = LocalDate.now().plusMonths(1).format(DateTimeFormatter.ofPattern("yyyy-MM"));
         String nextNextMonth = LocalDate.now().plusMonths(2).format(DateTimeFormatter.ofPattern("yyyy-MM"));
 
-        // 鏈湀涓変繚璁″垝鎬绘暟
-        List<EamThirdMaintenanceOrder> currentMonthOrders = this.baseMapper.queryAsList(
-                queryWrapper.clone().like("maintenance_date", currentYearMonth)
-        );
-        echartsDtos.add(new EchartsDto("bysbzs", "鏈湀涓変繚璁″垝鎬绘暟",
-                String.valueOf(currentMonthOrders.size()), null, null));
+        // 浠庡瓧鍏歌〃鑾峰彇閰嶇疆鍊�
+        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());
 
-        // 鏈湀瀹屾垚
-        List<EamThirdMaintenanceOrder> currentMonthCompleted = this.baseMapper.queryAsList(
-                queryWrapper.clone().like("maintenance_date", currentYearMonth)
+        // 鏇存柊缁熻椤癸紙浠呭湪閰嶇疆鍊间负"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())
-        );
-        echartsDtos.add(new EchartsDto("bwc", "鏈湀瀹屾垚",
-                String.valueOf(currentMonthCompleted.size()), null, null));
+                ).size());
 
-        // 涓嬫湀涓変繚璁″垝
-        List<EamThirdMaintenanceOrder> nextMonthOrders = this.baseMapper.queryAsList(
-                queryWrapper.clone().like("maintenance_date", nextMonth)
-        );
-        echartsDtos.add(new EchartsDto("xysb", "涓嬫湀涓変繚璁″垝",
-                String.valueOf(nextMonthOrders.size()), null, null));
+        updateEchartsValue(echartsMap, "xysb", dictValues,
+                () -> this.baseMapper.queryAsList(queryWrapper.clone().like("maintenance_date", nextMonth)).size());
 
-        // 涓嬩笅鏈堜笁淇濊鍒�
-        List<EamThirdMaintenanceOrder> nextNextMonthOrders = this.baseMapper.queryAsList(
-                queryWrapper.clone().like("maintenance_date", nextNextMonth)
-        );
-        echartsDtos.add(new EchartsDto("xxysb", "涓嬩笅鏈堜笁淇濊鍒�",
-                String.valueOf(nextNextMonthOrders.size()), null, null));
+        updateEchartsValue(echartsMap, "xxysb", dictValues,
+                () -> this.baseMapper.queryAsList(queryWrapper.clone().like("maintenance_date", nextNextMonth)).size());
 
-        return echartsDtos;
+        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 {
+            echartsMap.get(key).setValue(dictValues.get(key));
+        }
+    }
+
+    /**
+     * 璁惧绠$悊棣栭〉-涓変繚璁″垝鍒楄〃
+     * @param code
+     * @param page
+     * @return
+     */
+    @Override
+    public IPage<EamThirdMaintenanceOrder> maintenanceList(String code,String productionCode, Page<EamThirdMaintenanceOrder> page){
+        IPage<EamThirdMaintenanceOrder> eamThirdMaintenanceOrderIPage=new Page<>();
+        QueryWrapper<EamThirdMaintenanceOrder> queryWrapper = new QueryWrapper<>();
+        if (StrUtil.isEmpty(productionCode)) {
+            // 鐢ㄦ埛鏁版嵁鏉冮檺妫�鏌�
+            LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
+            if (sysUser == null) {
+                return eamThirdMaintenanceOrderIPage; // 杩斿洖绌哄垪琛ㄨ�岄潪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 eamThirdMaintenanceOrderIPage; // 杩斿洖绌哄垪琛ㄨ�岄潪null
+                }
+                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);
+            }
+        }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 eamThirdMaintenanceOrderIPage;
+            }
+            //缁勮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);
+        }
+        queryWrapper.orderByDesc("wmo.create_time");
+        // 鑾峰彇褰撳墠骞存湀鍜屼笅涓や釜鏈堢殑骞存湀瀛楃涓�
+        String currentYearMonth = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyy-MM"));
+        switch (code){
+            case "bysbzs":
+                // 鏈湀涓変繚璁″垝鎬绘暟
+                eamThirdMaintenanceOrderIPage=this.baseMapper.queryPageList(page,queryWrapper.clone().like("maintenance_date", currentYearMonth));
+                break;
+            case "bwc":
+                // 鏈湀瀹屾垚
+                eamThirdMaintenanceOrderIPage=this.baseMapper.queryPageList(page,queryWrapper.clone().like("maintenance_date", currentYearMonth)
+                        .eq("maintenance_status", ThirdMaintenanceStatusEnum.COMPLETE.name()));
+                break;
+            case "xysb":
+                // 涓嬫湀涓変繚璁″垝
+                String nextMonth = LocalDate.now().plusMonths(1).format(DateTimeFormatter.ofPattern("yyyy-MM"));
+                eamThirdMaintenanceOrderIPage=this.baseMapper.queryPageList(page,queryWrapper.clone().like("maintenance_date", nextMonth));
+                break;
+            case "xxysb":
+                // 涓嬩笅鏈堜笁淇濊鍒�
+                String nextNextMonth = LocalDate.now().plusMonths(2).format(DateTimeFormatter.ofPattern("yyyy-MM"));
+                eamThirdMaintenanceOrderIPage=this.baseMapper.queryPageList(page,queryWrapper.clone().like("maintenance_date", nextNextMonth));
+                break;
+            default:
+                break;
+        }
+        eamThirdMaintenanceOrderIPage.getRecords().forEach(eamThirdMaintenanceOrder -> {
+            eamThirdMaintenanceOrder.setTechnologyStatus(eamEquipmentExtendService.getById(eamThirdMaintenanceOrder.getEquipmentId()).getTechnologyStatus());
+        });
+        return eamThirdMaintenanceOrderIPage;
     }
 
     /**

--
Gitblit v1.9.3