|
@@ -63,6 +63,7 @@ import java.util.stream.Collectors;
|
|
|
public class ArchiveTreeContractServiceImpl extends BaseServiceImpl<ArchiveTreeContractMapper, ArchiveTreeContract> implements IArchiveTreeContractService {
|
|
|
|
|
|
private final ArchiveTreeContractMapper archiveTreeContractMapper;
|
|
|
+ private final ArchiveTreeMapper archiveTreeMapper;
|
|
|
|
|
|
private final IArchiveTreeService archiveTreeService;
|
|
|
|
|
@@ -80,6 +81,35 @@ public class ArchiveTreeContractServiceImpl extends BaseServiceImpl<ArchiveTreeC
|
|
|
|
|
|
@Override
|
|
|
public boolean initTree2(String tenantId, Long projectId){
|
|
|
+ if (projectId == null || projectId == 0) {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+
|
|
|
+ //判断客户级是否存在
|
|
|
+ QueryWrapper<ArchiveTreeContract> queryWrapper = new QueryWrapper<>();
|
|
|
+ queryWrapper.eq("parent_id", 0);
|
|
|
+ queryWrapper.eq("project_id", projectId);
|
|
|
+ queryWrapper.eq("status", 1);
|
|
|
+ queryWrapper.eq("is_deleted", 0);
|
|
|
+ ArchiveTreeContract archiveTreeContract1 = archiveTreeContractMapper.selectOne(queryWrapper);
|
|
|
+ if (archiveTreeContract1 != null) {
|
|
|
+ throw new ServiceException("根节点已存在,请先删除后再进行初始化");
|
|
|
+ }
|
|
|
+
|
|
|
+ //判断项目级是否存在
|
|
|
+ QueryWrapper<ArchiveTree> queryWrapper1 = new QueryWrapper<>();
|
|
|
+ queryWrapper1.eq("parent_id", 0);
|
|
|
+ queryWrapper1.eq("project_id", projectId);
|
|
|
+ queryWrapper1.eq("status", 1);
|
|
|
+ queryWrapper1.eq("is_deleted", 0);
|
|
|
+ ArchiveTree archiveTree = archiveTreeMapper.selectOne(queryWrapper1);
|
|
|
+ if (archiveTree == null ) {
|
|
|
+ boolean bRet = archiveTreeService.initArchiveProjectTree(projectId);
|
|
|
+ if (bRet == false) {
|
|
|
+ return bRet;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
|
|
|
List<ArchiveTreeVO2> tree = archiveTreeService.tree2(AuthUtil.getTenantId(), projectId,null, null,null,false);
|
|
|
if (tree == null || tree.size() == 0) {
|
|
@@ -107,6 +137,7 @@ public class ArchiveTreeContractServiceImpl extends BaseServiceImpl<ArchiveTreeC
|
|
|
|
|
|
ArchiveTreeContractVO2 newTree = arTreeContractInitService.getTree(archiveTreeContracts);
|
|
|
|
|
|
+
|
|
|
List<ArchiveTreeContract> addNodes = arTreeContractInitService.getContractProcExtNodes(tenantId,projectId,wbsId,newTree);
|
|
|
archiveTreeContracts.addAll(addNodes);
|
|
|
|