From 5c24b36b6f590e665e023c1ce6aa9d0cf78ad68d Mon Sep 17 00:00:00 2001
From: wuxw <928255095@qq.com>
Date: 星期二, 08 十一月 2022 01:16:08 +0800
Subject: [PATCH] 优化保养计划添加功能

---
 service-community/src/main/java/com/java110/community/cmd/maintainancePlan/SaveMaintainancePlanCmd.java |   16 ++++++++++++++++
 1 files changed, 16 insertions(+), 0 deletions(-)

diff --git a/service-community/src/main/java/com/java110/community/cmd/maintainancePlan/SaveMaintainancePlanCmd.java b/service-community/src/main/java/com/java110/community/cmd/maintainancePlan/SaveMaintainancePlanCmd.java
index 6fe3a30..acf0317 100644
--- a/service-community/src/main/java/com/java110/community/cmd/maintainancePlan/SaveMaintainancePlanCmd.java
+++ b/service-community/src/main/java/com/java110/community/cmd/maintainancePlan/SaveMaintainancePlanCmd.java
@@ -24,10 +24,12 @@
 import com.java110.core.event.cmd.CmdEvent;
 import com.java110.core.factory.GenerateCodeFactory;
 import com.java110.dto.equipmentAccount.EquipmentAccountDto;
+import com.java110.dto.user.UserDto;
 import com.java110.intf.common.IEquipmentAccountV1InnerServiceSMO;
 import com.java110.intf.community.IMaintainancePlanMachineV1InnerServiceSMO;
 import com.java110.intf.community.IMaintainancePlanStaffV1InnerServiceSMO;
 import com.java110.intf.community.IMaintainancePlanV1InnerServiceSMO;
+import com.java110.intf.user.IUserV1InnerServiceSMO;
 import com.java110.po.maintainancePlan.MaintainancePlanPo;
 import com.java110.po.maintainancePlanMachine.MaintainancePlanMachinePo;
 import com.java110.po.maintainancePlanStaff.MaintainancePlanStaffPo;
@@ -70,6 +72,9 @@
     @Autowired
     private IEquipmentAccountV1InnerServiceSMO equipmentAccountV1InnerServiceSMOImpl;
 
+    @Autowired
+    private IUserV1InnerServiceSMO userV1InnerServiceSMOImpl;
+
     @Override
     public void validate(CmdEvent event, ICmdDataFlowContext cmdDataFlowContext, JSONObject reqJson) {
         Assert.hasKeyAndValue(reqJson, "planName", "璇锋眰鎶ユ枃涓湭鍖呭惈planName");
@@ -106,7 +111,18 @@
     @Java110Transactional
     public void doCmd(CmdEvent event, ICmdDataFlowContext cmdDataFlowContext, JSONObject reqJson) throws CmdException {
 
+        String userId = cmdDataFlowContext.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,"鍛樺伐涓嶅瓨鍦�");
+
         MaintainancePlanPo maintainancePlanPo = BeanConvertUtil.covertBean(reqJson, MaintainancePlanPo.class);
+        maintainancePlanPo.setCreateUserId(userId);
+        maintainancePlanPo.setCreateUserName(userDtos.get(0).getName());
         maintainancePlanPo.setPlanId(GenerateCodeFactory.getGeneratorId(CODE_PREFIX_ID));
         int flag = maintainancePlanV1InnerServiceSMOImpl.saveMaintainancePlan(maintainancePlanPo);
 

--
Gitblit v1.8.0