fj-dc/.svn/pristine/d3/d3dc741a5e01434f2f8c3671d8cc749604248337.svn-base

package cn.com.goldenwater.dcproj.controller.pblm;

import cn.com.goldenwater.core.web.BaseController;
import cn.com.goldenwater.core.web.BaseResponse;
import cn.com.goldenwater.dcproj.dao.ObjInspPblmsDao;
import cn.com.goldenwater.dcproj.dto.ObjInspPblmsDto;
import cn.com.goldenwater.dcproj.model.ObjInspPblms;
import cn.com.goldenwater.dcproj.model.ObjTypeCheck;
import cn.com.goldenwater.dcproj.model.OlBisInspOrg;
import cn.com.goldenwater.dcproj.param.ObjInspPblmsParam;
import cn.com.goldenwater.dcproj.param.ObjInspPblmsParams;
import cn.com.goldenwater.dcproj.param.WagaPblm;
import cn.com.goldenwater.dcproj.service.ObjInspPblmsService;
import cn.com.goldenwater.dcproj.service.OlBisInspOrgService;
import cn.com.goldenwater.id.util.UuidUtil;
import cn.com.goldenwater.target.CheckException;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.util.Assert;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletRequest;
import java.util.*;

import static cn.com.goldenwater.dcproj.util.CheckUtil.check;

/**
 * @author lune
 * @date 2019-2-18
 */
@Api(value = "小水库督查问题分类标准库", tags = "小水库督查问题分类标准库")
@RestController
@RequestMapping("/dc/insp/pblms")
public class ObjInspPblmsController extends BaseController {

    private Logger logger = LoggerFactory.getLogger(getClass());

    @Autowired
    private ObjInspPblmsService objInspPblmsService;
    @Autowired
    private OlBisInspOrgService olBisInspOrgService;

    @ApiOperation(value = "添加问题分类标准库")
    @RequestMapping(value = "", method = RequestMethod.POST)
    public BaseResponse<String> insert(@ApiParam(name = "objInspPblms", value = "ObjInspPblms", required = true) @RequestBody ObjInspPblms objInspPblms) {
        String uuid = UuidUtil.uuid();
        objInspPblms.setGuid(uuid);
        int ret = objInspPblmsService.insert(objInspPblms);
        return buildSuccessResponse(uuid);
    }

    @ApiOperation(value = "获取问题类别")
    @RequestMapping(value = "/getPblmType", method = {RequestMethod.GET, RequestMethod.POST})
    public BaseResponse<List<ObjInspPblms>> getPblmType(ObjInspPblmsParam objInspPblmsParam) {
        if (StringUtils.isBlank(objInspPblmsParam.getType()) &&
                StringUtils.isBlank(objInspPblmsParam.getPguid())) {
            return buildSuccessResponse(objInspPblmsService.findList(null));
        }
        String orgId = getCurrentOrgId();
        OlBisInspOrg olBisInspOrg = olBisInspOrgService.get(orgId);
        Optional.ofNullable(olBisInspOrg).orElseThrow(() -> new CheckException("当前机构不存在!!"));

        if ("1".equals(olBisInspOrg.getPblmLogo()) && "1".equals(objInspPblmsParam.getType())) {
            objInspPblmsParam.setOrgId(olBisInspOrg.getQuoteOrgId());
        } else if ("22".equals(objInspPblmsParam.getType())) {
            objInspPblmsParam.setPguid(null);
            objInspPblmsParam.setOrgId(null);
//            if ("063".equals(orgId)) {
//                objInspPblmsParam.setOrgId(orgId);
//            }
        } else {
            objInspPblmsParam.setOrgId(null);
        }
        List<ObjInspPblms> objInspPblms;
        objInspPblms = objInspPblmsService.findList(objInspPblmsParam);
        //如果没找到问题清单，则设置通用问题清单
        if (objInspPblms.size() == 0) {
            ObjInspPblmsParam objInspPblmsParam1 = new ObjInspPblmsParam();
            objInspPblmsParam1.setType("111");
            objInspPblms = objInspPblmsService.findList(objInspPblmsParam1);
            for (ObjInspPblms item : objInspPblms) {
                item.setType(objInspPblmsParam.getType());
            }

        }
        return buildSuccessResponse(objInspPblms);
    }

