From 23855599412c4d61b38d78f0f3abd3430a48b5b1 Mon Sep 17 00:00:00 2001
From: zhangherong <571457620@qq.com>
Date: 星期三, 25 六月 2025 11:51:38 +0800
Subject: [PATCH] Merge branch 'mdc_hyjs_master'

---
 lxzn-module-dnc/src/main/java/org/jeecg/modules/dnc/service/impl/WorkStepServiceImpl.java |  485 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 485 insertions(+), 0 deletions(-)

diff --git a/lxzn-module-dnc/src/main/java/org/jeecg/modules/dnc/service/impl/WorkStepServiceImpl.java b/lxzn-module-dnc/src/main/java/org/jeecg/modules/dnc/service/impl/WorkStepServiceImpl.java
new file mode 100644
index 0000000..56a9a0d
--- /dev/null
+++ b/lxzn-module-dnc/src/main/java/org/jeecg/modules/dnc/service/impl/WorkStepServiceImpl.java
@@ -0,0 +1,485 @@
+package org.jeecg.modules.dnc.service.impl;
+
+import cn.hutool.core.bean.BeanUtil;
+import cn.hutool.core.util.StrUtil;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.apache.commons.collections4.ListUtils;
+import org.apache.shiro.SecurityUtils;
+import org.jeecg.common.system.vo.LoginUser;
+import org.jeecg.modules.dnc.constant.DocAttributionTypeEnum;
+import org.jeecg.modules.dnc.entity.*;
+import org.jeecg.modules.dnc.exception.ExceptionCast;
+import org.jeecg.modules.dnc.mapper.WorkStepMapper;
+import org.jeecg.modules.dnc.request.DocInfoQueryRequest;
+import org.jeecg.modules.dnc.request.TreeInfoRequest;
+import org.jeecg.modules.dnc.response.CommonCode;
+import org.jeecg.modules.dnc.response.ProcessInfoCode;
+import org.jeecg.modules.dnc.response.ProductInfoCode;
+import org.jeecg.modules.dnc.response.UcenterCode;
+import org.jeecg.modules.dnc.service.*;
+import org.jeecg.modules.dnc.utils.ValidateUtil;
+import org.jeecg.modules.system.entity.MdcProduction;
+import org.jeecg.modules.system.entity.SysUser;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Lazy;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.*;
+import java.util.stream.Collectors;
+
+@Service
+public class WorkStepServiceImpl extends ServiceImpl<WorkStepMapper, WorkStep> implements IWorkStepService {
+    @Autowired
+    private IPermissionStreamNewService permissionStreamNewService;
+
+    @Autowired
+    private IWorkStepDepartmentService workStepDepartmentService;
+
+    @Autowired
+    private IWorkStepPermissionService workStepPermissionService;
+
+    @Autowired
+    private IDocRelativeService iDocRelativeService;
+
+    @Autowired
+    private IProductPermissionService productPermissionService;
+
+    @Autowired
+    private IProductMixService productMixService;
+
+    @Autowired
+    @Lazy
+    private IDocInfoService docInfoService;
+    @Autowired
+    private IDeviceTypeService deviceTypeService;
+
+    @Override
+    public List<WorkStep> getByUserPerms(String userId) {
+        if(!ValidateUtil.validateString(userId))
+            return Collections.emptyList();
+        return super.getBaseMapper().getByUserPerms(userId);
+    }
+
+    /**
+     * 鏍规嵁鐢ㄦ埛id鑾峰彇鎺堟潈鐨勫伐姝ヤ俊鎭�
+     * @param userId
+     * @param queryParam 鏌ヨ鏉′欢
+     * @return
+     */
+    @Override
+    public List<WorkStep> getByUserPerms(String userId,String queryParam){
+        if(!ValidateUtil.validateString(userId))
+            return Collections.emptyList();
+        if(!ValidateUtil.validateString(queryParam))
+            return Collections.emptyList();
+        LambdaQueryWrapper<WorkStep> queryWrapper = Wrappers.lambdaQuery();
+        if(ValidateUtil.validateString(queryParam)) {
+            queryWrapper.and(wrapper->wrapper.like(WorkStep::getStepName, queryParam)
+                    .or()
+                    .like(WorkStep::getStepName, queryParam));
+        }
+        queryWrapper.orderByAsc(WorkStep::getCreateTime);
+        return super.list(queryWrapper);
+    }
+
+    /**
+     * 鏂板浜у搧淇℃伅
+     * @param workStep
+     * @return
+     */
+    @Override
+    public boolean addWorkStep(WorkStep workStep){
+        if(workStep == null)
+            ExceptionCast.cast(CommonCode.INVALID_PARAM);
+        if(!ValidateUtil.validateString(workStep.getProductId()))
+            ExceptionCast.cast(ProcessInfoCode.PROCESS_PRODUCT_NONE);
+        if(!ValidateUtil.validateString(workStep.getComponentId()))
+            ExceptionCast.cast(ProcessInfoCode.PROCESS_COMPONENT_NONE);
+        if(!ValidateUtil.validateString(workStep.getStepName()))
+            ExceptionCast.cast(ProcessInfoCode.PROCESS_NAME_NONE);
+        if(!ValidateUtil.validateString(workStep.getStepCode()))
+            ExceptionCast.cast(ProcessInfoCode.PROCESS_CODE_NONE);
+        LoginUser user = (LoginUser) SecurityUtils.getSubject().getPrincipal();
+        String userId = user.getId();
+        if(!ValidateUtil.validateString(userId))
+            ExceptionCast.cast(UcenterCode.UCENTER_ACCOUNT_NOT_EXIST);
+        WorkStep en =getByWorkStepNo(workStep.getProcessId(), workStep.getCraftNo());
+        if(en != null) {
+            ExceptionCast.cast(ProcessInfoCode.WORKSTEP_NOT_EXIST);
+        }
+        boolean b =super.save(workStep);
+        if(!b)
+            ExceptionCast.cast(CommonCode.FAIL);
+        //娣诲姞閮ㄩ棬鏉冮檺
+        List<PermissionStreamNew> oldDepartPermList = permissionStreamNewService.loadPermissionStreamNewByBusinessId(workStep.getProcessId(), DocAttributionTypeEnum.PROCESS.getCode().toString(),"1");
+        if(oldDepartPermList != null && !oldDepartPermList.isEmpty()) {
+            List<WorkStepDepartment> workStepDepartmentList = new ArrayList<>();
+            List<PermissionStreamNew> permissionStreamList = new ArrayList<>();
+            oldDepartPermList.forEach(item -> {
+                WorkStepDepartment pd = new WorkStepDepartment();
+                pd.setDepartId(item.getDepartId());
+                pd.setStepId(workStep.getId());
+                workStepDepartmentList.add(pd);
+                PermissionStreamNew perm = new PermissionStreamNew();
+                perm.setDepartId(item.getDepartId());
+                perm.setBusinessId(workStep.getId());
+                perm.setBusinessType(DocAttributionTypeEnum.WORKSITE.getCode().toString());
+                permissionStreamList.add(perm);
+            });
+            if(!workStepDepartmentList.isEmpty()) {
+                b = workStepDepartmentService.saveBatch(workStepDepartmentList);
+                if(!b)
+                    ExceptionCast.cast(CommonCode.FAIL);
+            }
+            if(!permissionStreamList.isEmpty()) {
+                b = permissionStreamNewService.saveBatch(permissionStreamList);
+                if(!b)
+                    ExceptionCast.cast(CommonCode.FAIL);
+            }
+        }
+        //娣诲姞鐢ㄦ埛鏉冮檺
+        b = productPermissionService.add(workStep.getId(), userId,DocAttributionTypeEnum.WORKSITE.getCode().toString());
+        if (!b) {
+            ExceptionCast.cast(ProductInfoCode.PRODUCT_SAVE_ERROR);
+        }
+        //娣诲姞缁撴瀯鏍�
+        ProductMix productMix = new ProductMix(Long.parseLong(workStep.getId()),Long.parseLong(workStep.getProcessId())
+                ,workStep.getStepName(),workStep.getStepCode(),DocAttributionTypeEnum.WORKSITE.getCode(),new Date());
+        productMixService.save(productMix);
+        //娣诲姞鏉冮檺楠岃瘉
+        PermissionStreamNew permissionStream = new PermissionStreamNew();
+        permissionStream.setUserId(userId);
+        permissionStream.setBusinessId(workStep.getId());
+        permissionStream.setBusinessType(DocAttributionTypeEnum.WORKSITE.getCode().toString());
+        return permissionStreamNewService.addPermissionStreamNew(permissionStream);
+    }
+
+    /**
+     * 缂栬緫浜у搧淇℃伅
+     * @param id
+     * @param workStep
+     * @return
+     */
+    @Override
+    public boolean editWorkStep(String id ,WorkStep workStep){
+        if(!ValidateUtil.validateString(id) || workStep == null)
+            ExceptionCast.cast(CommonCode.INVALID_PARAM);
+        if(!ValidateUtil.validateString(workStep.getStepName()))
+            ExceptionCast.cast(ProcessInfoCode.PROCESS_NAME_NONE);
+        WorkStep en = super.getById(id);
+        if(en == null)
+            ExceptionCast.cast(ProcessInfoCode.WORKSTEP_NOT_EXIST);
+        //鍚屾淇敼缁撴瀯鏍�
+        ProductMix productMix = productMixService.getById(Long.parseLong(id));
+        productMix.setTreeName(workStep.getStepName());
+        productMix.setTreeCode(workStep.getStepCode());
+        productMixService.updateById(productMix);
+        return super.updateById(workStep);
+    }
+
+    @Override
+    public WorkStep getByWorkStepNo(String processId,String craftNo){
+        if(ValidateUtil.validateString(processId)) {
+            List<WorkStep> list = super.lambdaQuery().eq(WorkStep::getProcessId, processId).eq(WorkStep::getCraftNo,craftNo).list();
+            if(list == null || list.isEmpty())
+                return null;
+            return list.get(0);
+        }
+        return null;
+
+    }
+
+    @Override
+    @Transactional(rollbackFor = {Exception.class})
+    public boolean assignAddDepart(WorkStep workStep, Collection<MdcProduction> departmentList) {
+        if(workStep == null || departmentList == null || departmentList.isEmpty())
+            ExceptionCast.cast(CommonCode.INVALID_PARAM);
+        List<WorkStepDepartment> workStepDepartmentList = new ArrayList<>();
+        List<PermissionStreamNew> permissionStreamList = new ArrayList<>();
+        departmentList.forEach(item -> {
+            WorkStepDepartment en = workStepDepartmentService.getByStepIdAndDepartId(workStep.getId(), item.getId());
+            if(en == null) {
+                en = new WorkStepDepartment();
+                en.setDepartId(item.getId());
+                en.setStepId(workStep.getId());
+                workStepDepartmentList.add(en);
+            }
+            PermissionStreamNew stream = permissionStreamNewService.loadPermissionStreamNewByBusinessIdAndDepartId(workStep.getId(),item.getId(),DocAttributionTypeEnum.WORKSITE.getCode().toString());
+            if(stream == null) {
+                stream = new PermissionStreamNew();
+                stream.setDepartId(item.getId());
+                stream.setBusinessId(workStep.getId());
+                stream.setBusinessType(DocAttributionTypeEnum.WORKSITE.getCode().toString());
+                permissionStreamList.add(stream);
+            }
+        });
+        if(!workStepDepartmentList.isEmpty()) {
+            boolean b = workStepDepartmentService.saveBatch(workStepDepartmentList);
+            if(!b) {
+                ExceptionCast.cast(ProductInfoCode.PRODUCT_USER_PERM_ERROR);
+            }
+        }
+        if(!permissionStreamList.isEmpty()) {
+            boolean b = permissionStreamNewService.saveBatch(permissionStreamList);
+            if(!b) {
+                ExceptionCast.cast(ProductInfoCode.PRODUCT_USER_PERM_ERROR);
+            }
+        }
+        return true;
+    }
+
+    @Override
+    public List<WorkStep> getByProcessIds(String productId,List<String> processIds) {
+        if(!ValidateUtil.validateString(productId) || processIds == null || processIds.isEmpty())
+            return null;
+        List<WorkStep> list = new ArrayList<>();
+        if(processIds.size() > 1000){
+            List<List<String>> processListArr = ListUtils.partition(processIds, 100);
+            for(List<String> processList : processListArr){
+                List<WorkStep> workStepList = super.lambdaQuery().eq(WorkStep::getProductId, productId).in(WorkStep::getProcessId, processList).list();
+                if(workStepList != null && !workStepList.isEmpty()){
+                    list.addAll(workStepList);
+                }
+            }
+        }else {
+            list = super.lambdaQuery().eq(WorkStep::getProductId, productId).in(WorkStep::getProcessId, processIds).list();
+        }
+        if(list == null || list.isEmpty())
+            return null;
+        return list;
+    }
+
+    /**
+     * 鍒嗛厤閮ㄩ棬鏉冮檺
+     * @param workStep
+     * @param departmentList
+     * @return
+     */
+    @Override
+    public boolean assignRemoveDepart(WorkStep workStep, Collection<MdcProduction> departmentList){
+        if(workStep == null || departmentList == null || departmentList.isEmpty())
+            ExceptionCast.cast(CommonCode.INVALID_PARAM);
+        List<WorkStepDepartment> workStepDepartmentList = new ArrayList<>();
+        List<PermissionStreamNew> permissionStreamList = new ArrayList<>();
+        departmentList.forEach(item -> {
+            WorkStepDepartment en = workStepDepartmentService.getByStepIdAndDepartId(workStep.getId(), item.getId());
+            if(en != null) {
+                workStepDepartmentList.add(en);
+            }
+            PermissionStreamNew stream = permissionStreamNewService.loadPermissionStreamNewByBusinessIdAndDepartId(workStep.getId(),item.getId(),DocAttributionTypeEnum.WORKSITE.getCode().toString());
+            if(stream != null) {
+                permissionStreamList.add(stream);
+            }
+        });
+        if(!workStepDepartmentList.isEmpty()) {
+            boolean b = workStepDepartmentService.removeByCollection(workStepDepartmentList);
+            if(!b) {
+                ExceptionCast.cast(ProductInfoCode.PRODUCT_USER_PERM_ERROR);
+            }
+        }
+        if(!permissionStreamList.isEmpty()) {
+            boolean b = permissionStreamNewService.deletePermissionStreamNewByList(permissionStreamList);
+            if(!b) {
+                ExceptionCast.cast(ProductInfoCode.PRODUCT_USER_PERM_ERROR);
+            }
+        }
+        return true;
+    }
+
+    /**
+     * 鍒犻櫎宸ユ淇℃伅 閫昏緫鍒犻櫎
+     * @param id
+     * @return
+     */
+    public boolean deleteWorkStep(String id){
+        if(!ValidateUtil.validateString(id))
+            ExceptionCast.cast(CommonCode.INVALID_PARAM);
+        WorkStep en = super.getById(id);
+        if(en == null)
+            ExceptionCast.cast(ProcessInfoCode.PROCESS_NOT_EXIST);
+        //楠岃瘉鏄惁瀛樺湪鏂囨。
+        List<DocRelative> docRelativeList=iDocRelativeService.list(new QueryWrapper<DocRelative>().eq("attribution_type",DocAttributionTypeEnum.WORKSITE.getCode().toString()).eq("attribution_id",id));
+        if (!docRelativeList.isEmpty()){
+            ExceptionCast.cast(ProcessInfoCode.WORKSTEP_HAS_DOC);
+        }
+        boolean b=productMixService.removeById(id);
+        if(!b) {
+            ExceptionCast.cast(CommonCode.FAIL);
+        }
+        return super.removeById(id);
+    }
+
+    @Override
+    @Transactional(rollbackFor = {Exception.class})
+    public boolean assignAddUser(WorkStep workStep, Collection<SysUser> userList) {
+        if(workStep == null || userList == null || userList.isEmpty())
+            ExceptionCast.cast(CommonCode.INVALID_PARAM);
+        List<WorkStepPermission> permissionList = new ArrayList<>();
+        List<PermissionStreamNew> permissionStreamList = new ArrayList<>();
+        userList.forEach(item -> {
+            WorkStepPermission en = workStepPermissionService.getByStepIdAndUserId(workStep.getId(), item.getId());
+            if(en == null) {
+                en = new WorkStepPermission();
+                en.setUserId(item.getId());
+                en.setStepId(workStep.getId());
+                permissionList.add(en);
+            }
+            PermissionStreamNew stream = permissionStreamNewService.loadPermissionStreamNewByBusinessIdAndUserId(workStep.getId(),item.getId(),DocAttributionTypeEnum.WORKSITE.getCode().toString());
+            if(stream == null) {
+                stream = new PermissionStreamNew();
+                stream.setUserId(item.getId());
+                stream.setBusinessId(workStep.getId());
+                stream.setBusinessType(DocAttributionTypeEnum.WORKSITE.getCode().toString());
+                permissionStreamList.add(stream);
+            }
+        });
+        if(!permissionList.isEmpty()) {
+            boolean b = workStepPermissionService.saveBatch(permissionList);
+            if(!b) {
+                ExceptionCast.cast(ProductInfoCode.PRODUCT_USER_PERM_ERROR);
+            }
+        }
+        if(!permissionStreamList.isEmpty()) {
+            boolean b = permissionStreamNewService.saveBatch(permissionStreamList);
+            if(!b) {
+                ExceptionCast.cast(ProductInfoCode.PRODUCT_USER_PERM_ERROR);
+            }
+        }
+        return true;
+    }
+
+    @Override
+    @Transactional(rollbackFor = {Exception.class})
+    public boolean assignRemoveUser(WorkStep workStep, Collection<SysUser> userList) {
+        if(workStep == null || userList == null || userList.isEmpty())
+            ExceptionCast.cast(CommonCode.INVALID_PARAM);
+        List<WorkStepPermission> permissionList = new ArrayList<>();
+        List<PermissionStreamNew> permissionStreamList = new ArrayList<>();
+        userList.forEach(item -> {
+            WorkStepPermission en = workStepPermissionService.getByStepIdAndUserId(workStep.getId(), item.getId());
+            if(en != null) {
+                permissionList.add(en);
+            }
+            PermissionStreamNew stream = permissionStreamNewService.loadPermissionStreamNewByBusinessIdAndUserId(workStep.getId(),item.getId(),DocAttributionTypeEnum.WORKSITE.getCode().toString());
+            if(stream != null) {
+                permissionStreamList.add(stream);
+            }
+        });
+        //娓呯┖鐢ㄦ埛鏉冮檺鏍¢獙
+        if(!permissionList.isEmpty()) {
+            boolean b = workStepPermissionService.removeByCollection(permissionList);
+            if(!b) {
+                ExceptionCast.cast(ProductInfoCode.PRODUCT_USER_PERM_ERROR);
+            }
+        }
+        if(!permissionStreamList.isEmpty()) {
+            boolean b = permissionStreamNewService.deletePermissionStreamNewByList(permissionStreamList);
+            if(!b) {
+                ExceptionCast.cast(ProductInfoCode.PRODUCT_USER_PERM_ERROR);
+            }
+        }
+        return true;
+    }
+
+    /**
+     * 閫氳繃宸ユ鍙枫�佸伐姝ュ悕绉扮瓑鏌ヨ瀵瑰簲NC鏂囦欢
+     * NC鏂囦欢瀛樺湪璁惧绫讳笅闈�
+     * @param treeInfoRequest
+     * @return
+     */
+    @Override
+    public List<DocInfo> getByWorkStepOtherFile(TreeInfoRequest treeInfoRequest){
+        LambdaQueryWrapper<WorkStep> queryWrapper = new LambdaQueryWrapper<>();
+        if (treeInfoRequest.getProductIds() != null && !treeInfoRequest.getProductIds().isEmpty()) {
+            queryWrapper.in(WorkStep::getProductId, treeInfoRequest.getProductIds());
+        }
+        if (treeInfoRequest.getComponentIds() != null && !treeInfoRequest.getComponentIds().isEmpty()) {
+            queryWrapper.in(WorkStep::getComponentId, treeInfoRequest.getComponentIds());
+        }
+        if (treeInfoRequest.getPartsIds() != null && !treeInfoRequest.getPartsIds().isEmpty()) {
+            queryWrapper.in(WorkStep::getPartsId, treeInfoRequest.getPartsIds());
+        }
+        if (treeInfoRequest.getPsvIds() != null && !treeInfoRequest.getPsvIds().isEmpty()) {
+            queryWrapper.in(WorkStep::getPsvId, treeInfoRequest.getPsvIds());
+        }
+        if (treeInfoRequest.getProcessIds() != null && !treeInfoRequest.getProcessIds().isEmpty()) {
+            queryWrapper.in(WorkStep::getProcessId, treeInfoRequest.getProcessIds());
+        }
+        if (Objects.equals(treeInfoRequest.getAttributionType(), DocAttributionTypeEnum.WORKSITE.getCode())){
+            queryWrapper.eq(StrUtil.isNotEmpty(treeInfoRequest.getAttributionId()),WorkStep::getId,treeInfoRequest.getAttributionId());
+        }
+        queryWrapper.like(StrUtil.isNotEmpty(treeInfoRequest.getTreeName()),WorkStep::getStepName, treeInfoRequest.getTreeName());
+        queryWrapper.like(StrUtil.isNotEmpty(treeInfoRequest.getTreeCode()),WorkStep::getStepCode, treeInfoRequest.getTreeName());
+        queryWrapper.orderByDesc(WorkStep::getCreateTime);
+        List<WorkStep> list = super.list(queryWrapper);
+        List<DocInfo> docInfos;
+        if (list == null || list.isEmpty() || StrUtil.isNotBlank(treeInfoRequest.getStructureType())){
+            return new ArrayList<>();
+        }else {
+            String ids=list.stream().map(WorkStep::getId).collect(Collectors.joining(","));
+            DocInfoQueryRequest docQuery = new DocInfoQueryRequest();
+            BeanUtil.copyProperties(treeInfoRequest,docQuery);
+            docQuery.setAttributionIds(ids);
+            docQuery.setDocClassCode("OTHER");
+            docQuery.setAttributionType(DocAttributionTypeEnum.WORKSITE.getCode());
+            docInfos=docInfoService.findListByDocQuery(docQuery);
+        }
+        return docInfos;
+    }
+
+    /**
+     * 閫氳繃宸ユ鍙枫�佸伐姝ュ悕绉扮瓑鏌ヨ瀵瑰簲NC鏂囦欢
+     * NC鏂囦欢瀛樺湪璁惧绫讳笅闈�
+     * @param treeInfoRequest
+     * @return
+     */
+    @Override
+    public List<DocInfo> getByWorkStepNCFile(TreeInfoRequest treeInfoRequest){
+        LambdaQueryWrapper<WorkStep> queryWrapper = new LambdaQueryWrapper<>();
+        if (treeInfoRequest.getProductIds() != null && !treeInfoRequest.getProductIds().isEmpty()) {
+            queryWrapper.in(WorkStep::getProductId, treeInfoRequest.getProductIds());
+        }
+        if (treeInfoRequest.getComponentIds() != null && !treeInfoRequest.getComponentIds().isEmpty()) {
+            queryWrapper.in(WorkStep::getComponentId, treeInfoRequest.getComponentIds());
+        }
+        if (treeInfoRequest.getPartsIds() != null && !treeInfoRequest.getPartsIds().isEmpty()) {
+            queryWrapper.in(WorkStep::getPartsId, treeInfoRequest.getPartsIds());
+        }
+        if (treeInfoRequest.getPsvIds() != null && !treeInfoRequest.getPsvIds().isEmpty()) {
+            queryWrapper.in(WorkStep::getPsvId, treeInfoRequest.getPsvIds());
+        }
+        if (treeInfoRequest.getProcessIds() != null && !treeInfoRequest.getProcessIds().isEmpty()) {
+            queryWrapper.in(WorkStep::getProcessId, treeInfoRequest.getProcessIds());
+        }
+        if (Objects.equals(treeInfoRequest.getAttributionType(), DocAttributionTypeEnum.WORKSITE.getCode())){
+            queryWrapper.eq(StrUtil.isNotEmpty(treeInfoRequest.getAttributionId()),WorkStep::getProcessId,treeInfoRequest.getAttributionId());
+        }
+        queryWrapper.like(StrUtil.isNotEmpty(treeInfoRequest.getTreeName()),WorkStep::getStepName, treeInfoRequest.getTreeName());
+        queryWrapper.like(StrUtil.isNotEmpty(treeInfoRequest.getTreeCode()),WorkStep::getStepCode, treeInfoRequest.getTreeName());
+        queryWrapper.orderByDesc(WorkStep::getCreateTime);
+        List<WorkStep> list = super.list(queryWrapper);
+        List<DocInfo> docInfos = new ArrayList<>();
+        //NC鏂囦欢瀛樺湪瀵瑰簲鐨勮澶囩被涓嬶紝鎵�浠ラ渶瑕佽繃婊�
+        if (!list.isEmpty()) {
+            List<String> workStepIds = list.stream().map(WorkStep::getId).collect(Collectors.toList());
+            List<DeviceType> deviceTypeList = deviceTypeService.getDeviceTypeByAttribution(workStepIds,
+                    DocAttributionTypeEnum.WORKSITE.getCode(),treeInfoRequest.getDeviceManagementName(),treeInfoRequest.getDeviceManagementCode());
+            if (deviceTypeList != null && !deviceTypeList.isEmpty()) {
+                List<String> deviceTypeIds=deviceTypeList.stream().map(DeviceType::getId).collect(Collectors.toList());
+                DocInfoQueryRequest docQuery = new DocInfoQueryRequest();
+                BeanUtil.copyProperties(treeInfoRequest,docQuery);
+                docQuery.setAttributionIds(String.join(",",deviceTypeIds));
+                docQuery.setDocClassCode("NC");
+                docQuery.setAttributionType(DocAttributionTypeEnum.WORKSITE.getCode());
+                docInfos=docInfoService.findListByDocQuery(docQuery);
+            }
+            return docInfos;
+        }
+        return new ArrayList<>();
+    }
+
+}

--
Gitblit v1.9.3