解决查询出错

13693261870

2024-03-25 1c285b38777e9f37ee8d112e994443d3640b78c1

 src/main/java/com/lf/server/service/sys/ReportService.java

@@ -1,37 +1,74 @@
package com.lf.server.service.sys;

import com.lf.server.entity.all.RedisCacheKey;
import com.lf.server.entity.all.StaticData;
import com.lf.server.entity.ctrl.CoordinateEntity;
import com.lf.server.entity.ctrl.CountEntity;
import com.lf.server.entity.data.DownloadEntity;
import com.lf.server.entity.sys.AttachEntity;
import com.lf.server.entity.sys.ReportEntity;
import com.lf.server.helper.StringHelper;
import com.lf.server.helper.WordHelper;
import com.lf.server.entity.sys.UserEntity;
import com.lf.server.helper.*;
import com.lf.server.mapper.sys.ReportMapper;
import org.springframework.beans.factory.annotation.Autowired;
import com.lf.server.service.all.RedisService;
import com.lf.server.service.data.DirService;
import com.lf.server.service.data.DownloadService;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import java.util.ArrayList;
import java.util.List;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.lang.reflect.Method;
import java.util.*;

/**
 * 报告模板
 *
 * @author WWW
 * @date 2024-03-06
 */
@Service
@SuppressWarnings("ALL")
public class ReportService implements ReportMapper {
    @Autowired
    @Value("${sys.turfServer}")
    private String turfServer;

    @Resource
    ReportMapper reportMapper;

    @Override
    public Integer selectCount(String name) {
        name = StringHelper.getLikeStr(name);
    @Resource
    AttachService attachService;

        return reportMapper.selectCount(name);
    @Resource
    PathHelper pathHelper;

    @Resource
    DownloadService downloadService;

    @Resource
    RedisService redisService;

    @Resource
    DirService dirService;

    private final static Log log = LogFactory.getLog(ReportService.class);

    @Override
    public Integer selectCount(String name, String code) {
        name = StringHelper.getLikeUpperStr(name);

        return reportMapper.selectCount(name, code);
    }

    @Override
    public List<ReportEntity> selectByPage(String name, Integer limit, Integer offset) {
        name = StringHelper.getLikeStr(name);
    public List<ReportEntity> selectByPage(String name, String code, Integer limit, Integer offset) {
        name = StringHelper.getLikeUpperStr(name);

        return reportMapper.selectByPage(name, limit, offset);
        return reportMapper.selectByPage(name, code, limit, offset);
    }

    @Override
@@ -80,6 +117,11 @@
    }

    @Override
    public List<CountEntity> countSizesByType() {
        return reportMapper.countSizesByType();
    }

    @Override
    public List<CountEntity> countServices() {
        return reportMapper.countServices();
    }
@@ -89,69 +131,394 @@
        return reportMapper.countOperates();
    }

