From 591f744651c3f14a01ea7685899e47b9df3c2ad3 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期五, 03 二月 2023 18:19:19 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/config/InitConfig.java               |    2 ++
 src/main/java/com/lf/server/service/all/UploadAttachService.java |   27 +++++++++++++++++++++++++++
 src/main/resources/application.yml                               |    2 ++
 3 files changed, 31 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/lf/server/config/InitConfig.java b/src/main/java/com/lf/server/config/InitConfig.java
index 4f24dde..a7b8c77 100644
--- a/src/main/java/com/lf/server/config/InitConfig.java
+++ b/src/main/java/com/lf/server/config/InitConfig.java
@@ -13,6 +13,7 @@
 import com.lf.server.mapper.bd.DlgagnpMapper;
 import com.lf.server.service.all.BaseQueryService;
 import com.lf.server.service.all.PermsService;
+import com.lf.server.service.all.UploadAttachService;
 import com.lf.server.service.show.LocateService;
 import com.lf.server.service.sys.ArgsService;
 import com.lf.server.service.sys.BlacklistService;
@@ -90,6 +91,7 @@
 
             // 鍒濆鍖�
             GdalHelper.init(env.getProperty("sys.gdal_path"));
+            UploadAttachService.init(env.getProperty("sys.attachTabs"));
             pathHelper.init();
             argsService.initSettingData();
 
diff --git a/src/main/java/com/lf/server/service/all/UploadAttachService.java b/src/main/java/com/lf/server/service/all/UploadAttachService.java
index 5f6dd56..53590fe 100644
--- a/src/main/java/com/lf/server/service/all/UploadAttachService.java
+++ b/src/main/java/com/lf/server/service/all/UploadAttachService.java
@@ -5,6 +5,7 @@
 import com.lf.server.entity.all.HttpStatus;
 import com.lf.server.entity.all.ResponseMsg;
 import com.lf.server.entity.all.SettingData;
+import com.lf.server.entity.all.StaticData;
 import com.lf.server.entity.sys.AttachEntity;
 import com.lf.server.entity.sys.UserEntity;
 import com.lf.server.helper.FileHelper;
@@ -20,6 +21,8 @@
 
 import javax.servlet.http.HttpServletResponse;
 import java.io.File;
+import java.util.HashMap;
+import java.util.Map;
 
 /**
  * 涓婁紶闄勪欢鏈嶅姟绫�
@@ -33,11 +36,35 @@
     @Autowired
     AttachService attachService;
 
+    private static Map<String, String> attachTabs = new HashMap<>();
+
     private final static Log log = LogFactory.getLog(UploadAttachService.class);
 
     private static final String NO_FILE = JSON.toJSONString(new ResponseMsg<String>(HttpStatus.NOT_FOUND, "鏂囦欢鎵句笉鍒�"));
 
     /**
+     * 鍒濆鍖栭檮浠惰〃
+     */
+    public static void init(String cfg) {
+        if (StringHelper.isEmpty(cfg)) {
+            return;
+        }
+
+        String[] strs = cfg.split(StaticData.COMMA);
+        if (strs.length == 0) {
+            return;
+        }
+
+        for (String str : strs) {
+            if (attachTabs.containsKey(str)) {
+                continue;
+            }
+
+            attachTabs.put(str, str.contains("bd.") ? "materiname" : "photono");
+        }
+    }
+
+    /**
      * 涓婁紶鏂囦欢
      */
     public ResponseMsg<String> upload(UserEntity ue, String tab, MultipartFile file, BaseController ctrl) {
diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml
index 3347818..46e5680 100644
--- a/src/main/resources/application.yml
+++ b/src/main/resources/application.yml
@@ -162,3 +162,5 @@
     publishMpt: D:\LF\publishMpt
     # 鍙戝竷3DML鐩綍
     publish3dml: D:\LF\publish3dml
+  # 涓婁紶闄勪欢琛�
+  attachTabs: bd.b_pac_hydrogeology,bd.b_pac_frozensoil,bd.b_pac_geologic_hazard,bd.b_pac_marine_meteorological,bd.b_pac_meteorological,bs.m_equipment_nameplate,bs.m_hydraulic_protection,bs.m_marker,bs.s_borehole
\ No newline at end of file

--
Gitblit v1.9.3