From b36dfcb807af748dabdbc76a134b0667196563f6 Mon Sep 17 00:00:00 2001
From: buhuazhen <hua100783@gmail.com>
Date: 星期二, 10 三月 2026 14:37:43 +0800
Subject: [PATCH] feat(projectManagement): 完善项目信息管理功能

---
 src/main/java/com/ruoyi/projectManagement/service/impl/PlanServiceImpl.java |   98 +++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 83 insertions(+), 15 deletions(-)

diff --git a/src/main/java/com/ruoyi/projectManagement/service/impl/PlanServiceImpl.java b/src/main/java/com/ruoyi/projectManagement/service/impl/PlanServiceImpl.java
index 4262f89..fd5f028 100644
--- a/src/main/java/com/ruoyi/projectManagement/service/impl/PlanServiceImpl.java
+++ b/src/main/java/com/ruoyi/projectManagement/service/impl/PlanServiceImpl.java
@@ -1,21 +1,29 @@
 package com.ruoyi.projectManagement.service.impl;
 
 import cn.hutool.core.bean.BeanUtil;
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.lang.Assert;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.ruoyi.basic.service.CustomerFollowUpFileService;
+import com.ruoyi.common.exception.ServiceException;
 import com.ruoyi.projectManagement.mapper.PlanMapper;
+import com.ruoyi.projectManagement.mapper.PlanNodeMapper;
 import com.ruoyi.projectManagement.pojo.Plan;
+import com.ruoyi.projectManagement.pojo.PlanNode;
 import com.ruoyi.projectManagement.service.PlanService;
-import com.ruoyi.projectManagement.vo.PlanVo;
-import com.ruoyi.projectManagement.vo.SavePlanVo;
-import com.ruoyi.projectManagement.vo.SearchPlanVo;
+import com.ruoyi.projectManagement.vo.*;
 import lombok.RequiredArgsConstructor;
+import org.jetbrains.annotations.Nullable;
+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.Collections;
-import java.util.Optional;
+import java.util.*;
+import java.util.stream.Collectors;
+import java.util.stream.IntStream;
 
 /**
  * @author buhuazhen
@@ -31,6 +39,12 @@
 
     private final CustomerFollowUpFileService customerFollowUpFileService;
 
+    private final PlanNodeMapper planNodeMapper;
+
+    @Lazy
+    @Autowired
+    private PlanService planService;
+
     @Override
     @Transactional(rollbackFor = Exception.class)
     public void savePlan(SavePlanVo savePlanVo) {
@@ -38,18 +52,60 @@
         // 闄勪欢澶勭悊 , 鎷兼帴
         String attachments = String.join(",", Optional.ofNullable(savePlanVo.getAttachmentIds()).orElse(Collections.emptyList()));
         plan.setAttachment(attachments);
-
-
         if (savePlanVo.getId() == null) {
             planMapper.insert(plan);
         } else {
             planMapper.updateById(plan);
         }
-
-        // todo@ 鑺傜偣淇濆瓨
-
-
+        planService.savePlanNode(plan.getId(), savePlanVo.getSavePlanNodeList());
     }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public void savePlanNode(Long planId, @Nullable List<SavePlanNodeVo> savePlanNodeVos) {
+        Assert.notNull(planId, () -> new ServiceException("璁″垝ID涓嶈兘涓虹┖"));
+        if (CollUtil.isEmpty(savePlanNodeVos)) {
+            return;
+        }
+
+        // 鍒犻櫎澶氫綑鑺傜偣
+        List<Long> existNodeIds = savePlanNodeVos.stream().map(SavePlanNodeVo::getId).filter(Objects::nonNull).collect(Collectors.toList());
+        LambdaQueryWrapper<PlanNode> planNodeLambdaQueryWrapper = new LambdaQueryWrapper<PlanNode>()
+                .select(PlanNode::getId)
+                .eq(PlanNode::getProjectManagementPlanId, planId);
+        if(CollUtil.isNotEmpty(existNodeIds)){
+            planNodeLambdaQueryWrapper.notIn(PlanNode::getId, existNodeIds);
+        }
+        List<PlanNode> needDeleteNode = planNodeMapper.selectList(planNodeLambdaQueryWrapper);
+
+        deletePlanNode(needDeleteNode.stream().map(PlanNode::getId).collect(Collectors.toList()));
+
+        List<PlanNode> planNodes = BeanUtil.copyToList(savePlanNodeVos, PlanNode.class);
+        // 璁剧疆鎺掑簭绱㈠紩
+        IntStream.range(0, savePlanNodeVos.size()).forEach(i -> {
+            planNodes.get(i).setSort(i);
+            planNodes.get(i).setProjectManagementPlanId(planId);
+            if (planNodes.get(i).getId() == null) {
+                planNodeMapper.insert(planNodes.get(i));
+            } else {
+                planNodeMapper.updateById(planNodes.get(i));
+            }
+        });
+    }
+
+    @Override
+    public List<PlanNode> getPlanNodeByPlanId(Long planId) {
+        return planNodeMapper.selectList(new LambdaQueryWrapper<PlanNode>()
+                .eq(PlanNode::getIsDelete, 0)
+                .eq(PlanNode::getProjectManagementPlanId, planId).orderByAsc(PlanNode::getSort));
+    }
+
+    private List<PlanNode> getPlanNodeByPlanIds(List<Long> planIds) {
+        return planNodeMapper.selectList(new LambdaQueryWrapper<PlanNode>()
+                .eq(PlanNode::getIsDelete, 0)
+                .in(PlanNode::getProjectManagementPlanId, planIds).orderByAsc(PlanNode::getSort));
+    }
+
 
     @Override
     @Transactional(rollbackFor = Exception.class)
@@ -58,9 +114,18 @@
                 new LambdaUpdateWrapper<Plan>()
                         .eq(Plan::getId, id)
                         .set(Plan::getIsDelete, 1));
-        // todo@ 瀵瑰簲鑺傜偣鍏ㄩ儴鍒犻櫎
+        planService.deletePlanNode(getPlanNodeByPlanId(id).stream().map(PlanNode::getId).collect(Collectors.toList()));
+    }
 
-
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public void deletePlanNode(List<Long> ids) {
+        if (CollUtil.isNotEmpty(ids)) {
+            planNodeMapper.update(null,
+                    new LambdaUpdateWrapper<PlanNode>()
+                            .in(PlanNode::getId, ids)
+                            .set(PlanNode::getIsDelete, 1));
+        }
     }
 
     @Override
@@ -69,8 +134,11 @@
         IPage<PlanVo> resultPage = planIPage.convert(plan -> BeanUtil.copyProperties(plan, PlanVo.class));
         // 鏂囦欢鑾峰彇
         customerFollowUpFileService.fillAttachment(resultPage.getRecords(), PlanVo::getAttachment, PlanVo::setAttachmentList);
-        // todo@ node 鑺傜偣鑾峰彇
-
+        Map<Long, List<PlanNodeVo>> collect = getPlanNodeByPlanIds(resultPage.getRecords().stream().map(PlanVo::getId).collect(Collectors.toList()))
+                .stream()
+                .map(it -> BeanUtil.copyProperties(it, PlanNodeVo.class))
+                .collect(Collectors.groupingBy(PlanNodeVo::getProjectManagementPlanId, Collectors.toList()));
+        resultPage.getRecords().forEach(planVo -> planVo.setPlanNodeList(collect.getOrDefault(planVo.getId(), Collections.emptyList())));
         return resultPage;
     }
 

--
Gitblit v1.9.3