    @ApiOperation(value = "获取根据附件表问题类别")
    @RequestMapping(value = "/getPblmTypeInAttach", method = {RequestMethod.POST})
    public BaseResponse<List<ObjInspPblms>> getPblmTypeInAttach(@RequestBody ObjInspPblmsParams objInspPblms) {
        List<ObjInspPblms> obj = new ArrayList<>();
        try {
            obj = objInspPblmsService.getPblmTypeInAttach(objInspPblms);
        } catch (Exception e) {
            e.printStackTrace();
            logger.error(e.getMessage());
            return buildFailResponse(e.getMessage());
        }
        return buildSuccessResponse(obj);
    }

    @ApiOperation(value = "根据ID删除问题分类标准库")
    @RequestMapping(value = "/{id}", method = RequestMethod.POST)
    public BaseResponse delete(@ApiParam(name = "id", value = "id", required = true) @PathVariable String id) {
        int ret = objInspPblmsService.delete(id);
        return buildSuccessResponse();
    }

    @ApiOperation(value = "更新问题分类标准库信息")
    @RequestMapping(value = "/update", method = RequestMethod.POST)
    public BaseResponse update(@ApiParam(name = "objInspPblms", value = "ObjInspPblms", required = true) @RequestBody ObjInspPblms objInspPblms) {
        Assert.notNull(objInspPblms.getGuid(), "主键id为必填参数");
        int ret = objInspPblmsService.update(objInspPblms);
        return buildSuccessResponse();
    }

    @ApiOperation(value = "根据ID获取问题分类标准库（单表）")
    @RequestMapping(value = "/{id}", method = RequestMethod.GET)
    public BaseResponse<ObjInspPblms> get(@ApiParam(name = "id", value = "id", required = true) @PathVariable String id) {
        ObjInspPblms objInspPblms = objInspPblmsService.get(id);
        return buildSuccessResponse(objInspPblms);
    }


    @ApiOperation(value = "获取整个标准库数据")
    @RequestMapping(value = "/findAll", method = RequestMethod.GET)
    public BaseResponse<List<ObjInspPblms>> findAll() {
        List<ObjInspPblms> pblmsList = objInspPblmsService.findList(null);
        return buildSuccessResponse(pblmsList);
    }


    @ApiOperation(value = "获取所有数据类型")
    @RequestMapping(value = "/getPblmsType", method = {RequestMethod.GET})
    public BaseResponse<List<String>> getPblmsType() {
        List<String> pblmsType = objInspPblmsService.findPblmsName(getCurrentOrgId());
        return buildSuccessResponse(pblmsType);
    }

    @ApiOperation(value = "获取所有的问题名称，去重")
    @RequestMapping(value = "/getPblmNames", method = RequestMethod.GET)
    public BaseResponse<List<String>> getPblmNames(ObjInspPblmsParam objInspPblmsParam) {
        OlBisInspOrg olBisInspOrg = olBisInspOrgService.get(getCurrentOrgId());
        if (olBisInspOrg == null) {
            throw new CheckException("当前机构不存在!!");
        }
        if ("1".equals(olBisInspOrg.getPblmLogo()) && "1".equals(objInspPblmsParam.getType())) {
            objInspPblmsParam.setOrgId(olBisInspOrg.getQuoteOrgId());
        } else {
            objInspPblmsParam.setOrgId("");
        }
        List<String> pblmNames = objInspPblmsService.getPblmNames(objInspPblmsParam);
        return buildSuccessResponse(pblmNames);
    }

    @ApiOperation(value = "获取所有数据检查项")
    @RequestMapping(value = "/getCheckPoints", method = {RequestMethod.POST})
    public BaseResponse<List<String>> getPblmsType(@RequestBody ObjInspPblmsParam objInspPblmsParam) {
        if (StringUtils.isBlank(objInspPblmsParam.getType())) {
            objInspPblmsParam.setType("1");
        }
        List<String> pblmsType = objInspPblmsService.findPblmsCheckPoints(objInspPblmsParam, getCurrentOrgId());
        return buildSuccessResponse(pblmsType);
    }

    @ApiOperation(value = "获取所有问题描述信息")
    @RequestMapping(value = "/getPblmsDesc", method = {RequestMethod.POST})
    public BaseResponse<List<String>> getPblmsDesc(@RequestBody ObjInspPblmsParam objInspPblmsParam) {
        if (StringUtils.isBlank(objInspPblmsParam.getType())) {
            objInspPblmsParam.setType("1");
        }
        OlBisInspOrg olBisInspOrg = olBisInspOrgService.get(getCurrentOrgId());
        if (olBisInspOrg == null) {
            throw new CheckException("当前机构不存在!!");
        }
        Map<String, String> params = new HashMap<>();
        params.put("type", objInspPblmsParam.getType());
        params.put("inspPblmsName", objInspPblmsParam.getInspPblmsName());
        params.put("checkPoint", objInspPblmsParam.getCheckPoint());
        if ("1".equals(olBisInspOrg.getPblmLogo()) && "1".equals(objInspPblmsParam.getType())) {
            params.put("orgId", olBisInspOrg.getQuoteOrgId());
        }
        List<String> pblmsType = objInspPblmsService.findPblmsDesc(params);
        return buildSuccessResponse(pblmsType);
    }

