fj-dc/.svn/pristine/2c/2c31bf97705a3124e43d7e313c72b293917c016c.svn-base

package cn.com.goldenwater.dcproj.service.impl.general;

import cn.com.goldenwater.dcproj.constValue.SplitValue;
import cn.com.goldenwater.dcproj.dao.BisInspAllDao;
import cn.com.goldenwater.dcproj.dao.BisInspAllObjDao;
import cn.com.goldenwater.dcproj.dao.BisInspPblmDao;
import cn.com.goldenwater.dcproj.dao.BisInspRsvrRgstrDao;
import cn.com.goldenwater.dcproj.dao.BisInspWtdstDao;
import cn.com.goldenwater.dcproj.dao.GeneralDownloadDao;
import cn.com.goldenwater.dcproj.dto.BisInspWtdstDetailDto;
import cn.com.goldenwater.dcproj.model.BisInspAll;
import cn.com.goldenwater.dcproj.model.BisInspAllRlation;
import cn.com.goldenwater.dcproj.model.BisInspKeyRegister;
import cn.com.goldenwater.dcproj.model.BisInspPblm;
import cn.com.goldenwater.dcproj.param.BisInspPblmParam;
import cn.com.goldenwater.dcproj.param.BisInspWtdstDtoListParam;
import cn.com.goldenwater.dcproj.param.TypeExportParam;
import cn.com.goldenwater.dcproj.param.TypeParam;
import cn.com.goldenwater.dcproj.service.GeneralDownloadService;
import cn.com.goldenwater.dcproj.service.OlBisInspOrgService;
import cn.com.goldenwater.dcproj.utils.ZipUtil;
import cn.com.goldenwater.dcproj.utils.export.CreateExcelHelper;
import cn.com.goldenwater.dcproj.utils.export.CreateRyTables;
import cn.com.goldenwater.dcproj.utils.export.CreateShTables;
import cn.com.goldenwater.dcproj.utils.export.TableEnum;
import cn.com.goldenwater.dcproj.utils.export.template.CreateRyTablesByTemplate;
import cn.com.goldenwater.dcproj.utils.export.template.CreateSkTablesByTemplate;
import cn.com.goldenwater.util.common.FileUtils;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * @ClassName GeneralDownloadServiceImpl
 * @Description 通用下载实现类
 * @Author liyz
 * @Date 2019/3/6 10:56
 * @Version 1.0
 **/

@Service
@Transactional(rollbackFor = Exception.class)
public class GeneralDownloadServiceImpl implements GeneralDownloadService {

    private final SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
    @Value("${export.basePath}")
    public String exportBasePath;

    @Autowired
    private CreateRyTables createRyTables;
    @Autowired
    private CreateRyTablesByTemplate createRyTablesByTemplate;
    @Autowired
    private CreateShTables createShTables;
    @Autowired
    private CreateExcelHelper createExcelHelper;
    @Autowired
    private BisInspAllObjDao bisInspAllObjDao;
    @Autowired
    private BisInspWtdstDao bisInspWtdstDao;
    @Autowired
    private BisInspRsvrRgstrDao bisInspRsvrRgstrDao;
    @Autowired
    private BisInspPblmDao bisInspPblmDao;
    @Autowired
    private CreateSkTablesByTemplate createSkTablesByTemplate;
    @Autowired
    private GeneralDownloadDao generalDownloadDao;
    @Autowired
    private BisInspAllDao bisInspAllDao;

    public GeneralDownloadServiceImpl() {
        super();
    }

    /**
     * 人饮相关
     *
     * @param typeExportParam typeExportParam
     * @param printWriter     printWriter
     */
    @Override
    public void createExcel(TypeExportParam typeExportParam, PrintWriter printWriter) {

        if (StringUtils.isNotBlank(typeExportParam.getEngIds()) && StringUtils.isNotBlank(typeExportParam.getTableIds())) {
            try {
                printWriter.println("准备导出，开始创建导出目录<br>");
                String filePath = UUID.randomUUID().toString();
                String dir = exportBasePath + File.separator + filePath + File.separator;
                String zipFile = exportBasePath + File.separator + filePath + ".zip";
                FileUtils.mkFolder(dir);

                // String转List,表id不过滤
                List<String> tablesList = Arrays.asList(typeExportParam.getTableIds().split(SplitValue.DOUHAO_SPLIT));
                // String转List,engIds过滤
                List<String> endIdsList = Stream
                        .of(typeExportParam.getEngIds().split(SplitValue.DOUHAO_SPLIT))
                        // 过滤掉空字符串
                        .filter(StringUtils::isNotBlank)
                        // 转成集合
                        .collect(Collectors.toList());

                int zipStatus = 0;
                // 如果合并
                if ("1".equals(typeExportParam.getSumState())) {
                    zipStatus = initSumTable(tablesList, endIdsList, dir, printWriter, zipStatus);
                } else {
                    zipStatus = initNotSumTable(tablesList, endIdsList, dir, printWriter, zipStatus);
                }

                if (zipStatus > 0) {
                    // 指定压缩源，可以是目录或文件
                    String src = dir;
                    // 压缩包路径
                    String archive = zipFile;
                    // 压缩包注释
                    String comment = "Java Zip SLDC";
                    // 压缩文件或目录
                    ZipUtil.zip(src, archive, comment);

                    printWriter.write("创建完成。压缩完成。 <a href='/api/dc/gd/base/download/" + filePath + "' download>点击下载</a>");
                }

            } catch (IOException e) {
                e.printStackTrace();
            } finally {
                printWriter.close();
            }
        } else {
            printWriter.println("请选择地区和表");
        }

    }

