From fb5aa087eaf6a68cf7ea8114beb270551016e7dd Mon Sep 17 00:00:00 2001
From: lyh <925863403@qq.com>
Date: 星期二, 29 七月 2025 11:27:45 +0800
Subject: [PATCH] 添加定型指派设备

---
 lxzn-module-eam-common/src/main/java/org/jeecg/modules/eam/service/impl/EamEquipmentServiceImpl.java |  230 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 228 insertions(+), 2 deletions(-)

diff --git a/lxzn-module-eam-common/src/main/java/org/jeecg/modules/eam/service/impl/EamEquipmentServiceImpl.java b/lxzn-module-eam-common/src/main/java/org/jeecg/modules/eam/service/impl/EamEquipmentServiceImpl.java
index 9122818..c2ec226 100644
--- a/lxzn-module-eam-common/src/main/java/org/jeecg/modules/eam/service/impl/EamEquipmentServiceImpl.java
+++ b/lxzn-module-eam-common/src/main/java/org/jeecg/modules/eam/service/impl/EamEquipmentServiceImpl.java
@@ -1,7 +1,17 @@
 package org.jeecg.modules.eam.service.impl;
 
+import cn.hutool.core.collection.CollectionUtil;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.apache.commons.lang3.StringUtils;
+import org.apache.shiro.SecurityUtils;
 import org.jeecg.common.constant.CommonConstant;
+import org.jeecg.common.constant.DataBaseConstant;
+import org.jeecg.common.system.vo.LoginUser;
+import org.jeecg.common.util.oConvertUtils;
 import org.jeecg.modules.eam.aspect.annotation.EquipmentHistoryLog;
 import org.jeecg.modules.eam.constant.AssetStatusEnum;
 import org.jeecg.modules.eam.constant.EquipmentMaintenanceStatus;
@@ -10,18 +20,29 @@
 import org.jeecg.modules.eam.entity.EamEquipment;
 import org.jeecg.modules.eam.entity.EamEquipmentExtend;
 import org.jeecg.modules.eam.mapper.EamEquipmentMapper;
+import org.jeecg.modules.eam.request.EamEquipmentQuery;
 import org.jeecg.modules.eam.service.IEamEquipmentExtendService;
 import org.jeecg.modules.eam.service.IEamEquipmentService;
+import org.jeecg.modules.eam.tree.FindsEquipmentProductionUtil;
+import org.jeecg.modules.eam.vo.EamEquipmentTree;
+import org.jeecg.modules.eam.vo.EquipmentSearchResult;
+import org.jeecg.modules.system.entity.MdcProduction;
+import org.jeecg.modules.system.service.IMdcProductionService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
 import javax.annotation.Resource;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * @Description: 璁惧鍙拌处
  * @Author: jeecg-boot
- * @Date:   2025-03-19
+ * @Date: 2025-03-19
  * @Version: V1.0
  */
 @Service
@@ -31,6 +52,8 @@
     private EamEquipmentMapper eamEquipmentMapper;
     @Autowired
     private IEamEquipmentExtendService equipmentExtendService;
+    @Autowired
+    private IMdcProductionService mdcProductionService;
 
     @Override
     @Transactional(rollbackFor = Exception.class)
@@ -52,7 +75,210 @@
 
         equipmentExtendService.save(eamEquipmentExtend);
 
-        //鎻掑叆璁惧灞ュ巻 TODO
+        //鎻掑叆璁惧灞ュ巻   @EquipmentHistoryLog
         return eamEquipment;
     }
