wuye / MicroCommunit

  package com.java110.community.cmd.inspection;
  
  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.dto.inspection.InspectionTaskDetailDto;
  import com.java110.dto.inspection.InspectionTaskDto;

  import com.java110.dto.user.UserDto;

  import com.java110.intf.community.IInspectionTaskDetailV1InnerServiceSMO;
  import com.java110.intf.community.IInspectionTaskV1InnerServiceSMO;

  import com.java110.intf.user.IUserInnerServiceSMO;

  import com.java110.utils.exception.CmdException;
  import com.java110.utils.util.Assert;
  import com.java110.utils.util.BeanConvertUtil;
  import com.java110.vo.ResultVo;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.http.HttpStatus;
  import org.springframework.http.ResponseEntity;
  
  import java.util.ArrayList;
  import java.util.List;
  

  @Java110Cmd(serviceCode = "inspection.queryStaffTodayTasks")
  public class QueryStaffTodayTasksCmd extends Cmd {
  
      @Autowired
      private IInspectionTaskV1InnerServiceSMO inspectionTaskV1InnerServiceSMOImpl;
  
      @Autowired
      private IInspectionTaskDetailV1InnerServiceSMO inspectionTaskDetailV1InnerServiceSMOImpl;
  

      @Autowired
      private IUserInnerServiceSMO userInnerServiceSMOImpl;
  

      @Override
      public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException {

          Assert.hasKeyAndValue(reqJson, "staffCode", "请求报文中未包含staffCode");

      }
  
      @Override
      public void doCmd(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException {

          String staffCode = reqJson.getString("staffCode");
          String communityId = reqJson.getString("communityId") != null ? reqJson.getString("communityId") : "";
  
          // 根据员工编号查询用户
          UserDto userDto = new UserDto();
          userDto.setName(staffCode);
          List<UserDto> users = userInnerServiceSMOImpl.getUsers(userDto);
          String staffId = "-1";
          String staffName = staffCode;
          if (users != null && users.size() > 0) {
              staffId = users.get(0).getUserId();
              staffName = users.get(0).getName();
          }

          InspectionTaskDto inspectionTaskDto = new InspectionTaskDto();
          inspectionTaskDto.setPlanUserId(staffId);
          inspectionTaskDto.setCommunityId(communityId);
          List<InspectionTaskDto> tasks = inspectionTaskV1InnerServiceSMOImpl.queryInspectionTasks(inspectionTaskDto);
  
          JSONObject result = new JSONObject();
          result.put("staffId", staffId);

          result.put("staffName", staffName);
          result.put("staffCode", staffCode);
          result.put("communityId", communityId);

  
          List<JSONObject> taskList = new ArrayList<>();

          for (InspectionTaskDto task : tasks) {

              InspectionTaskDetailDto detailDto = new InspectionTaskDetailDto();
              detailDto.setTaskId(task.getTaskId());
              detailDto.setCommunityId(communityId);

              List<InspectionTaskDetailDto> details = BeanConvertUtil.covertBeanList(
                      inspectionTaskDetailV1InnerServiceSMOImpl.queryInspectionTaskDetails(detailDto),
                      InspectionTaskDetailDto.class);

  
              JSONObject taskObj = new JSONObject();
              taskObj.put("taskId", task.getTaskId());

              taskObj.put("communityId", communityId);

              taskObj.put("planName", task.getInspectionPlanName());
              taskObj.put("planInsTime", task.getPlanInsTime());
              taskObj.put("planEndTime", task.getPlanEndTime());
              taskObj.put("planUserId", task.getPlanUserId());
              taskObj.put("planUserName", task.getPlanUserName());
              taskObj.put("signType", task.getSignType());
              taskObj.put("state", task.getState());
  
              List<JSONObject> detailList = new ArrayList<>();
              for (InspectionTaskDetailDto detail : details) {
                  JSONObject detailObj = new JSONObject();
                  detailObj.put("taskDetailId", detail.getTaskDetailId());
                  detailObj.put("taskId", detail.getTaskId());
                  detailObj.put("inspectionId", detail.getInspectionId());
                  detailObj.put("inspectionName", detail.getInspectionName());
                  detailObj.put("longitude", detail.getLongitude());
                  detailObj.put("latitude", detail.getLatitude());
                  detailObj.put("state", detail.getState());
                  detailObj.put("inspectionState", detail.getInspectionState());
                  detailObj.put("pointStartTime", detail.getPointStartTime());
                  detailObj.put("pointEndTime", detail.getPointEndTime());
                  detailObj.put("sortNumber", detail.getSortNumber());
                  detailObj.put("lng", detail.getLng());
                  detailObj.put("lat", detail.getLat());
                  detailList.add(detailObj);
              }
              taskObj.put("details", detailList);
              taskList.add(taskObj);
          }
  
          result.put("staffName", staffName);

          result.put("tasks", taskList);
  

          context.setResponseEntity(new ResponseEntity<>(
                  ResultVo.createResponseEntity(result).toString(), HttpStatus.OK));

      }
  }