hdy
2024-04-18 e2670e7912ea7316d1f9fb518d980b8a4ef94055
jcdm-main/src/main/java/com/jcdm/main/em/inspectionPlanItemsProject/controller/EmInspectionPlanItemsProjectController.java
@@ -7,8 +7,10 @@
import javax.servlet.http.HttpServletResponse;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.jcdm.common.exception.ServiceException;
import com.jcdm.framework.websocket.WebSocketUsers;
import com.jcdm.main.em.inspectionPlanTask.domain.EmInspectionPlanTask;
import com.jcdm.main.em.inspectionPlanTask.service.IEmInspectionPlanTaskService;
@@ -44,16 +46,24 @@
    @GetMapping("/findPlanItemsProjectByPlanId")
    public AjaxResult list2(EmInspectionPlanItemsProject emInspectionPlanItemsProject)
    {
        EmInspectionPlanTask byId = emInspectionPlanTaskService.getById(emInspectionPlanItemsProject.getPlanTaskId());
        List<EmInspectionPlanItemsProject> planTaskIdList = emInspectionPlanItemsProjectService
                .list(new LambdaQueryWrapper<EmInspectionPlanItemsProject>()
                        .eq(EmInspectionPlanItemsProject::getPlanTaskId,emInspectionPlanItemsProject.getPlanTaskId())
                        .eq(EmInspectionPlanItemsProject::getItemsType,emInspectionPlanItemsProject.getItemsType()));
        if (StrUtil.isBlank(emInspectionPlanItemsProject.getPlanTaskCode())){
            throw new ServiceException("未获取到计划编号,请重试!");
        }
        Map<String,Object> resultMap = new HashMap<>();
        resultMap.put("planItemsProject", planTaskIdList);
        resultMap.put("planTask",byId);
        List<EmInspectionPlanTask> list = emInspectionPlanTaskService.list(new LambdaQueryWrapper<EmInspectionPlanTask>()
                .eq(EmInspectionPlanTask::getPlanCode, emInspectionPlanItemsProject.getPlanTaskCode()));
        if (CollUtil.isNotEmpty(list)){
            EmInspectionPlanTask byId = list.get(0);
            List<EmInspectionPlanItemsProject> planTaskIdList = emInspectionPlanItemsProjectService
                    .list(new LambdaQueryWrapper<EmInspectionPlanItemsProject>()
                            .eq(EmInspectionPlanItemsProject::getPlanTaskId,byId.getId())
                            .eq(EmInspectionPlanItemsProject::getItemsType,emInspectionPlanItemsProject.getItemsType()));
            resultMap.put("planItemsProject", planTaskIdList);
            resultMap.put("planTask",byId);
        }else {
            throw new ServiceException("未获取到计划数据!");
        }
        return AjaxResult.success(resultMap);
    }