    /**
     * 创建 用户流量统计 Word
     */
    public void createCountSizesWord() {
        List<CountEntity> list = countOperates();
        if (null == list || list.isEmpty()) {
            return;
    @Override
    public List<CountEntity> countSizesByPrj() {
        return reportMapper.countSizesByPrj();
    }

    @Override
    public List<CountEntity> countExplorationPoints() {
        String key = RedisCacheKey.dataCountKey("countExplorationPoints");
        List<CountEntity> list = redisService.getListByKey(key);
        if (null == list) {
            list = reportMapper.countExplorationPoints();
            countExplorationArea(list);
            redisService.saveListByKey(key, list, StaticData.I2);
        }

        int rows = 1;
        String inputFile = "D:\\LF\\用户流量统计.docx";
        String outPutFile = "D:\\LF\\用户流量统计_new.docx";
        return list;
    }

        ArrayList<String[]> addList = new ArrayList<>();
        for (CountEntity ce : list) {
            String[] strs = new String[]{"" + rows++, ce.getM1(), ce.getM2(), ce.getCount().toString()};
            addList.add(strs);
        }
    @Override
    public List<CoordinateEntity> selectExplorationPoints(String code) {
        return reportMapper.selectExplorationPoints(StringHelper.getRightLike(code));
    }

        WordHelper.generateWord(inputFile, outPutFile, null, addList);
    @Override
    public List<CountEntity> countDemAreaByPrj() {
        return reportMapper.countDemAreaByPrj();
    }

    @Override
    public List<CountEntity> countMptAreaByPrj() {
        return reportMapper.countMptAreaByPrj();
    }

    @Override
    public List<CountEntity> countOsgbAreaByPrj() {
        return reportMapper.countOsgbAreaByPrj();
    }

    @Override
    public List<CountEntity> countLasAreaByPrj() {
        return reportMapper.countLasAreaByPrj();
    }

    @Override
    public List<CountEntity> countSurveyWorksiteByPrj() {
        return reportMapper.countSurveyWorksiteByPrj();
    }

    @Override
    public List<CountEntity> countExplorationReportByPrj() {
        return reportMapper.countExplorationReportByPrj();
    }

    @Override
    public List<CountEntity> countCollapseByPrj() {
        return reportMapper.countCollapseByPrj();
    }

    @Override
    public List<CountEntity> countDebrisFlowByPrj() {
        return reportMapper.countDebrisFlowByPrj();
    }

    @Override
    public List<CountEntity> countGroundCollapseByPrj() {
        return reportMapper.countGroundCollapseByPrj();
    }

    @Override
    public List<CountEntity> countHighSteepSlopeByPrj() {
        return reportMapper.countHighSteepSlopeByPrj();
    }

    @Override
    public List<CountEntity> countLandSlideByPrj() {
        return reportMapper.countLandSlideByPrj();
    }

    @Override
    public List<CountEntity> countUnstableSlopeByPrj() {
        return reportMapper.countUnstableSlopeByPrj();
    }

    @Override
    public List<CountEntity> countWaterDamageByPrj() {
        return reportMapper.countWaterDamageByPrj();
    }

    @Override
    public List<CountEntity> countDlgAreaByPrj() {
        return reportMapper.countDlgAreaByPrj();
    }

    @Override
    public List<CountEntity> countDomAreaByPrj() {
        return reportMapper.countDomAreaByPrj();
    }

    @Override
    public List<CountEntity> countLineLength() {
        return reportMapper.countLineLength();
    }

    @Override
    public List<CountEntity> countExplorationPointByPrj() {
        return reportMapper.countExplorationPointByPrj();
    }

    @Override
    public List<CountEntity> countGeoModelAreaByPrj() {
        return reportMapper.countGeoModelAreaByPrj();
    }

    @Override
    public List<CountEntity> countGeologicHazardByPrj() {
        return reportMapper.countGeologicHazardByPrj();
    }

    @Override
    public List<CountEntity> countVariousDataByPrj(String code) {
        code = StringUtils.isEmpty(code) || "ALL".equals(code) ? null : StringHelper.getRightLike(code);
        return reportMapper.countVariousDataByPrj(code);
    }

    /**
     * 创建 服务调用量统计 Word
     * 统计钻孔点面积
     */
    public void createCountServicesWord() {
        List<CountEntity> list = countServices();
        if (null == list || list.isEmpty()) {
    private void countExplorationArea(List<CountEntity> list) {
        if (null == list) {
            return;
        }

        int rows = 1;
        String inputFile = "D:\\LF\\服务调用量统计.docx";
        String outPutFile = "D:\\LF\\服务调用量统计_new.docx";

        ArrayList<String[]> addList = new ArrayList<>();
        for (CountEntity ce : list) {
            String[] strs = new String[]{"" + rows++, ce.getM1(), ce.getCount().toString()};
            addList.add(strs);
        }
            if (0 == ce.getCount()) {
                continue;
            }

        WordHelper.generateWord(inputFile, outPutFile, null, addList);
            double area = calcPolygonArea(ce.getM3());
            ce.setArea(area);
        }
    }

    /**
     * 创建 用户流量统计 Word
     * 计算多边形面积
     */
    public void createCountOperatesWord() {
        List<CountEntity> list = countOperates();
        if (null == list || list.isEmpty()) {
    private double calcPolygonArea(String code) {
        try {
            String url = turfServer + "/Call/CalcArea?code=" + code;
            String str = RestHelper.get(url);
            if (StringHelper.isEmpty(str)) {
                return 0d;
            }

            return Double.parseDouble(str);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            return 0d;
        }
    }

    /**
     * 调用JS方法 *
     */
    private double callJsFn(String str) {
        String staticPath = PathHelper.getStaticPath();
        Object obj = JsHelper.callJsFn(staticPath + "js/turf.min.6.5.js", "pointsToPolygon", str);
        if (null == obj) {
            return 0d;
        }

        return (double) obj;
    }

    /**
     * 创建报告
     */
    public void createReport(UserEntity ue, ReportEntity re, String code, HttpServletResponse res) throws Exception {
        AttachEntity ae = attachService.selectByGuid(re.getGuid());
        if (null == ae) {
            return;
        }

        int rows = 1;
        String inputFile = "D:\\LF\\服务调用量统计.docx";
        String outPutFile = "D:\\LF\\服务调用量统计_new.docx";
        String source = pathHelper.getConfig().getUploadPath() + File.separator + ae.getPath();
        String targetName = StringHelper.YMDHMS2_FORMAT.format(new Date()) + FileHelper.getExtension(ae.getName());
        String target = pathHelper.getDownloadFullPath() + File.separator + targetName;

        ArrayList<String[]> addList = new ArrayList<>();
        for (CountEntity ce : list) {
            String[] strs = new String[]{"" + rows++, ce.getM1(), ce.getCount().toString()};
            addList.add(strs);
        File sourceFile = new File(source);
        if (!sourceFile.exists() || sourceFile.isDirectory()) {
            return;
        }
        generateReport(source, target, re, code);

        File targetFile = new File(target);
        if (!targetFile.exists() || sourceFile.isDirectory()) {
            return;
        }

        WordHelper.generateWord(inputFile, outPutFile, null, addList);
        DownloadEntity de = getDownloadEntity(ue, target);
        int rows = downloadService.insert(de);
        if (rows > 0) {
            WebHelper.download(target, targetName, res);
        }
    }

    /**
     * 生成报告
     */
    private void generateReport(String source, String target, ReportEntity re, String code) {
        if (StaticData.S1.equals(re.getType())) {
            createWord(source, target, re, code);
        } else {
            createExcel(source, target, re, code);
        }
    }

    /**
     * 生成Word
     */
    private void createWord(String source, String target, ReportEntity re, String code) {
        List<CountEntity> list = selectListByMethodName(re, code);
        list = processListData(list);

        Map<String, String> textMap = new HashMap<>();
        if (!StringHelper.isEmpty(code)) {
            textMap.put("title", "ALL".equals(code) ? "所有项目" : dirService.selectNameByCode(code));
            list.remove(list.size() - 1);
        }

        List<String[]> tableList = new ArrayList<>();
        for (CountEntity ce : list) {
            switch (re.getCode()) {
                case "countSizesByType":
                case "countSizesByPrj":
                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), "" + ce.getCount(), ce.getM2()});
                    break;
                case "countSizes":
                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), ce.getM2()});
                    break;
                case "countOperates":
                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), ce.getM2(), "" + ce.getCount()});
                    break;
                case "countExplorationPoints":
                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), "" + ce.getCount(), ce.getM2(), ce.getM3()});
                    break;
                case "countDemAreaByPrj":
                case "countMptAreaByPrj":
                case "countOsgbAreaByPrj":
                case "countLasAreaByPrj":
                case "countDlgAreaByPrj":
                case "countDomAreaByPrj":
                case "countGeoModelAreaByPrj":
                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), ce.getM3()});
                    break;
                case "countLineLength":
                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), "" + ce.getLen()});
                    break;
                case "countVariousDataByPrj":
                    String str = "count".equals(ce.getM2()) ? ce.getCount() + "个" : ("area".equals(ce.getM2()) ? ce.getM3() : ce.getLen() + "千米");
                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), str});
                    break;
                default:
                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), "" + ce.getCount()});
                    break;
            }
        }

        WordHelper.generateWord(source, target, textMap, tableList);
    }

    /**
     * 生成Excel
     */
    private void createExcel(String source, String target, ReportEntity re, String code) {
        List<CountEntity> list = selectListByMethodName(re, code);
        list = processListData(list);

        Map<String, Object> map = new HashMap<>();
        if (!StringHelper.isEmpty(code)) {
            map.put("title", "ALL".equals(code) ? "所有项目" : dirService.selectNameByCode(code));
            list.remove(list.size() - 1);
            for (CountEntity ce : list) {
                ce.setM2("count".equals(ce.getM2()) ? ce.getCount() + "个" : ("area".equals(ce.getM2()) ? ce.getM3() : ce.getLen() + "千米"));
            }
        }

        Map<String, List<CountEntity>> listMap = new HashMap<>(1);
        listMap.put("data", list);

        ExcelHelper.writeToTemplate(source, target, map, listMap);
    }

    /**
     * 处理列表数据
     */
    private List<CountEntity> processListData(List<CountEntity> list) {
        if (null == list || list.isEmpty()) {
            return null;
        }

        int rows = 1;
        long allCount = 0;
        Double allSize = 0d, allArea = 0d, allLen = 0d;
        for (CountEntity ce : list) {
            ce.setNo(rows++);
            if (StringUtils.isEmpty(ce.getM2())) {
                ce.setM2(FileHelper.getSizes(ce.getSizes()));
            }
            ce.setM3(FileHelper.getSquareMeter(ce.getArea()));
            allCount += ce.getCount();
            allSize += ce.getSizes();
            allArea += ce.getArea();
            allLen += ce.getLen();
        }
        if (list.size() > 0) {
            CountEntity ce = new CountEntity();
            ce.setNo(rows);
            ce.setM1("总计");
            ce.setCount(allCount);
            if (StringUtils.isEmpty(list.get(0).getM2())) {
                ce.setM2(FileHelper.getSizes(allSize));
            }
            ce.setM3(FileHelper.getSquareMeter(allArea));
            ce.setLen(allLen);
            list.add(ce);
        }

        return list;
    }

    /**
     * 根据方法名称查询数据列表
     */
    private List<CountEntity> selectListByMethodName(ReportEntity re, String code) {
        List<CountEntity> list = null;
        try {
            Object obj;
            if (null != code) {
                Method method = ReportService.class.getDeclaredMethod(re.getCode(), String.class);
                obj = method.invoke(this, code);
            } else {
                Method method = ReportService.class.getDeclaredMethod(re.getCode());
                obj = method.invoke(this);
            }
            if (obj instanceof List<?>) {
                list = (List<CountEntity>) obj;
            }
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
        }

        if (null == list || list.isEmpty()) {
            return list;
        }
        for (CountEntity ce : list) {
            if (null == ce.getCount()) {
                ce.setCount(0L);
            }
            if (null == ce.getSizes()) {
                ce.setSizes(0d);
            }
            if (null == ce.getArea()) {
                ce.setArea(0d);
            }
            if (null == ce.getLen()) {
                ce.setLen(0d);
            }
        }

        return list;
    }

    /**
     * 获取下载实体类
     */
    private DownloadEntity getDownloadEntity(UserEntity ue, String file) {
        DownloadEntity de = new DownloadEntity();
        de.setName(FileHelper.getFileName(file));
        // 1-Shp文件，2-专题图，3-元数据，4-业务数据，5-管道分析，6-统计报告，7-附件文件，8-瓦片文件
        de.setType(6);
        de.setSizes(FileHelper.sizeToMb(new File(file).length()));
        de.setDepid(ue.getDepid());
        de.setDcount(1);
        de.setPwd(null);
        de.setUrl(FileHelper.getRelativePath(file));
        de.setDescr("统计报告");
        de.setGuid(FileHelper.getFileMd5(file));
        de.setCreateUser(ue.getId());
        // de.setGeom(null)

        return de;
    }
}