From 4d8dd71b7d13fa41132c46831090c6282af502cb Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期六, 12 十一月 2022 09:03:08 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/service/show/ExportService.java |   64 ++++++++++++++++++++++++++++----
 1 files changed, 56 insertions(+), 8 deletions(-)

diff --git a/src/main/java/com/lf/server/service/show/ExportService.java b/src/main/java/com/lf/server/service/show/ExportService.java
index 2132174..400573f 100644
--- a/src/main/java/com/lf/server/service/show/ExportService.java
+++ b/src/main/java/com/lf/server/service/show/ExportService.java
@@ -2,18 +2,21 @@
 
 import com.alibaba.fastjson.JSON;
 import com.lf.server.entity.all.ResponseMsg;
+import com.lf.server.entity.data.DownloadEntity;
 import com.lf.server.entity.show.ExportEntity;
+import com.lf.server.entity.sys.UserEntity;
+import com.lf.server.helper.FileHelper;
+import com.lf.server.helper.PathHelper;
 import com.lf.server.helper.RestHelper;
 import com.lf.server.helper.StringHelper;
-import org.apache.http.NameValuePair;
-import org.apache.http.message.BasicNameValuePair;
+import com.lf.server.service.data.DownloadService;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 
+import java.io.File;
 import java.lang.reflect.Field;
-import java.util.ArrayList;
 import java.util.HashMap;
-import java.util.List;
 import java.util.Map;
 
 /**
@@ -25,8 +28,22 @@
     @Value("${sys.path.exportServer}")
     private String exportServer;
 
-    public String post(ExportEntity entity) {
-        Map<String, Object> map = getPostEntity(entity);
+    @Autowired
+    PathHelper pathHelper;
+
+    @Autowired
+    DownloadService downloadService;
+
+    /**
+     * POST璇锋眰鍑哄浘鏈嶅姟
+     *
+     * @param ue     鐢ㄦ埛绫�
+     * @param entity 鍦ㄧ嚎鍒跺浘绫�
+     * @return 鎴愬姛
+     * @throws Exception
+     */
+    public String post(UserEntity ue, ExportEntity entity) throws Exception {
+        Map<String, Object> map = getMapData(entity);
         String url = exportServer + "/Export/Start";
 
         String str = RestHelper.postForRest(url, map);
@@ -39,10 +56,22 @@
             return null;
         }
 
-        return msg.getResult();
+        String file = pathHelper.getConfig().getDownloadPath() + File.separator + msg.getResult();
+        File f = new File(file);
+        if (!f.exists() && !f.isDirectory()) {
+            return null;
+        }
+
+        DownloadEntity de = getDownloadEntity(ue, file);
+        int rows = downloadService.insert(de);
+
+        return rows > 0 ? de.getGuid() : null;
     }
 
-    private Map<String, Object> getPostEntity(ExportEntity entity) {
+    /**
+     * 鑾峰彇Map鏁版嵁
+     */
+    private Map<String, Object> getMapData(ExportEntity entity) {
         Map<String, Object> map = new HashMap<String, Object>();
 
         Field[] fields = entity.getClass().getDeclaredFields();
@@ -63,4 +92,23 @@
 
         return map;
     }
+
+    /**
+     * 鑾峰彇涓嬭浇瀹炰綋绫�
+     */
+    private DownloadEntity getDownloadEntity(UserEntity ue, String file) throws Exception {
+        DownloadEntity de = new DownloadEntity();
+        de.setName(FileHelper.getFileName(file));
+        de.setType(2);
+        de.setDepid(ue.getDepid());
+        de.setDcount(0);
+        // 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