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 { //Assert.hasKeyAndValue(reqJson, "xxx", "xxx"); Assert.hasKeyAndValue(reqJson, "inspectionPlanName", "必填,请填写巡检计划名称"); Assert.hasKeyAndValue(reqJson, "inspectionRouteId", "必填,请填写巡检路线"); Assert.hasKeyAndValue(reqJson, "inspectionPlanPeriod", "必填,请选择执行周期"); Assert.hasKeyAndValue(reqJson, "startTime", "必填,请选择计划开始时间"); Assert.hasKeyAndValue(reqJson, "endTime", "必填,请选择结束时间"); Assert.hasKeyAndValue(reqJson, "signType", "必填,请填写签到方式"); Assert.hasKeyAndValue(reqJson, "state", "必填,请填写签到方式"); } @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 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); } } }