From 5405910566dad07e41e7bb1919afaf28385dbd1f Mon Sep 17 00:00:00 2001
From: xueqingkun <xueqingkun@126.com>
Date: Tue, 9 Jan 2024 13:08:36 +0800
Subject: [PATCH] =?UTF-8?q?manage:=E4=BF=AE=E6=94=B9=E7=96=BE=E7=97=85?=
 =?UTF-8?q?=E5=A4=84=E7=BD=AE=E8=AE=A1=E5=88=92=20=E4=BF=9D=E5=AD=98?=
 =?UTF-8?q?=E6=8E=A5=E5=8F=A3(=E6=B7=BB=E5=8A=A0=20=E5=85=B3=E8=81=94?=
 =?UTF-8?q?=E8=8D=AF=E7=89=A9id)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../DiseaseTreatmentPlanManageServiceImpl.java   | 16 ++++++++++++++++
 .../supervision/model/DiseaseTreatmentPlan.java  |  6 ++++++
 .../controller/TreatmentPlanController.java      |  2 +-
 3 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/virtual-patient-manage/src/main/java/com/supervision/manage/service/impl/DiseaseTreatmentPlanManageServiceImpl.java b/virtual-patient-manage/src/main/java/com/supervision/manage/service/impl/DiseaseTreatmentPlanManageServiceImpl.java
index 2eafdf18..fb03a688 100644
--- a/virtual-patient-manage/src/main/java/com/supervision/manage/service/impl/DiseaseTreatmentPlanManageServiceImpl.java
+++ b/virtual-patient-manage/src/main/java/com/supervision/manage/service/impl/DiseaseTreatmentPlanManageServiceImpl.java
@@ -7,6 +7,8 @@ import cn.hutool.core.util.StrUtil;
 import com.supervision.manage.service.ConfigTreatmentPlanManageService;
 import com.supervision.manage.service.DiseaseTreatmentPlanManageService;
 import com.supervision.model.DiseaseTreatmentPlan;
+import com.supervision.model.DiseaseTreatmentPlanDrug;
+import com.supervision.service.DiseaseTreatmentPlanDrugService;
 import com.supervision.service.DiseaseTreatmentPlanService;
 import com.supervision.vo.manage.DiseaseTreatmentPlanResVo;
 import com.supervision.vo.manage.DiseaseTreatmentPlanTreeNode;
@@ -25,6 +27,8 @@ public class DiseaseTreatmentPlanManageServiceImpl implements DiseaseTreatmentPl
 
     private final ConfigTreatmentPlanManageService configTreatmentPlanManageService;
 
+    private final DiseaseTreatmentPlanDrugService diseaseTreatmentPlanDrugService;;
+
     @Override
     public List<DiseaseTreatmentPlanResVo> queryListByDiseaseId(String diseaseId) {
 
@@ -39,6 +43,18 @@ public class DiseaseTreatmentPlanManageServiceImpl implements DiseaseTreatmentPl
 
         diseaseTreatmentPlanService.save(diseaseTreatmentPlan);
 
+        // 保存处置计划药品信息
+        if (CollUtil.isNotEmpty(diseaseTreatmentPlan.getDrugIds())){
+            List<DiseaseTreatmentPlanDrug> diseaseTreatmentPlanDrugList = diseaseTreatmentPlan.getDrugIds().stream().map(item -> {
+                DiseaseTreatmentPlanDrug diseaseTreatmentPlanDrug = new DiseaseTreatmentPlanDrug();
+                diseaseTreatmentPlanDrug.setDiseaseId(diseaseTreatmentPlan.getDiseaseId());
+                diseaseTreatmentPlanDrug.setTreatmentPlanId(diseaseTreatmentPlan.getTreatmentPlanId());
+                diseaseTreatmentPlanDrug.setDrugId(item);
+                return diseaseTreatmentPlanDrug;
+            }).collect(Collectors.toList());
+            diseaseTreatmentPlanDrugService.saveBatch(diseaseTreatmentPlanDrugList);
+        }
+
         return diseaseTreatmentPlan;
 
     }
diff --git a/virtual-patient-model/src/main/java/com/supervision/model/DiseaseTreatmentPlan.java b/virtual-patient-model/src/main/java/com/supervision/model/DiseaseTreatmentPlan.java
index 2de95634..b0a12f75 100644
--- a/virtual-patient-model/src/main/java/com/supervision/model/DiseaseTreatmentPlan.java
+++ b/virtual-patient-model/src/main/java/com/supervision/model/DiseaseTreatmentPlan.java
@@ -5,6 +5,9 @@ import com.baomidou.mybatisplus.annotation.TableId;
 import com.baomidou.mybatisplus.annotation.TableName;
 import java.io.Serializable;
 import java.util.Date;
+import java.util.List;
+
+import io.swagger.annotations.ApiModelProperty;
 import lombok.Data;
 
 /**
@@ -36,6 +39,9 @@ public class DiseaseTreatmentPlan implements Serializable {
      */
     private String treatmentPlanId;
 
+
+    @ApiModelProperty("药物列表")
+    private List<String> drugIds;
     /**
      * 创建人ID
      */
diff --git a/virtual-patient-web/src/main/java/com/supervision/controller/TreatmentPlanController.java b/virtual-patient-web/src/main/java/com/supervision/controller/TreatmentPlanController.java
index 9174cd02..bcfb046c 100644
--- a/virtual-patient-web/src/main/java/com/supervision/controller/TreatmentPlanController.java
+++ b/virtual-patient-web/src/main/java/com/supervision/controller/TreatmentPlanController.java
@@ -19,7 +19,7 @@ import java.util.Map;
 @Slf4j
 @Api(tags = "处置计划")
 @RestController
-@RequestMapping("treatmentPlan")
+@RequestMapping("/treatmentPlan")
 @RequiredArgsConstructor
 public class TreatmentPlanController {