+
+    @Override
+    public List<EamEquipmentTree> loadTreeListByProductionIds(String ids) {
+        List<String> productionIds = Arrays.asList(ids.split(","));
+        //鑾峰彇鎵�鏈変骇绾挎暟鎹�
+        List<MdcProduction> productionList = mdcProductionService.list(new LambdaQueryWrapper<MdcProduction>().eq(MdcProduction::getDelFlag, CommonConstant.DEL_FLAG_0.toString()).orderByAsc(MdcProduction::getProductionOrder));
+        List<String> allProductionIds = new ArrayList<>();
+        //鎵惧埌鎵�鏈変骇绾縤d鐨勪笂绾d
+        if (!productionIds.isEmpty()) {
+            for (String productionId : productionIds) {
+                this.getAllProductionIds(productionList, productionId, allProductionIds);
+            }
+        }
+        //杩囨护浜х嚎鏁版嵁
+        List<MdcProduction> list = productionList.stream().filter((MdcProduction mdcProduction) -> allProductionIds.contains(mdcProduction.getId())).collect(Collectors.toList());
+        //缁勮浜х嚎璁惧鏍�
+        List<EamEquipmentTree> treeList = FindsEquipmentProductionUtil.wrapEquipmentProductionTreeList(list);
+        //濉厖璁惧鏁版嵁
+        fillEquipmentByProduction(treeList);
+        return treeList;
+    }
+
+    @Override
+    public IPage<EamEquipment> queryPageList(IPage<EamEquipment> page, EamEquipmentQuery eamEquipment) {
+        QueryWrapper<EamEquipment> queryWrapper = new QueryWrapper<>();
+        queryWrapper.eq("e.del_flag", CommonConstant.DEL_FLAG_0);
+        //鐢ㄦ埛鏁版嵁鏉冮檺
+        LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
+        if (sysUser == null) {
+            return page;
+        }
+        if (StringUtils.isNotBlank(sysUser.getEquipmentIds())) {
+            //閫夋嫨浜嗚澶囷紝鏍规嵁璁惧id杩囨护璁惧
+            List<String> equipArr = Arrays.asList(sysUser.getEquipmentIds().split(","));
+            queryWrapper.in("e.equipment_code", equipArr);
+        } else {
+            //娌℃湁閫夋嫨璁惧锛屾牴鎹溅闂磋繃婊よ澶�
+            queryWrapper.exists("select 1 from mdc_user_production t where t.user_id={0} and t.pro_id=e.org_id", sysUser.getId());
+        }
+        //鏌ヨ鏉′欢杩囨护
+        if (eamEquipment != null) {
+            if (StringUtils.isNotBlank(eamEquipment.getEquipmentCode())) {
+                queryWrapper.like("e.equipment_code", eamEquipment.getEquipmentCode());
+            }
+            if (StringUtils.isNotBlank(eamEquipment.getEquipmentName())) {
+                queryWrapper.like("e.equipment_name", eamEquipment.getEquipmentName());
+            }
+            if (StringUtils.isNotBlank(eamEquipment.getEquipmentImportance())) {
+                queryWrapper.eq("e.equipment_importance", eamEquipment.getEquipmentImportance());
+            }
+            if (StringUtils.isNotBlank(eamEquipment.getAssetStatus())) {
+                queryWrapper.like("e.asset_status", eamEquipment.getAssetStatus());
+            }
+            if (StringUtils.isNotBlank(eamEquipment.getTechnologyStatus())) {
+                queryWrapper.like("e.technology_status", eamEquipment.getTechnologyStatus());
+            }
+            if (StringUtils.isNotBlank(eamEquipment.getOperationSystem())) {
+                queryWrapper.like("e.operation_system", eamEquipment.getOperationSystem());
+            }
+            if (StringUtils.isNotBlank(eamEquipment.getOrgId())) {
+                //杞﹂棿缂栫爜涓嶄负绌�
+                List<String> productIds = mdcProductionService.recursionChildren(eamEquipment.getOrgId());
+                if (CollectionUtil.isNotEmpty(productIds)) {
+                    //涓嶅仛绛涢��
+                    queryWrapper.in("e.org_id", productIds);
+                }
+            }
+            if (StringUtils.isNotBlank(eamEquipment.getEquipmentCategory())) {
+                queryWrapper.like("e.equipment_category", eamEquipment.getEquipmentCategory());
+            }
+            if (StringUtils.isNotBlank(eamEquipment.getId())) {
+                queryWrapper.eq("e.id", eamEquipment.getId());
+            }
+            //鎺掑簭
+            if (StringUtils.isNotBlank(eamEquipment.getColumn()) && StringUtils.isNotBlank(eamEquipment.getOrder())) {
+                //queryWrapper.like("column", eamEquipment.getColumn());
+                String column = eamEquipment.getColumn();
+                if (column.endsWith(CommonConstant.DICT_TEXT_SUFFIX)) {
+                    column = column.substring(0, column.lastIndexOf(CommonConstant.DICT_TEXT_SUFFIX));
+                }
+                if (DataBaseConstant.SQL_ASC.equalsIgnoreCase(eamEquipment.getOrder())) {
+                    queryWrapper.orderByAsc("e." + oConvertUtils.camelToUnderline(column));
+                } else {
+                    queryWrapper.orderByDesc("e." + oConvertUtils.camelToUnderline(column));
+                }
+            } else {
+                queryWrapper.orderByDesc("e.create_time");
+            }
+        } else {
+            queryWrapper.orderByDesc("e.create_time");
+        }
+
+        IPage<EamEquipment> ipage = eamEquipmentMapper.queryPageList(page, queryWrapper);
+        return ipage;
+    }
+
+    @Override
+    public List<EquipmentSearchResult> asyncLoadEquipment(String keyword, Integer pageSize, String id) {
+        if (StringUtils.isNotBlank(id)) {
+            EamEquipment eamEquipment = eamEquipmentMapper.selectById(id);
+            if (eamEquipment != null) {
+                List<EquipmentSearchResult> resultList = new ArrayList<>();
+                resultList.add(new EquipmentSearchResult(eamEquipment));
+                return resultList;
+            }
+        }
+        IPage<EamEquipment> page = new Page<>(1, pageSize);
+        QueryWrapper<EamEquipment> queryWrapper = new QueryWrapper<>();
+        //鐢ㄦ埛鏁版嵁鏉冮檺
+        LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
+        if (sysUser == null) {
+            return Collections.emptyList();
+        }
+        if (StringUtils.isNotBlank(sysUser.getEquipmentIds())) {
+            //閫夋嫨浜嗚澶囷紝鏍规嵁璁惧id杩囨护璁惧
+            List<String> equipArr = Arrays.asList(sysUser.getEquipmentIds().split(","));
+            queryWrapper.in("equipment_code", equipArr);
+        } else {
+            //娌℃湁閫夋嫨璁惧锛屾牴鎹溅闂磋繃婊よ澶�
+            queryWrapper.exists("select 1 from mdc_user_production t where t.user_id={0} and t.pro_id=org_id", sysUser.getId());
+        }
+        if (StringUtils.isNotBlank(keyword)) {
+            queryWrapper.like("equipment_code", keyword);
+            queryWrapper.or().like("equipment_name", keyword);
+        }
+        queryWrapper.eq("del_flag", CommonConstant.DEL_FLAG_0);
+        IPage<EamEquipment> pageResult = eamEquipmentMapper.queryPageList(page, queryWrapper);
+        if (pageResult != null && CollectionUtil.isNotEmpty(pageResult.getRecords())) {
+            List<EquipmentSearchResult> resultList = new ArrayList<>();
+            pageResult.getRecords().forEach((record) -> {
+                resultList.add(new EquipmentSearchResult(record));
+            });
+            return resultList;
+        }
+        return Collections.emptyList();
+    }
+
+    @Override
+    public EamEquipment selectByEquipmentCode(String equipmentCode) {
+        if (StringUtils.isBlank(equipmentCode)) {
+            return null;
+        }
+        QueryWrapper<EamEquipment> queryWrapper = new QueryWrapper<>();
+        //鐢ㄦ埛鏁版嵁鏉冮檺
+        LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
+        if (sysUser == null) {
+            return null;
+        }
+        if (StringUtils.isNotBlank(sysUser.getEquipmentIds())) {
+            //閫夋嫨浜嗚澶囷紝鏍规嵁璁惧id杩囨护璁惧
+            List<String> equipArr = Arrays.asList(sysUser.getEquipmentIds().split(","));
+            queryWrapper.in("equipment_code", equipArr);
+        } else {
+            //娌℃湁閫夋嫨璁惧锛屾牴鎹溅闂磋繃婊よ澶�
+            queryWrapper.exists("select 1 from mdc_user_production t where t.user_id={0} and t.pro_id=org_id", sysUser.getId());
+        }
+        queryWrapper.eq("equipment_code", equipmentCode);
+        queryWrapper.eq("del_flag", CommonConstant.DEL_FLAG_0);
+
+        return eamEquipmentMapper.selectOne(queryWrapper);
+    }
+
+    /**
+     * 鑾峰彇鎵�鏈夌殑浜х嚎id锛堝寘鍚墍鏈変笂绾э級
+     */
+    private void getAllProductionIds(List<MdcProduction> productionList, String productionId, List<String> allProductionIds) {
+        if (!allProductionIds.contains(productionId)) {
+            allProductionIds.add(productionId);
+        }
+        for (MdcProduction mdcProduction : productionList) {
+            if (StringUtils.isEmpty(mdcProduction.getParentId())) {
+                continue;
+            }
+            if (productionId.equals(mdcProduction.getId())) {
+                if (!allProductionIds.contains(mdcProduction.getParentId())) {
+                    allProductionIds.add(mdcProduction.getParentId());
+                    getAllProductionIds(productionList, mdcProduction.getParentId(), allProductionIds);
+                }
+            }
+        }
+    }
+
+    /**
+     * 浜х嚎璁惧鏍戝~鍏呰澶囨暟鎹�
+     */
+    private void fillEquipmentByProduction(List<EamEquipmentTree> treeList) {
+        for (EamEquipmentTree mdcEquipmentTree : treeList) {
+            List<EamEquipment> equipmentList = eamEquipmentMapper.queryByProductionId(mdcEquipmentTree.getKey());
+            if (CollectionUtil.isNotEmpty(equipmentList)) {
+                for (EamEquipment mdcEquipment : equipmentList) {
+                    EamEquipmentTree tree = new EamEquipmentTree().convert(mdcEquipment);
+                    tree.setParentId(mdcEquipmentTree.getKey());
+                    tree.setType(2);
+                    mdcEquipmentTree.getChildren().add(tree);
+                }
+                mdcEquipmentTree.setLeaf(false);
+            }
+            if (CollectionUtil.isNotEmpty(mdcEquipmentTree.getChildren())) {
+                fillEquipmentByProduction(mdcEquipmentTree.getChildren());
+            }
+        }
+    }
+
 }

--
Gitblit v1.9.3