From 5dbdab581d8654f37acb964d6e31e16eeb0be347 Mon Sep 17 00:00:00 2001
From: whycxzp <perryhsu@163.com>
Date: 星期四, 18 五月 2023 10:17:09 +0800
Subject: [PATCH] 更新放电计划接口

---
 src/main/java/com/whyc/service/WorkflowMainService.java |   18 +++++++++++++++++-
 1 files changed, 17 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/whyc/service/WorkflowMainService.java b/src/main/java/com/whyc/service/WorkflowMainService.java
index cb6abb2..5d91988 100644
--- a/src/main/java/com/whyc/service/WorkflowMainService.java
+++ b/src/main/java/com/whyc/service/WorkflowMainService.java
@@ -6,6 +6,7 @@
 import com.github.pagehelper.PageInfo;
 import com.whyc.dto.Response;
 import com.whyc.mapper.WorkflowMainMapper;
+import com.whyc.pojo.BattDischargePlanTemp;
 import com.whyc.pojo.UserInf;
 import com.whyc.pojo.WorkflowLink;
 import com.whyc.pojo.WorkflowMain;
@@ -27,6 +28,9 @@
 
     @Autowired
     private WorkflowLinkService linkService;
+
+    @Autowired
+    private BattDischargePlanTempService tempService;
 
     /**
      * 鍒嗘淳鍗曞彿:
@@ -77,7 +81,19 @@
 
 
     public WorkflowMain getBaseInfo(Integer mainId) {
-        return mapper.getBaseInfo(mainId);
+        //鏍规嵁mainId鏌ヨ鏄摢绉嶇被鍨�
+        QueryWrapper<WorkflowMain> query = Wrappers.query();
+        query.eq("id",mainId).last(" limit 1");
+        WorkflowMain workflowMain = mapper.selectOne(query);
+        Integer type = workflowMain.getType();
+        if(type ==1){
+            List<BattDischargePlanTemp> tempList = tempService.getListByMainId(mainId);
+            workflowMain.setTempList(tempList);
+        }
+        List<WorkflowLink> linkList = linkService.getWorkflowInfo(mainId);
+        workflowMain.setLinkList(linkList);
+
+        return workflowMain;
     }
 
     public List<WorkflowLink> getAssignReply(Integer mainId) {

--
Gitblit v1.9.1