    @ApiOperation(value = "获取标准表单个信息")
    @RequestMapping(value = "/getByPblms", method = {RequestMethod.POST})
    public BaseResponse<ObjInspPblms> getByPblms(@RequestBody ObjInspPblmsParam objInspPblmsParam) {
        check(StringUtils.isNotBlank(objInspPblmsParam.getType()), "type.no");

        Map<String, String> params = new HashMap<>(6);
        params.put("inspPblmsName", objInspPblmsParam.getInspPblmsName());
        params.put("checkPoint", objInspPblmsParam.getCheckPoint());
        params.put("pblmDesc", objInspPblmsParam.getPblmDesc());
        params.put("type", objInspPblmsParam.getType());
        params.put("guid", objInspPblmsParam.getGuid());
        String orgId = getCurrentOrgId();
        OlBisInspOrg olBisInspOrg = olBisInspOrgService.get(orgId);
        if ("1".equals(olBisInspOrg.getPblmLogo()) && "1".equals(objInspPblmsParam.getType())) {
            params.put("orgId", olBisInspOrg.getQuoteOrgId());
        } else {
            params.put("orgId", "");
        }
        ObjInspPblms pblmsType = objInspPblmsService.getByPblms(params);
        return buildSuccessResponse(pblmsType);
    }

    @ApiOperation(value = "获取标准表单个信息")
    @RequestMapping(value = "/getPblmsByType", method = {RequestMethod.POST})
    public BaseResponse<ObjInspPblms> getPblmsByType(@RequestBody ObjInspPblmsParam objInspPblmsParam) {
        String orgId = getCurrentOrgId();
        OlBisInspOrg olBisInspOrg = olBisInspOrgService.get(orgId);
        if (olBisInspOrg == null) {
            throw new CheckException("当前机构不存在!!");
        }
        if ("1".equals(olBisInspOrg.getPblmLogo()) && "1".equals(objInspPblmsParam.getType())) {
            objInspPblmsParam.setOrgId(olBisInspOrg.getQuoteOrgId());
        } else {
            objInspPblmsParam.setOrgId(null);
        }
        ObjInspPblms pblmsType = objInspPblmsService.getPblmsByType(objInspPblmsParam);
        return buildSuccessResponse(pblmsType);
    }


    @Autowired
    private ObjInspPblmsDao objInspPblmsDao;

    @ApiOperation(value = "获取标准表单个信息(未启用)")
    @RequestMapping(value = "/getByWagaPblm", method = {RequestMethod.POST})
    public BaseResponse<Map<String, Object>> getByPblms(@RequestBody WagaPblm wagaPblm) {
        String sn = wagaPblm.getSn();
        if (StringUtils.isNotBlank(sn)) {
            sn = sn.replace("至", "-");
            sn = sn.replace("到", "-");
            StringBuffer buffer = new StringBuffer();
            if (sn.contains(",")) {
                String[] arrays = sn.split(",");
                for (String array : arrays) {
                    if (array.contains("-")) {
                        String[] startAndEnd = array.split("-");
                        if (startAndEnd.length == 2) {
                            int start = Integer.parseInt(startAndEnd[0]);
                            int end = Integer.parseInt(startAndEnd[1]);
                            for (int i = start; i <= end; i++) {
                                buffer.append(i + ",");
                            }
                        } else {
                            buffer.append(startAndEnd[0] + ",");
                        }
                    } else {
                        buffer.append(array + ",");
                    }
                }
            } else if (sn.contains("-")) {
                String[] startAndEnd = sn.split("-");
                if (startAndEnd.length == 2) {
                    int start = Integer.parseInt(startAndEnd[0]);
                    int end = Integer.parseInt(startAndEnd[1]);
                    for (int i = start; i <= end; i++) {
                        buffer.append(i + ",");
                    }
                } else {
                    buffer.append(startAndEnd[0] + ",");
                }
            } else {
                buffer.append(sn);
            }
            String snStr = buffer.toString();
            if (snStr.endsWith(",")) {
                snStr = snStr.substring(0, snStr.length() - 1);
            }
            wagaPblm.setSn(snStr);
            List<ObjInspPblms> inspPblms = objInspPblmsDao.findWagaPblm(wagaPblm.getSn(), wagaPblm.getAttach(), getCurrentOrgId());
            Map<String, Object> retMap = new HashMap<>();
            List<String> pblmNames = new ArrayList<>();
            for (ObjInspPblms pblms : inspPblms) {
                if (!pblmNames.contains(pblms.getInspPblmsName())) {
                    pblmNames.add(pblms.getInspPblmsName());
                }
            }
            retMap.put("pblms", pblmNames);
            retMap.put("datas", inspPblms);
            return buildSuccessResponse(retMap);
        }
        return buildSuccessResponse(null);

    }

