修改生成Excel功能

13693261870

2024-03-06 58f087ab3989a72eaab14307eccd68b7f5124427

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

@@ -1,37 +1,67 @@
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.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 java.util.ArrayList;
import java.util.List;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.util.*;

/**
 * 报告模板
 *
 * @author WWW
 * @date 2024-03-06
 */
@Service
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;

    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 +110,11 @@
    }

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

    @Override
    public List<CountEntity> countServices() {
        return reportMapper.countServices();
    }
@@ -89,25 +124,390 @@
        return reportMapper.countOperates();
    }

    @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);
        }

        return list;
    }

    @Override
    public List<CoordinateEntity> selectExplorationPoints(String code) {
        return reportMapper.selectExplorationPoints(StringHelper.getRightLike(code));
    }

    @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();
    }

    /**
     * 统计钻孔点面积
     */
    private void countExplorationArea(List<CountEntity> list) {
        if (null == list) {
            return;
        }
        for (CountEntity ce : list) {
            if (0 == ce.getCount()) {
                continue;
            }

            double area = calcPolygonArea(ce.getM3());
            ce.setArea(area);
        }
    }

    /**
     * 计算多边形面积
     */
    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, HttpServletResponse res) throws Exception {
        AttachEntity ae = attachService.selectByGuid(re.getGuid());
        if (null == ae) {
            return;
        }

        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;

        File sourceFile = new File(source);
        if (!sourceFile.exists() || sourceFile.isDirectory()) {
            return;
        }
        generateReport(source, target, re);

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

        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) {
        if (StaticData.S1.equals(re.getType())) {
            createWord(source, target, re);
        } else {
            createExcel(source, target, re);
        }
    }

    /**
     * 生成Word
     */
    private void createWord(String source, String target, ReportEntity re) {
        //
    }

    /**
     * 生成Excel
     */
    private void createExcel(String source, String target, ReportEntity re) {
        List<CountEntity> list = countExplorationPoints();//
        if (null == list || list.isEmpty()) {
            return;
        }

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

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

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

    /**
     * 获取下载实体类
     */
    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;
    }

    /**
     * 创建 用户流量统计 Word
     */
    public void createCountSizesWord() {
    public void createCountOperatesWord(String source, String target) {
        List<CountEntity> list = countOperates();
        if (null == list || list.isEmpty()) {
            return;
        }

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

        int rows = 1, allCount = 0;
        ArrayList<String[]> addList = new ArrayList<>();
        for (CountEntity ce : list) {
            allCount += ce.getCount();
            String[] strs = new String[]{"" + rows++, ce.getM1(), ce.getM2(), ce.getCount().toString()};
            addList.add(strs);
        }
        if (addList.size() > 0) {
            addList.add(new String[]{"" + rows, "总计", " ", "" + allCount});
        }

        WordHelper.generateWord(inputFile, outPutFile, null, addList);
        WordHelper.generateWord(source, target, null, addList);
    }

    /**
     * 创建 数据量统计 Word
     */
    public void createCountSizesWord(String source, String target) {
        List<CountEntity> list = countSizes();
        if (null == list || list.isEmpty()) {
            return;
        }

        int rows = 1;
        Double allSize = 0d;
        ArrayList<String[]> addList = new ArrayList<>();
        for (CountEntity ce : list) {
            allSize += ce.getSizes();
            String[] strs = new String[]{"" + rows++, ce.getM1(), FileHelper.getSizes(ce.getSizes())};
            addList.add(strs);
        }
        if (addList.size() > 0) {
            addList.add(new String[]{"" + rows, "总计", FileHelper.getSizes(allSize)});
        }

        WordHelper.generateWord(source, target, null, addList);
    }

    /**
     * 创建 服务调用量统计 Word
     */
    public void createCountServicesWord(String source, String target) {
        List<CountEntity> list = countServices();
        if (null == list || list.isEmpty()) {
            return;
        }

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

        WordHelper.generateWord(source, target, null, addList);
    }

    /**
     * 创建 钻孔数据统计 Word
     */
    public void createCountExplorationPointsWord(String source, String target) {
        List<CountEntity> list = countExplorationPoints();
        if (null == list || list.isEmpty()) {
            return;
        }

        int rows = 1, allCount = 0;
        Double allSize = 0d, allArea = 0d;
        ArrayList<String[]> addList = new ArrayList<>();
        for (CountEntity ce : list) {
            allSize += ce.getSizes();
            allCount += ce.getCount();
            allArea += ce.getArea();
            String[] strs = new String[]{"" + rows++, ce.getM1(), "" + ce.getCount(), FileHelper.getSizes(ce.getSizes()), FileHelper.getSquareMeter(ce.getArea())};
            addList.add(strs);
        }
        if (addList.size() > 0) {
            addList.add(new String[]{"" + rows, "总计", "" + allCount, FileHelper.getSizes(allSize), FileHelper.getSquareMeter(allArea)});
        }

        WordHelper.generateWord(source, target, null, addList);
    }

    /**
     * 创建 数据格式统计 Word
     */
    public void createCountSizesByTypeWord(String source, String target) {
        List<CountEntity> list = countSizesByType();
        if (null == list || list.isEmpty()) {
            return;
        }

        int rows = 1, allCount = 0;
        Double allSize = 0d;
        ArrayList<String[]> addList = new ArrayList<>();
        for (CountEntity ce : list) {
            allSize += ce.getSizes();
            allCount += ce.getCount();
            String[] strs = new String[]{"" + rows++, ce.getM1(), "" + ce.getCount(), FileHelper.getSizes(ce.getSizes())};
            addList.add(strs);
        }
        if (addList.size() > 0) {
            addList.add(new String[]{"" + rows, "总计", "" + allCount, FileHelper.getSizes(allSize)});
        }

        WordHelper.generateWord(source, target, null, addList);
    }

    /**
     * 创建 项目数据统计 Word
     */
    public void createCountSizesByPrjWord(String source, String target) {
        List<CountEntity> list = countSizesByPrj();
        if (null == list || list.isEmpty()) {
            return;
        }

        int rows = 1, allCount = 0;
        Double allSize = 0d;
        ArrayList<String[]> addList = new ArrayList<>();
        for (CountEntity ce : list) {
            allSize += ce.getSizes();
            allCount += ce.getCount();
            String[] strs = new String[]{"" + rows++, ce.getM1(), "" + ce.getCount(), FileHelper.getSizes(ce.getSizes())};
            addList.add(strs);
        }
        if (addList.size() > 0) {
            addList.add(new String[]{"" + rows, "总计", "" + allCount, FileHelper.getSizes(allSize)});
        }

        WordHelper.generateWord(source, target, null, addList);
    }
}