From d99758709271e4d183dbedd2374f7ba96538a22b Mon Sep 17 00:00:00 2001
From: wuxw <928255095@qq.com>
Date: 星期六, 13 八月 2022 02:12:00 +0800
Subject: [PATCH] 优化代码

---
 service-community/src/main/java/com/java110/community/cmd/inspectionPlan/SaveInspectionPlanCmd.java |   19 +++++++++++++++++++
 1 files changed, 19 insertions(+), 0 deletions(-)

diff --git a/service-community/src/main/java/com/java110/community/cmd/inspectionPlan/SaveInspectionPlanCmd.java b/service-community/src/main/java/com/java110/community/cmd/inspectionPlan/SaveInspectionPlanCmd.java
index 728c121..39c53a6 100644
--- a/service-community/src/main/java/com/java110/community/cmd/inspectionPlan/SaveInspectionPlanCmd.java
+++ b/service-community/src/main/java/com/java110/community/cmd/inspectionPlan/SaveInspectionPlanCmd.java
@@ -6,18 +6,25 @@
 import com.java110.core.event.cmd.Cmd;
 import com.java110.core.event.cmd.CmdEvent;
 import com.java110.core.factory.GenerateCodeFactory;
+import com.java110.dto.user.UserDto;
 import com.java110.intf.community.IInspectionPlanV1InnerServiceSMO;
+import com.java110.intf.user.IUserV1InnerServiceSMO;
 import com.java110.po.inspection.InspectionPlanPo;
 import com.java110.utils.exception.CmdException;
 import com.java110.utils.util.Assert;
 import com.java110.utils.util.BeanConvertUtil;
 import org.springframework.beans.factory.annotation.Autowired;
 
+import java.util.List;
+
 @Java110Cmd(serviceCode = "inspectionPlan.saveInspectionPlan")
 public class SaveInspectionPlanCmd extends Cmd {
 
     @Autowired
     private IInspectionPlanV1InnerServiceSMO inspectionPlanV1InnerServiceSMOImpl;
+
+    @Autowired
+    private IUserV1InnerServiceSMO userV1InnerServiceSMOImpl;
 
     @Override
     public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException {
@@ -34,10 +41,22 @@
 
     @Override
     public void doCmd(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException {
+
+        String userId = context.getReqHeaders().get("user-id");
+        UserDto userDto = new UserDto();
+        userDto.setUserId(userId);
+        userDto.setRow(1);
+        userDto.setPage(1);
+       List<UserDto> userDtos =  userV1InnerServiceSMOImpl.queryUsers(userDto);
+
+       Assert.listOnlyOne(userDtos,"鍛樺伐涓嶅瓨鍦�");
+
         JSONObject businessInspectionPlan = new JSONObject();
         businessInspectionPlan.putAll(reqJson);
         businessInspectionPlan.put("inspectionPlanId", GenerateCodeFactory.getGeneratorId(GenerateCodeFactory.CODE_PREFIX_inspectionPlanId));
         InspectionPlanPo inspectionPlanPo = BeanConvertUtil.covertBean(businessInspectionPlan, InspectionPlanPo.class);
+        inspectionPlanPo.setCreateUserId(userId);
+        inspectionPlanPo.setCreateUserName(userDtos.get(0).getUserName());
         int flag = inspectionPlanV1InnerServiceSMOImpl.saveInspectionPlan(inspectionPlanPo);
         if (flag < 1) {
             throw new CmdException("淇濆瓨宸℃璁″垝澶辫触");

--
Gitblit v1.8.0