lxzn-module-dnc/src/main/java/org/jeecg/modules/dnc/service/impl/ComponentInfoSeServiceImpl.java
@@ -4,9 +4,12 @@
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.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.apache.shiro.SecurityUtils;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.system.vo.LoginUser;
import org.jeecg.modules.dnc.dto.ComponentExt;
import org.jeecg.modules.dnc.entity.*;
@@ -57,6 +60,8 @@
    @Autowired
    @Lazy
    private IDocInfoService docInfoService;
    @Autowired
    private IDocRelativeService docRelativeService;
    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean addComponentInfo(ComponentInfo componentInfo) {
@@ -69,7 +74,7 @@
        if(!ValidateUtil.validateString(componentInfo.getComponentCode()))
            ExceptionCast.cast(ComponentInfoCode.COMPONENT_CODE_NONE);
        ComponentInfo en = getByCode(componentInfo.getComponentCode());
        if(en != null)
        if(en != null&&componentInfo.getBorrowId()==null)
            ExceptionCast.cast(ComponentInfoCode.COMPONENT_IS_EXIST);
        LoginUser user = (LoginUser) SecurityUtils.getSubject().getPrincipal();
        String userId = user.getId();
@@ -137,6 +142,20 @@
                    ,componentInfo.getComponentName(),componentInfo.getComponentCode(),2,new Date());
        }
        productMixService.save(productMix);
        if (StrUtil.isNotEmpty(componentInfo.getBorrowId())){
            //借用部件,同步添加文件
            List<DocRelative> docRelativeList=docRelativeService.
                    list(new QueryWrapper<DocRelative>().eq("attribution_type","2").eq("attribution_id",componentInfo.getBorrowId()));
            if (!docRelativeList.isEmpty()){
                docRelativeList.forEach(item -> {
                    DocRelative docRelative = new DocRelative();
                    BeanUtil.copyProperties(item,docRelative);
                    docRelative.setId(null);
                    docRelative.setAttributionId(componentInfo.getComponentId());
                    docRelativeService.save(docRelative);
                });
            }
        }
        //添加权限
        PermissionStreamNew stream = new PermissionStreamNew();
        stream.setUserId(userId);
@@ -189,7 +208,11 @@
            stream.setBusinessType("2");
            return permissionStreamNewService.save(stream);
        }
        return b;
        List<ComponentInfo> list = super.lambdaQuery().eq(ComponentInfo::getBorrowId,id).list();
        if (list != null && !list.isEmpty()) {
            list.forEach(item -> editComponentInfo(item.getComponentId(),componentInfo));
        }
        return true;
    }
    @Override
@@ -267,14 +290,20 @@
        if(childrenList != null && !childrenList.isEmpty()) {
            ExceptionCast.cast(ComponentInfoCode.COMPONENT_CHILD_EXIST);
        }
        //验证产品下是否有零件
        //验证部件下是否有零件
        List<PartsInfo> partsInfoList = partsInfoService.getByComponentId(componentInfo.getProductId(), componentInfo.getComponentId());
        if(partsInfoList != null && !partsInfoList.isEmpty()) {
            ExceptionCast.cast(ComponentInfoCode.COMPONENT_PARTS_EXIST);
        }
        //验证部件下是否有工序信息
        List<ProcessStream> processStreamList = processStreamService.findByComponentId(id);
        if(processStreamList != null && !processStreamList.isEmpty()) {
            ExceptionCast.cast(ComponentInfoCode.COMPONENT_PROCESS_EXIST);
        }
        //验证部件是否被借用
        List<ComponentInfo> list = super.lambdaQuery().eq(ComponentInfo::getBorrowId,id).list();
        if (list != null && !list.isEmpty()) {
            ExceptionCast.cast(ComponentInfoCode.COMPONENT_BORROW_EXIST);
        }
        boolean b;
        //验证是否存在文档
