优化巡检计划查询

mrzcc

2020-02-19 cbc0ac27033820a970bbeb013891f798e3be83da

优化巡检计划查询

 Api/src/main/java/com/java110/api/listener/inspectionPlan/ListInspectionPlansListener.java

@@ -2,9 +2,11 @@

import com.alibaba.fastjson.JSONObject;
import com.java110.api.listener.AbstractServiceApiListener;
import com.java110.core.smo.inspectionRoute.IInspectionRouteInnerServiceSMO;
import com.java110.core.smo.org.IOrgInnerServiceSMO;
import com.java110.core.smo.org.IOrgStaffRelInnerServiceSMO;
import com.java110.core.smo.user.IUserInnerServiceSMO;
import com.java110.dto.inspectionRoute.InspectionRouteDto;
import com.java110.dto.org.OrgStaffRelDto;
import com.java110.dto.user.UserDto;
import com.java110.utils.constant.ServiceCodeInspectionPlanConstant;
@@ -22,6 +24,7 @@
import org.springframework.http.ResponseEntity;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;

@@ -36,6 +39,8 @@
    private IInspectionPlanInnerServiceSMO inspectionPlanInnerServiceSMOImpl;
    @Autowired
    private IOrgStaffRelInnerServiceSMO iOrgStaffRelInnerServiceSMO;
    @Autowired
    private IInspectionRouteInnerServiceSMO inspectionRouteInnerServiceSMOImpl;


    @Override
@@ -80,9 +85,12 @@
        if (count > 0) {
            inspectionPlans = BeanConvertUtil.covertBeanList(inspectionPlanInnerServiceSMOImpl.queryInspectionPlans(inspectionPlanDto), ApiInspectionPlanDataVo.class);
            ArrayList staffIds = new ArrayList<>();
            ArrayList inspectionRouteIds = new ArrayList<>();
            for( ApiInspectionPlanDataVo Plans : inspectionPlans){
                if(Plans.getStaffId() != null){
                    staffIds.add(Plans.getStaffId());
                staffIds.add(Plans.getStaffId());
                String[] ids = Plans.getInspectionRouteId().split(",");
                for( String s : ids){
                    inspectionRouteIds.add(s);
                }
            }
            if(staffIds.size() > 0){
@@ -102,6 +110,32 @@
                    }
                }
            }
            if(inspectionRouteIds.size() > 0){
                //去重
                HashSet set = new HashSet(inspectionRouteIds);
                inspectionRouteIds.clear();
                inspectionRouteIds.addAll(set);
                InspectionRouteDto inspectionRouteDto = new InspectionRouteDto();
                String[] routeIds = (String[]) inspectionRouteIds.toArray(new String[inspectionRouteIds.size()]);
                inspectionRouteDto.setInspectionRouteIds(routeIds);
                List<InspectionRouteDto> inspectionRouteDtoList = inspectionRouteInnerServiceSMOImpl.queryInspectionRoutes(inspectionRouteDto);
                for( ApiInspectionPlanDataVo planDataVo : inspectionPlans){
                    String[] routeIdArray = planDataVo.getInspectionRouteId().split(",");
                    for( String s : routeIdArray){
                        for( InspectionRouteDto inspectionRouteDto1 : inspectionRouteDtoList){
                            if(inspectionRouteDto1.getInspectionRouteId().equals(s)){
                                if(planDataVo.getInspectionRouteName() == null){
                                    planDataVo.setInspectionRouteName(inspectionRouteDto1.getRouteName());
                                }else{
                                    planDataVo.setInspectionRouteName(planDataVo.getInspectionRouteName()+","+inspectionRouteDto1.getRouteName());
                                }
                            }
                        }

                    }
                }
            }

        } else {
            inspectionPlans = new ArrayList<>();
        }