From b20fd29dc3864405af4afa0aeb99d13d74fbfcdc Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期四, 21 三月 2024 15:07:44 +0800
Subject: [PATCH] 解决导出xls异常

---
 src/main/java/com/lf/server/service/sys/ReportService.java |  137 +++++++++++++++++++++++++++++++++++----------
 1 files changed, 106 insertions(+), 31 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 c897a98..376bd4c 100644
--- a/src/main/java/com/lf/server/service/sys/ReportService.java
+++ b/src/main/java/com/lf/server/service/sys/ReportService.java
@@ -11,11 +11,13 @@
 import com.lf.server.helper.*;
 import com.lf.server.mapper.sys.ReportMapper;
 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 javax.annotation.Resource;
 import javax.servlet.http.HttpServletResponse;
@@ -49,6 +51,9 @@
 
     @Resource
     RedisService redisService;
+
+    @Resource
+    DirService dirService;
 
     private final static Log log = LogFactory.getLog(ReportService.class);
 
@@ -214,6 +219,42 @@
         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 = StringHelper.getRightLike(code);
+        return reportMapper.countVariousDataByPrj(code);
+    }
+
     /**
      * 缁熻閽诲瓟鐐归潰绉�
      */
@@ -265,7 +306,7 @@
     /**
      * 鍒涘缓鎶ュ憡
      */
-    public void createReport(UserEntity ue, ReportEntity re, HttpServletResponse res) throws Exception {
+    public void createReport(UserEntity ue, ReportEntity re, String code, HttpServletResponse res) throws Exception {
         AttachEntity ae = attachService.selectByGuid(re.getGuid());
         if (null == ae) {
             return;
@@ -279,7 +320,7 @@
         if (!sourceFile.exists() || sourceFile.isDirectory()) {
             return;
         }
-        generateReport(source, target, re);
+        generateReport(source, target, re, code);
 
         File targetFile = new File(target);
         if (!targetFile.exists() || sourceFile.isDirectory()) {
@@ -296,66 +337,88 @@
     /**
      * 鐢熸垚鎶ュ憡
      */
-    private void generateReport(String source, String target, ReportEntity re) {
+    private void generateReport(String source, String target, ReportEntity re, String code) {
         if (StaticData.S1.equals(re.getType())) {
-            createWord(source, target, re);
+            createWord(source, target, re, code);
         } else {
-            createExcel(source, target, re);
+            createExcel(source, target, re, code);
         }
     }
 
     /**
      * 鐢熸垚Word
      */
-    private void createWord(String source, String target, ReportEntity re) {
-        List<CountEntity> list = selectListByMethodName(re);
+    private void createWord(String source, String target, ReportEntity re, String code) {
+        List<CountEntity> list = selectListByMethodName(re, code);
         list = processListData(list);
 
-        Map<String, List<CountEntity>> map = new HashMap<>(1);
-        map.put("data", list);
+        Map<String, String> textMap = new HashMap<>();
+        if (!StringHelper.isEmpty(code)) {
+            textMap.put("title", dirService.selectNameByCode(code));
+            list.remove(list.size() - 1);
+        }
 
-        List<String[]> addList = new ArrayList<>();
+        List<String[]> tableList = new ArrayList<>();
         for (CountEntity ce : list) {
             switch (re.getCode()) {
                 case "countSizesByType":
                 case "countSizesByPrj":
-                    addList.add(new String[]{"" + ce.getNo(), ce.getM1(), "" + ce.getCount(), ce.getM2()});
+                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), "" + ce.getCount(), ce.getM2()});
                     break;
                 case "countSizes":
-                    addList.add(new String[]{"" + ce.getNo(), ce.getM1(), ce.getM2()});
+                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), ce.getM2()});
                     break;
                 case "countOperates":
-                    addList.add(new String[]{"" + ce.getNo(), ce.getM1(), ce.getM2(), "" + ce.getCount()});
+                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), ce.getM2(), "" + ce.getCount()});
                     break;
                 case "countExplorationPoints":
-                    addList.add(new String[]{"" + ce.getNo(), ce.getM1(), "" + ce.getCount(), ce.getM2(), ce.getM3()});
+                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), "" + ce.getCount(), ce.getM2(), ce.getM3()});
                     break;
                 case "countDemAreaByPrj":
                 case "countMptAreaByPrj":
                 case "countOsgbAreaByPrj":
                 case "countLasAreaByPrj":
-                    addList.add(new String[]{"" + ce.getNo(), ce.getM1(), ce.getM3()});
+                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:
-                    addList.add(new String[]{"" + ce.getNo(), ce.getM1(), "" + ce.getCount()});
+                    tableList.add(new String[]{"" + ce.getNo(), ce.getM1(), "" + ce.getCount()});
                     break;
             }
         }
 
-        WordHelper.generateWord(source, target, null, addList);
+        WordHelper.generateWord(source, target, textMap, tableList);
     }
 
     /**
      * 鐢熸垚Excel
      */
-    private void createExcel(String source, String target, ReportEntity re) {
-        List<CountEntity> list = selectListByMethodName(re);
+    private void createExcel(String source, String target, ReportEntity re, String code) {
+        List<CountEntity> list = selectListByMethodName(re, code);
         list = processListData(list);
 
-        Map<String, List<CountEntity>> map = new HashMap<>(1);
-        map.put("data", list);
+        Map<String, Object> map = new HashMap<>();
+        if (!StringHelper.isEmpty(code)) {
+            map.put("title", 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() + "鍗冪背"));
+            }
+        }
 
-        ExcelHelper.writeToTemplate(source, target, map);
+        Map<String, List<CountEntity>> listMap = new HashMap<>(1);
+        listMap.put("data", list);
+
+        ExcelHelper.writeToTemplate(source, target, map, listMap);
     }
 
     /**
@@ -368,22 +431,28 @@
 
         int rows = 1;
         long allCount = 0;
-        Double allSize = 0d, allArea = 0d;
+        Double allSize = 0d, allArea = 0d, allLen = 0d;
         for (CountEntity ce : list) {
             ce.setNo(rows++);
-            ce.setM2(FileHelper.getSizes(ce.getSizes()));
+            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);
-            ce.setM2(FileHelper.getSizes(allSize));
+            if (StringUtils.isEmpty(list.get(0).getM2())) {
+                ce.setM2(FileHelper.getSizes(allSize));
+            }
             ce.setM3(FileHelper.getSquareMeter(allArea));
+            ce.setLen(allLen);
             list.add(ce);
         }
 
@@ -392,15 +461,18 @@
 
     /**
      * 鏍规嵁鏂规硶鍚嶇О鏌ヨ鏁版嵁鍒楄〃
-     *
-     * @param re
-     * @return
      */
-    private List<CountEntity> selectListByMethodName(ReportEntity re) {
+    private List<CountEntity> selectListByMethodName(ReportEntity re, String code) {
         List<CountEntity> list = null;
         try {
-            Method method = ReportService.class.getDeclaredMethod(re.getCode(), ReportService.class);
-            Object obj = method.invoke(this);
+            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;
             }
@@ -421,6 +493,9 @@
             if (null == ce.getArea()) {
                 ce.setArea(0d);
             }
+            if (null == ce.getLen()) {
+                ce.setLen(0d);
+            }
         }
 
         return list;

--
Gitblit v1.9.3