    private int initSumTable(List<String> tablesList, List<String> endIdsList, String dir, PrintWriter printWriter, int zipStatus) {
        for (int j = 0; j < tablesList.size(); j++) {
            String tableId = tablesList.get(j);
            TableEnum tableEnum = TableEnum.getTypeByValue(tableId);
            String tableName = TableEnum.getNameByValue(tableId);
            try {
                printWriter.println(sdf.format(new Date()) + " 开始创建 " + tableName + ".xls" + "<br>");
                printWriter.flush();
                switch (tableEnum) {
                    case T1:
                        int zipStatusT1 = createRyTables.createBisInspSecsurveyVlgSumExcel(endIdsList, dir, tableName);
                        printMessage(zipStatusT1, printWriter, tableName);
                        zipStatus += zipStatusT1;
                        break;
                    case T2:
                        int zipStatusT2 = createRyTables.createBisInspVlgdrinkFacOperSumExcel(endIdsList, dir, tableName);
                        printMessage(zipStatusT2, printWriter, tableName);
                        zipStatus += zipStatusT2;
                        break;
                    case T3:
                        int zipStatusT3 = createRyTablesByTemplate.createBisInspWateruserInfoSumExcel(endIdsList, dir, tableName);
                        printMessage(zipStatusT3, printWriter, tableName);
                        zipStatus += zipStatusT3;
                        break;
                    case T4:
                        int zipStatusT4 = createRyTablesByTemplate.createBisInspVlgdrinkProjManageSumExcel(endIdsList, dir, tableName);
                        printMessage(zipStatusT4, printWriter, tableName);
                        zipStatus += zipStatusT4;
                        break;
                    case T5:
                        int zipStatusT5 = createRyTables.createBisInspProSourceProtectSumExcel(endIdsList, dir, tableName);
                        printMessage(zipStatusT5, printWriter, tableName);
                        zipStatus += zipStatusT5;
                        break;
                    case T6:
                        int zipStatusT6 = createRyTablesByTemplate.createBisInspManageResponsibilitySumExcel(endIdsList, dir, tableName);
                        printMessage(zipStatusT6, printWriter, tableName);
                        zipStatus += zipStatusT6;
                        break;
                    default:
                        break;
                }
            } catch (Exception e) {
                printWriter.close();
                e.printStackTrace();
            }
        }
        return zipStatus;
    }

    private int initNotSumTable(List<String> tablesList, List<String> endIdsList, String dir, PrintWriter printWriter, int zipStatus) {
        for (int i = 0; i < endIdsList.size(); i++) {
            String engId = endIdsList.get(i);
            String adCode = generalDownloadDao.getCodeByEngId(engId);
            String adName = createExcelHelper.getAdName(adCode);
            for (int j = 0; j < tablesList.size(); j++) {
                String tableId = tablesList.get(j);
                TableEnum tableEnum = TableEnum.getTypeByValue(tableId);
                String tableName = adName + TableEnum.getNameByValue(tableId);
                try {
                    printWriter.println(sdf.format(new Date()) + " 开始创建 " + tableName + ".xls" + "<br>");
                    printWriter.flush();
                    switch (tableEnum) {
                        case T1:
                            int zipStatusT1 = createRyTables.createBisInspSecsurveyVlgExcel(engId, dir, tableName);
                            printMessage(zipStatusT1, printWriter, tableName);
                            zipStatus += zipStatusT1;
                            break;
                        case T2:
                            int zipStatusT2 = createRyTables.createBisInspVlgdrinkFacOperExcel(engId, dir, tableName);
                            printMessage(zipStatusT2, printWriter, tableName);
                            zipStatus += zipStatusT2;
                            break;
                        case T3:
                            int zipStatusT3 = createRyTablesByTemplate.createBisInspWateruserInfoExcel(engId, dir, tableName);
                            printMessage(zipStatusT3, printWriter, tableName);
                            zipStatus += zipStatusT3;
                            break;
                        case T4:
                            int zipStatusT4 = createRyTablesByTemplate.createBisInspVlgdrinkProjManageExcel(engId, dir, tableName);
                            printMessage(zipStatusT4, printWriter, tableName);
                            zipStatus += zipStatusT4;
                            break;
                        case T5:
                            int zipStatusT5 = createRyTables.createBisInspProSourceProtectExcel(engId, dir, tableName);
                            printMessage(zipStatusT5, printWriter, tableName);
                            zipStatus += zipStatusT5;
                            break;
                        case T6:
                            int zipStatusT6 = createRyTablesByTemplate.createBisInspManageResponsibilityExcel(engId, dir, tableName);
                            printMessage(zipStatusT6, printWriter, tableName);
                            zipStatus += zipStatusT6;
                            break;
                        default:
                            break;
                    }
                    // 这是一个所有的判断，最好是能够每张表都判断一次

                } catch (Exception e) {
                    printWriter.close();
                    e.printStackTrace();
                }
            }
        }
        return zipStatus;
    }