@@ -566,4 +595,91 @@
        }
        return docInfos;
    }
    /**
     * 查询可以被引用的部件
     * @param componentInfo
     * @param pageNo
     * @param pageSize
     * @return
     */
    @Override
    public IPage<ComponentInfo> getByComponentId(ComponentInfo componentInfo, Integer pageNo, Integer pageSize){
        //权限排除,不能引用自己上下级部件
        LoginUser loginUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
        String productIds = loginUser.getProductionIds();
        IPage<ComponentInfo> page = new Page<>(pageNo, pageSize);
        if (productIds != null && !productIds.isEmpty()) {
            List<String> productIdList = Arrays.asList(productIds.split(","));
            List<ProductMix> productMixList = permissionStreamNewService.
                    loadProductMixByBusinessIdAndBusinessType(loginUser.getId(),productIdList,componentInfo.getComponentId(),"2");
            List<Long> ids = productMixList.stream().map(ProductMix::getId).collect(Collectors.toList());
            //获取已经被借用的部件
            LambdaQueryWrapper<ComponentInfo> borrowQueryWrapper = new LambdaQueryWrapper<>();
            borrowQueryWrapper.in(ComponentInfo::getBorrowId,ids);
            borrowQueryWrapper.eq(ComponentInfo::getParentId,componentInfo.getComponentId());
            List<String> borrowIds= super.list(borrowQueryWrapper).stream().map(ComponentInfo::getBorrowId).collect(Collectors.toList());
            LambdaQueryWrapper<ComponentInfo> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.like(StrUtil.isNotEmpty(componentInfo.getComponentName()),ComponentInfo::getComponentName, componentInfo.getComponentName());
            queryWrapper.like(StrUtil.isNotEmpty(componentInfo.getComponentCode()),ComponentInfo::getComponentCode, componentInfo.getComponentCode());
            queryWrapper.isNull(ComponentInfo::getBorrowId);
            queryWrapper.in(ComponentInfo::getComponentId,ids);
            if (!borrowIds.isEmpty()) {
                queryWrapper.notIn(ComponentInfo::getComponentId,borrowIds);
            }
            queryWrapper.orderByDesc(ComponentInfo::getCreateTime);
            page = super.page(page, queryWrapper);
        }
        return page;
    }
    /**
     * 借用部件(可批量) - 原有部件id,新部件ids
     * @param oldId,newIds
     * @return
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public Result<?> borrowComponent(String oldId, String newIds){
        LoginUser loginUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
        ComponentInfo componentInfo = super.getById(oldId);
        if(componentInfo == null) {
            return Result.error("部件不存在");
        }
        List<String> productIdList = Arrays.asList(loginUser.getProductionIds().split(","));
        //获取可以借用的部件
        List<ProductMix> productMixList = permissionStreamNewService.
                loadProductMixByBusinessIdAndBusinessType(loginUser.getId(),productIdList,componentInfo.getComponentId(),"2");
        Set<Long> allowedIds = productMixList.stream()
                .map(ProductMix::getId)
                .collect(Collectors.toCollection(HashSet::new));
        List<String> newIdList = Arrays.asList(newIds.split(","));
        boolean allAllowed = newIdList.parallelStream()
                .map(Long::parseLong)
                .allMatch(allowedIds::contains);
        if (!allAllowed) {
            return Result.error("存在未授权的部件借用");
        }
        //获取已经被借用的部件
        LambdaQueryWrapper<ComponentInfo> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.in(ComponentInfo::getBorrowId,newIdList);
        queryWrapper.eq(ComponentInfo::getParentId,componentInfo.getComponentId());
        List<ComponentInfo> list = super.list(queryWrapper);
        //判断是否已经被当前部件借用
        if(!list.isEmpty()) {
            return Result.error("选择的部件已被当前部件借用");
        }
        List<ComponentInfo> componentInfoList=listByIds(newIdList);
        componentInfoList.forEach(item->{
            //添加部件信息、层级信息、权限信息
            ComponentInfo newComponentInfo=new ComponentInfo();
            BeanUtil.copyProperties(item,newComponentInfo);
            newComponentInfo.setComponentId(null);
            newComponentInfo.setProductId(componentInfo.getProductId());
            newComponentInfo.setParentId(componentInfo.getComponentId());
            newComponentInfo.setBorrowId(item.getComponentId());
            addComponentInfo(newComponentInfo);
        });
        return Result.ok("部件借用成功");
    }
}