    @ApiOperation(value = "根据规则id和类型获取对应的描述信息")
    @RequestMapping(value = "/getPblmsByQuesNum/{id}/{type}", method = {RequestMethod.GET})
    public BaseResponse<List<ObjTypeCheck>> getPblmsByQuesNum(@ApiParam(name = "id", value = "规则id", required = true) @PathVariable String id,
                                                              @ApiParam(name = "type", value = "类型", required = true) @PathVariable String type) {
        // 1. 跟id取出对应的 规则(问题标准 0:全国标准，1:自定义标准)
        OlBisInspOrg olBisInspOrg = olBisInspOrgService.get(getCurrentOrgId());
        if (olBisInspOrg == null) {
            throw new CheckException("当前机构不存在!!");
        }

        List<ObjTypeCheck> bisInspItemQuesNumDtoList = objInspPblmsService.getPblmsByQuesNum(id, type, olBisInspOrg);
        return buildSuccessResponse(bisInspItemQuesNumDtoList);

    }

    @ApiOperation(value = "根据督查类型获取所有的问题标准的分类、检查项和描述信息")
    @RequestMapping(value = "/getAllPblmsByType/{type}", method = {RequestMethod.GET})
    public BaseResponse getAllPblmsByType(@ApiParam(name = "type", value = "类型", required = true)
                                          @PathVariable String type) {
        return buildSuccessResponse(objInspPblmsService.getAllPblmsByType(type, getCurrentOrgId()));
    }

    @Autowired
    private RedisTemplate redisTemplate;

    @ApiOperation(value = "获取小水库(水闸)标准数据tree类型：type=1水库，type=6水闸")
    @RequestMapping(value = "/getPblmsRverTree/{type}", method = {RequestMethod.GET})
    public BaseResponse<List<ObjInspPblmsDto>> getPblmsRverTree(@PathVariable(required = true) String type, HttpServletRequest request) {
        //第一节点

        OlBisInspOrg olBisInspOrg = olBisInspOrgService.get(getCurrentOrgId());
        String orgId = "";
        if ("1".equals(olBisInspOrg.getPblmLogo()) && "1".equals(type)) {
            orgId = olBisInspOrg.getQuoteOrgId();
        }
        String pblmsName = request.getParameter("pblmsName");
        String sort1 = request.getParameter("sort1");
        List<ObjInspPblmsDto> pblmsType = new ArrayList<>();
        List<ObjInspPblmsDto> pblmsList = new ArrayList<>();
        pblmsType = objInspPblmsService.findPblmsDtoName(type, pblmsName, orgId, sort1);
        pblmsList = objInspPblmsService.findPblmsDTO(type, pblmsName, orgId, sort1);
        Set<ObjInspPblmsDto> secondList = new HashSet<>();
        if (pblmsList == null) {
            throw new CheckException("未查到任何数据!!");
        }
        for (ObjInspPblmsDto pblm : pblmsList) {
            ObjInspPblmsDto secChild = new ObjInspPblmsDto();
            secChild.setCheckPoint(pblm.getCheckPoint());
            secChild.setInspPblmsName(pblm.getInspPblmsName());
            secondList.add(secChild);
        }
        for (ObjInspPblmsDto pblm : pblmsType) {
            List<ObjInspPblmsDto> chlist = new ArrayList<>();
            for (ObjInspPblmsDto sec : secondList) {
                if (pblm.getInspPblmsName().equals(sec.getInspPblmsName())) {
                    chlist.add(sec);
                }
            }
            pblm.setChildren(chlist);
        }

        for (ObjInspPblmsDto sec : secondList) {
            List<ObjInspPblmsDto> chlist = new ArrayList<>();
            for (ObjInspPblmsDto pblm : pblmsList) {
                if (sec.getInspPblmsName().equals(pblm.getInspPblmsName()) && sec.getCheckPoint().equals(pblm.getCheckPoint())) {
                    chlist.add(pblm);
                }
            }
            sec.setChildren(chlist);
        }

        return buildSuccessResponse(pblmsType);
    }

}