From d526cf637048d71f046667f26e76436e9d489861 Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: 星期日, 23 四月 2023 22:45:11 +0800
Subject: [PATCH] 优化代码
---
service-community/src/main/java/com/java110/community/cmd/inspectionPlan/SaveInspectionPlanCmd.java | 39 +++++++++++++++++++++++++++++++++++++++
1 files changed, 39 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..0102530 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
@@ -1,23 +1,36 @@
package com.java110.community.cmd.inspectionPlan;
+import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.java110.core.annotation.Java110Cmd;
import com.java110.core.context.ICmdDataFlowContext;
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.IInspectionPlanStaffV1InnerServiceSMO;
import com.java110.intf.community.IInspectionPlanV1InnerServiceSMO;
+import com.java110.intf.user.IUserV1InnerServiceSMO;
import com.java110.po.inspection.InspectionPlanPo;
+import com.java110.po.inspection.InspectionPlanStaffPo;
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;
+
+ @Autowired
+ private IInspectionPlanStaffV1InnerServiceSMO inspectionPlanStaffV1InnerServiceSMOImpl;
@Override
public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException {
@@ -34,13 +47,39 @@
@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).getName());
int flag = inspectionPlanV1InnerServiceSMOImpl.saveInspectionPlan(inspectionPlanPo);
if (flag < 1) {
throw new CmdException("淇濆瓨宸℃璁″垝澶辫触");
}
+
+ JSONArray staffs = reqJson.getJSONArray("staffs");
+ InspectionPlanStaffPo inspectionPlanStaffPo = null;
+ for(int staffIndex = 0; staffIndex < staffs.size() ; staffIndex++) {
+ inspectionPlanStaffPo = new InspectionPlanStaffPo();
+ inspectionPlanStaffPo.setCommunityId(reqJson.getString("communityId"));
+ inspectionPlanStaffPo.setEndTime(reqJson.getString("endTime"));
+ inspectionPlanStaffPo.setInspectionPlanId(inspectionPlanPo.getInspectionPlanId());
+ inspectionPlanStaffPo.setIpStaffId(GenerateCodeFactory.getGeneratorId("11"));
+ inspectionPlanStaffPo.setStaffId(staffs.getJSONObject(staffIndex).getString("userId"));
+ inspectionPlanStaffPo.setStaffName(staffs.getJSONObject(staffIndex).getString("name"));
+ inspectionPlanStaffPo.setStartTime(reqJson.getString("startTime"));
+ inspectionPlanStaffV1InnerServiceSMOImpl.saveInspectionPlanStaff(inspectionPlanStaffPo);
+ }
}
}
--
Gitblit v1.8.0