From ec7fa1df7b083e7b03ccb8c2dae9a03598f3c00f Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期四, 24 八月 2023 11:18:52 +0800
Subject: [PATCH] 修改MD架构的实体类

---
 src/main/java/com/lf/server/service/sys/ReportService.java |  149 ++++++++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 132 insertions(+), 17 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 df5813e..ced9c23 100644
--- a/src/main/java/com/lf/server/service/sys/ReportService.java
+++ b/src/main/java/com/lf/server/service/sys/ReportService.java
@@ -1,18 +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.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 javax.servlet.http.HttpServletResponse;
 import java.io.File;
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.List;
+import java.util.*;
 
 /**
  * 鎶ュ憡妯℃澘
@@ -29,18 +31,21 @@
     @Autowired
     PathHelper pathHelper;
 
-    @Override
-    public Integer selectCount(String name) {
-        name = StringHelper.getLikeStr(name);
+    @Autowired
+    DownloadService downloadService;
 
-        return reportMapper.selectCount(name);
+    @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
@@ -89,6 +94,11 @@
     }
 
     @Override
+    public List<CountEntity> countSizesByType() {
+        return reportMapper.countSizesByType();
+    }
+
+    @Override
     public List<CountEntity> countServices() {
         return reportMapper.countServices();
     }
@@ -98,10 +108,15 @@
         return reportMapper.countOperates();
     }
 
+    @Override
+    public List<CountEntity> countSizesByPrj() {
+        return reportMapper.countSizesByPrj();
+    }
+
     /**
      * 鍒涘缓鎶ュ憡
      */
-    public void createReport(ReportEntity re, HttpServletResponse res) throws Exception {
+    public void createReport(UserEntity ue, ReportEntity re, HttpServletResponse res) throws Exception {
         AttachEntity ae = attachService.selectByGuid(re.getGuid());
         if (null == ae) {
             return;
@@ -109,7 +124,7 @@
 
         String source = pathHelper.getConfig().getUploadPath() + File.separator + ae.getPath();
         String targetName = StringHelper.YMDHMS2_FORMAT.format(new Date()) + FileHelper.getExtension(ae.getName());
-        String target = pathHelper.getShareFullPath() + File.separator + targetName;
+        String target = pathHelper.getDownloadFullPath() + File.separator + targetName;
 
         File sourceFile = new File(source);
         if (!sourceFile.exists() || sourceFile.isDirectory()) {
@@ -122,27 +137,45 @@
             return;
         }
 
-        WebHelper.download(target, targetName, res);
+        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 ("1".equals(re.getType())) {
+        if (StaticData.S1.equals(re.getType())) {
             switch (re.getCode()) {
-                case "countOperates ":
+                case "countOperates":
                     createCountOperatesWord(source, target);
                     break;
                 case "countSizes":
                     createCountSizesWord(source, target);
                     break;
-                default:
+                case "countServices":
                     createCountServicesWord(source, target);
+                    break;
+                default:
                     break;
             }
         } else {
-            //
+            switch (re.getCode()) {
+                case "countOperates":
+                    createCountOperatesExcel(source, target);
+                    break;
+                case "countSizes":
+                    createCountSizesExcel(source, target);
+                    break;
+                case "countServices":
+                    createCountServicesExcel(source, target);
+                    break;
+                default:
+                    break;
+            }
         }
     }
 
@@ -202,4 +235,86 @@
 
         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-缁熻鎶ュ憡锛�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;
+    }
 }

--
Gitblit v1.9.3