    private void printMessage(int zipStatus, PrintWriter printWriter, String tableName) {
        if (zipStatus > 0) {
            printWriter.println(sdf.format(new Date()) + " 完成创建 " + tableName + ".xls" + "<br>");
            printWriter.flush();
        } else {
            printWriter.println(sdf.format(new Date()) + " 此表暂无数据可导出" + "<br>");
            printWriter.flush();
        }
    }

    /**
     * 水毁相关
     *
     * @param typeExportParam typeExportParam
     * @param printWriter     printWrite
     */
    @Override
    public void createShuiHuiExcel(TypeExportParam typeExportParam, PrintWriter printWriter) {

        if (StringUtils.isNotBlank(typeExportParam.getIds()) && StringUtils.isNotBlank(typeExportParam.getTableIds())) {
            try {
                // 获取adCodes
                String[] ids = typeExportParam.getIds().split(SplitValue.DOUHAO_SPLIT);
                List<String> adCodes = new ArrayList<>();
                for (String id : ids) {
                    List<Map<String, Object>> listId = generalDownloadDao.getCodeById(id);
                    if (listId.size() > 0) {
                        adCodes.addAll(listId.stream().filter(x -> x.get("CODE") != null).map(x -> x.get("CODE").toString()).collect(Collectors.toList()));
                    }
                }

                // 水毁的AdCodes 其实是WTDST_ID，需要转化一下
                List<String> adCodesList = adCodes
                        .stream()
                        .map(x -> (bisInspWtdstDao.get(x).getLocAd() + "").substring(0, 2))
                        .distinct()
                        .collect(Collectors.toList());
                String filePath = UUID.randomUUID().toString();
                String dir = exportBasePath + File.separator + filePath + File.separator;
                String zipFile = exportBasePath + File.separator + filePath + ".zip";
                FileUtils.mkFolder(dir);

                int i = 0;
                List<String> tablesList = Arrays.asList(typeExportParam.getTableIds().split(SplitValue.DOUHAO_SPLIT));
                for (String tableId : tablesList) {
                    TableEnum tableEnum = TableEnum.getTypeByValue(tableId);


                    switch (tableEnum) {
                        case T7:
                            if ("1".equals(typeExportParam.getSumState())) {
                                typeExportParam.setAdCodes(null);
                                i = createShTables.createBisInspWtdstDtoExcel(typeExportParam, printWriter, null, dir);
                            } else {
                                for (String adCode : adCodesList) {
                                    i = createShTables.createBisInspWtdstDtoExcel(typeExportParam, printWriter, adCode, dir);
                                }
                            }
                            break;
                    }
                }
                if (i == 0) {
                    printWriter.println("暂无数据可导出。<br>");
                    printWriter.flush();
                    return;
                }


                // 指定压缩源，可以是目录或文件
                String src = dir;
                // 压缩包路径
                String archive = zipFile;
                // 压缩包注释
                String comment = "Java Zip SLDC";
                // 压缩文件或目录
                ZipUtil.zip(src, archive, comment);

                printWriter.write("创建完成。压缩完成。 <a href='/api/dc/gd/base/download/" + filePath + "' download>点击下载</a>");
                printWriter.flush();

            } catch (Exception e) {
                e.printStackTrace();
            }

        } else {
            printWriter.println("请选择地区和表");
            printWriter.flush();
        }
        printWriter.close();

    }


