From cb9fbcd27a288d0c61d85fa13ff5fc8eb1f4deab Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期六, 18 三月 2023 16:44:56 +0800 Subject: [PATCH] 1 --- src/main/java/com/lf/server/service/sys/ReportService.java | 222 ++++++++++++++++++++++++++++++++++++++++++++++++++++--- 1 files changed, 208 insertions(+), 14 deletions(-) diff --git a/src/main/java/com/lf/server/service/sys/ReportService.java b/src/main/java/com/lf/server/service/sys/ReportService.java index fcce48c..8e0bfaa 100644 --- a/src/main/java/com/lf/server/service/sys/ReportService.java +++ b/src/main/java/com/lf/server/service/sys/ReportService.java @@ -1,15 +1,20 @@ package com.lf.server.service.sys; +import com.lf.server.entity.all.StaticData; 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 com.lf.server.service.data.DownloadService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; -import java.util.ArrayList; -import java.util.List; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.util.*; /** * 鎶ュ憡妯℃澘 @@ -20,18 +25,27 @@ @Autowired ReportMapper reportMapper; + @Autowired + AttachService attachService; + + @Autowired + PathHelper pathHelper; + + @Autowired + DownloadService downloadService; + @Override - public Integer selectCount(String name) { + public Integer selectCount(String name, String code) { name = StringHelper.getLikeStr(name); - return reportMapper.selectCount(name); + return reportMapper.selectCount(name, code); } @Override - public List<ReportEntity> selectByPage(String name, Integer limit, Integer offset) { + public List<ReportEntity> selectByPage(String name, String code, Integer limit, Integer offset) { name = StringHelper.getLikeStr(name); - return reportMapper.selectByPage(name, limit, offset); + return reportMapper.selectByPage(name, code, limit, offset); } @Override @@ -90,23 +104,203 @@ } /** + * 鍒涘缓鎶ュ憡 + */ + 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())) { + switch (re.getCode()) { + case "countOperates": + createCountOperatesWord(source, target); + break; + case "countSizes": + createCountSizesWord(source, target); + break; + default: + createCountServicesWord(source, target); + break; + } + } else { + switch (re.getCode()) { + case "countOperates": + createCountOperatesExcel(source, target); + break; + case "countSizes": + createCountSizesExcel(source, target); + break; + default: + createCountServicesExcel(source, target); + break; + } + } + } + + /** * 鍒涘缓 鐢ㄦ埛娴侀噺缁熻 Word */ - public void createCountSizesWord() { + public void createCountOperatesWord(String source, String target) { List<CountEntity> list = countOperates(); if (null == list || list.isEmpty()) { return; } - String inputFile = "D:\\LF\\鐢ㄦ埛娴侀噺缁熻.docx"; - String outPutFile = "D:\\LF\\鐢ㄦ埛娴侀噺缁熻_new.docx"; - + int rows = 1; ArrayList<String[]> addList = new ArrayList<>(); for (CountEntity ce : list) { - String[] strs = new String[]{ce.getM1(), ce.getM2(), ce.getCount().toString()}; + String[] strs = new String[]{"" + rows++, ce.getM1(), ce.getM2(), ce.getCount().toString()}; addList.add(strs); } - WordHelper.generateWord(inputFile, outPutFile, null, addList); + 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; + ArrayList<String[]> addList = new ArrayList<>(); + for (CountEntity ce : list) { + String[] strs = new String[]{"" + rows++, ce.getM1(), ce.getCount().toString()}; + addList.add(strs); + } + + 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; + ArrayList<String[]> addList = new ArrayList<>(); + for (CountEntity ce : list) { + String[] strs = new String[]{"" + rows++, ce.getM1(), FileHelper.getSizes(ce.getSizes())}; + addList.add(strs); + } + + WordHelper.generateWord(source, target, null, addList); + } + + /** + * 鍒涘缓 鐢ㄦ埛娴侀噺缁熻 Excel + */ + public void createCountOperatesExcel(String source, String target) { + List<CountEntity> list = countOperates(); + if (null == list || list.isEmpty()) { + return; + } + + int rows = 1; + for (CountEntity ce : list) { + ce.setNo(rows++); + } + + Map<String, List<CountEntity>> map = new HashMap<>(1); + map.put("data", list); + + ExcelHelper.writeToTemplate(source, target, map); + } + + /** + * 鍒涘缓 鏈嶅姟璋冪敤閲忕粺璁� Excel + */ + public void createCountServicesExcel(String source, String target) { + List<CountEntity> list = countServices(); + if (null == list || list.isEmpty()) { + return; + } + + int rows = 1; + for (CountEntity ce : list) { + ce.setNo(rows++); + } + + Map<String, List<CountEntity>> map = new HashMap<>(1); + map.put("data", list); + + ExcelHelper.writeToTemplate(source, target, map); + } + + /** + * 鍒涘缓 鏁版嵁閲忕粺璁� Excel + */ + public void createCountSizesExcel(String source, String target) { + List<CountEntity> list = countSizes(); + if (null == list || list.isEmpty()) { + return; + } + + int rows = 1; + for (CountEntity ce : list) { + ce.setNo(rows++); + ce.setM2(FileHelper.getSizes(ce.getSizes())); + } + + 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-缁熻鎶ュ憡 + 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; } } -- Gitblit v1.9.3