    /**
     * 水库相关
     *
     * @param typeExportParam typeExportParam
     * @param printWriter     printWriter
     */
    @Override
    public void createShuiKuExcel(TypeExportParam typeExportParam, PrintWriter printWriter) {

        if (StringUtils.isNotBlank(typeExportParam.getIds()) && StringUtils.isNotBlank(typeExportParam.getTableIds())) {

            //导出
            try {
                TableEnum tableEnum = TableEnum.getTypeByValue("08");
                switch (tableEnum) {
                    case T8:
                        createSkTablesByTemplate.createBisInspRsvrRgstrPcDtoExcel(typeExportParam, printWriter);
                        break;
                }

            } catch (Exception e) {
                e.printStackTrace();
            }
        } else {
            printWriter.println("请选择地区和表");
            printWriter.flush();
        }
        printWriter.close();

    }

    /**
     * 下载
     *
     * @param uuid 压缩包的名称
     */
    @Override
    public void downLoad(String uuid) {

    }


    //------------------------------- 其他业务方法 -------------------------------------------

    /**
     * 通过用户ID获取行政区划列表(截取前两位)
     *
     * @param presId 用户Id
     * @param ptype  类型 1水库 2人饮 3水毁
     * @return 行政区划(截取前两位)列表
     */
    @Override
    public String selectAdCodeByUserId(String presId, String ptype, String orgId) {
        return bisInspRsvrRgstrDao.selectAdCodeByUserId(presId, ptype, orgId);
    }

    @Override
    public String getPidByPersid(String persid, String type, String adCode) {
        if ("1".equals(type)) {
            type = "001";
        } else if ("2".equals(type)) {
            type = "002";
        } else if ("3".equals(type)) {
            type = "003";
        }
        List<BisInspAllRlation> list = bisInspAllObjDao.getIdByUserId(persid, type, adCode);
        if (list.size() > 0) {
            return list.get(0).getId();
        }
        return "000";
    }

    @Override
    public PageInfo<BisInspWtdstDetailDto> findDtoListAboutExport(BisInspWtdstDtoListParam p, TypeExportParam exp) {
        if (org.apache.commons.lang3.StringUtils.isNotBlank(p.getLocAd())) {
            exp.setAdCodes(p.getLocAd().substring(0, 2));
        }
        exp.setPageNum(p.getPageNum());
        exp.setPageSize(p.getPageSize());
        PageHelper.startPage(exp.getPageNum(), exp.getPageSize());
        List<BisInspWtdstDetailDto> list = bisInspWtdstDao.findDtoListAboutExport(exp);
        for (BisInspWtdstDetailDto dto : list) {
            BisInspPblmParam param = new BisInspPblmParam();
            param.setRegid(dto.getWtdstId());
            param.setObjId(dto.getObjId());
            List<BisInspPblm> pblms = bisInspPblmDao.findList(param);
            dto.setList(pblms);
            dto.setProblmNum(pblms.size());
        }
        return new PageInfo<BisInspWtdstDetailDto>(list);
    }

    @Autowired
    private OlBisInspOrgService olBisInspOrgService;

    @Override
    public List<BisInspAll> getRyNode(String userid, String orgType, String orgId) {
        String province = olBisInspOrgService.getProvince(orgId);
        List<BisInspAll> list = bisInspAllDao.getAllNode(userid, orgType, province, orgId, "", "", "", "","","");
        if (list.size() > 0) {
            List<BisInspAll> listEng = generalDownloadDao.getRyNode();
            list.addAll(listEng);
        }
        return list;
    }

    @Override
    public List<BisInspAll> getRyNodeProvincial(String userid, String orgType, String level, String province) {
        return generalDownloadDao.getRyNodeProvincial(userid, orgType, level, province);
    }

    @Override
    public List<BisInspAll> getSzyNodeProvincial(String userid) {
        return generalDownloadDao.getSzyNodeProvincial(userid);
    }

    @Override
    public List<BisInspAll> getLyNode(String userid, String orgType, String orgId) {
        return generalDownloadDao.getLyNode(userid, orgType, orgId);
    }

    /**
     * @see cn.com.goldenwater.dcproj.controller.rsvr.BisInspRsvrRgstrController
     */
    @Override
    public List<BisInspKeyRegister> getGcNode(TypeParam typeParam) {
        return generalDownloadDao.getGcNode(typeParam);
    }

    @Override
    public List<BisInspAll> getRsraqgNodeProvincial(String userid, String level, String province) {
        return generalDownloadDao.getRsraqgNodeProvincial(userid, level, province);
    }

    @Override
    public List<BisInspAll> getNodeProvincialExceptGroup(Map<String, String> param) {
        return generalDownloadDao.getNodeProvincialExceptGroup(param);
    }
}