From ed8c7a5effd0d423ce1118b680ecdca6fe732609 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 02 七月 2025 16:43:13 +0800
Subject: [PATCH] Merge branch 'master' of http://192.168.11.205:9000/r/P2022036_Service

---
 src/main/java/com/lf/server/service/show/DataLibService.java |  218 ++++++++++++++++++++++++++++++------------------------
 1 files changed, 121 insertions(+), 97 deletions(-)

diff --git a/src/main/java/com/lf/server/service/show/DataLibService.java b/src/main/java/com/lf/server/service/show/DataLibService.java
index 0bdeb1c..5abe6ac 100644
--- a/src/main/java/com/lf/server/service/show/DataLibService.java
+++ b/src/main/java/com/lf/server/service/show/DataLibService.java
@@ -1,8 +1,8 @@
 package com.lf.server.service.show;
 
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
-import com.lf.server.entity.all.BaseEntity;
 import com.lf.server.entity.ctrl.DownloadReqEntity;
+import com.lf.server.entity.ctrl.KeyValueEntity;
 import com.lf.server.entity.data.DownloadEntity;
 import com.lf.server.entity.sys.AttachEntity;
 import com.lf.server.entity.sys.UserEntity;
@@ -11,16 +11,16 @@
 import com.lf.server.mapper.all.GeomBaseMapper;
 import com.lf.server.mapper.data.DownloadMapper;
 import com.lf.server.service.all.BaseQueryService;
-import com.lf.server.service.data.DownloadService;
+import com.lf.server.service.data.MetaService;
 import net.lingala.zip4j.ZipFile;
 import net.lingala.zip4j.model.ZipParameters;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.util.StringUtils;
 
 import java.io.File;
-import java.io.FileInputStream;
 import java.util.*;
 
 /**
@@ -28,15 +28,16 @@
  * @author WWW
  */
 @Service
+@SuppressWarnings("ALL")
 public class DataLibService {
     @Autowired
     PathHelper pathHelper;
 
     @Autowired
-    DownloadMapper downloadMapper;
+    MetaService metaService;
 
     @Autowired
-    DownloadService downloadService;
+    DownloadMapper downloadMapper;
 
     @Autowired
     BaseQueryService baseQueryService;
@@ -67,9 +68,7 @@
 
             QueryWrapper wrapper = getWrapper4DbOverflow(ue, dr);
             Integer srid = baseQueryService.getSrid(baseMapper);
-            if (null != srid) {
-                wrapper.apply(String.format("ST_Intersects(ST_PolygonFromText('%s', %d), geom)", dr.getWkt(), srid));
-            }
+            wrapper.apply(String.format("ST_Intersects(ST_PolygonFromText('%s', %d), geom)", dr.getWkt(), srid));
 
             List<String> ids = baseMapper.selectObjs(wrapper);
             addDepCodes(rs, ids);
@@ -165,13 +164,13 @@
         ZipParameters params = Zip4jHelper.getZipParams(true);
         zip.addFolder(new File(gdbPath), params);
         // zip.addFolder(new File(annexPath), params)
-        addAnnex(zip, params, annexMap);
+        metaService.addAnnex(zip, params, annexMap);
 
         String dbPwd = Md5Helper.reverse(Md5Helper.generate(dr.getPwd()));
-        DownloadEntity downloadEntity = getDownloadEntity(ue, zipFile, dbPwd);
-        int rows = downloadMapper.insert(downloadEntity);
+        DownloadEntity de = getDownloadEntity(ue, zipFile, dbPwd);
+        int rows = downloadMapper.insert(de);
 
-        return rows > 0 ? downloadEntity.getGuid() : null;
+        return rows > 0 ? de.getGuid() : null;
     }
 
     /**
@@ -186,34 +185,7 @@
                 }
 
                 QueryWrapper wrapper = createQueryWrapper(baseMapper, dr);
-                List list = baseMapper.selectList(wrapper);
-                if (null == list || list.size() == 0) {
-                    continue;
-                }
-
-                if (!dataMap.containsKey(entity)) {
-                    dataMap.put(entity, list);
-                } else {
-                    dataMap.get(entity).addAll(list);
-                }
-                if (wrapper.isEmptyOfWhere()) {
-                    wrapper.apply("1 = 1");
-                }
-                if ("bborehole".equals(entity)) {
-                    wrapper.last("limit 100");
-                }
-
-                String tab = BaseQueryService.getTabName(baseMapper);
-                List<AttachEntity> annex = baseMapper.selectAnnex(tab, wrapper);
-                if (null == annex || annex.isEmpty()) {
-                    continue;
-                }
-
-                if (!annexMap.containsKey(entity)) {
-                    annexMap.put(tab.replace(".", "_"), annex);
-                } else {
-                    annexMap.get(tab.replace(".", "_")).addAll(annex);
-                }
+                metaService.addData(entity, baseMapper, wrapper, dataMap, annexMap);
             } catch (Exception ex) {
                 log.error(ex.getMessage(), ex);
             }
@@ -234,10 +206,10 @@
             // wrapper.apply(String.format("depid like '%s'", StringHelper.getRightLike(dr.getDepcode())))
             wrapper.likeRight("depid", dr.getDepcode());
         }
-        if (baseMapper instanceof GeomBaseMapper && !StringHelper.isEmpty(dr.getWkt())) {
+        if (baseMapper instanceof GeomBaseMapper) {
             wrapper.select("ST_AsText(geom) as geom, *");
-            Integer srid = baseQueryService.getSrid((GeomBaseMapper) baseMapper);
-            if (null != srid) {
+            if (!StringHelper.isEmpty(dr.getWkt())) {
+                Integer srid = baseQueryService.getSrid((GeomBaseMapper) baseMapper);
                 wrapper.apply(String.format("ST_Intersects(ST_PolygonFromText('%s', %d), geom)", dr.getWkt(), srid));
             }
         }
@@ -287,39 +259,12 @@
     }
 
     /**
-     * 娣诲姞闄勪欢
-     */
-    private void addAnnex(ZipFile zip, ZipParameters params, Map<String, List<AttachEntity>> annexMap) {
-        List<String> files = new ArrayList<>();
-        String uploadPath = pathHelper.getConfig().getUploadPath();
-        for (String key : annexMap.keySet()) {
-            for (AttachEntity ae : annexMap.get(key)) {
-                try {
-                    File srcFile = new File(uploadPath + File.separator + ae.getPath());
-                    if (!srcFile.exists() || srcFile.isDirectory()) {
-                        continue;
-                    }
-                    if (files.contains(srcFile.getPath())) {
-                        continue;
-                    }
-
-                    files.add(srcFile.getPath());
-                    params.setFileNameInZip("annex" + File.separator + key + File.separator + ae.getName());
-                    zip.addStream(new FileInputStream(srcFile), params);
-                } catch (Exception ex) {
-                    log.error(ex.getMessage(), ex);
-                }
-            }
-        }
-    }
-
-    /**
      * 鑾峰彇涓嬭浇瀹炰綋绫�
      */
     private DownloadEntity getDownloadEntity(UserEntity ue, String file, String pwd) {
         DownloadEntity de = new DownloadEntity();
         de.setName(FileHelper.getFileName(file));
-        // 1-Shp鏂囦欢锛�2-涓撻鍥撅紝3-鍏冩暟鎹紝4-涓氬姟鏁版嵁锛�5-绠¢亾鍒嗘瀽锛�6-缁熻鎶ュ憡
+        // 1-Shp鏂囦欢锛�2-涓撻鍥撅紝3-鍏冩暟鎹紝4-涓氬姟鏁版嵁锛�5-绠¢亾鍒嗘瀽锛�6-缁熻鎶ュ憡锛�7-闄勪欢鏂囦欢锛�8-鐡︾墖鏂囦欢
         de.setType(4);
         de.setSizes(FileHelper.sizeToMb(new File(file).length()));
         de.setDepid(ue.getDepid());
@@ -342,15 +287,23 @@
             return null;
         }
 
-        List<String> list = codesAsList(codes);
-        removeDuplicate(list);
-        setRightLike(list, field);
+        //List<String> list = Arrays.asList(codes.split(","));
+        //List<String> list = codesAsList(codes);
+        //list = copeDirCodes(list);
+        //setRightLike(list, field);
 
-        return "(" + StringHelper.join(list, " or ") + ")";
+        //return "(" + StringHelper.join(list, " or ") + ")";
+
+        List<String> list = new ArrayList<>();
+        for (String dir : codes.split(",")) {
+            list.add("^" + dir);
+        }
+
+        return field + " ~ '" + StringHelper.join(list, "|") + "'";
     }
 
     /**
-     * 鍗曚綅缂栫爜杞泦鍚�
+     * 鍗曚綅缂栫爜杞泦鍚� *
      */
     private static List<String> codesAsList(String codes) {
         List<String> list = Arrays.asList(codes.split(","));
@@ -363,42 +316,113 @@
     }
 
     /**
-     * 鍘婚櫎閲嶅
+     * 澶勭悊鐩綍缂栫爜 *
      */
-    private static void removeDuplicate(List<String> list) {
-        int i = 0;
-        while (i < list.size()) {
-            int j = findStr(list, i);
-            if (j > -1) {
-                list.remove(j);
+    private static List<String> copeDirCodes(List<String> list) {
+        List<String> prjList = getDirCodesByLen(list, 0, 2);
+        List<String> appList = getDirCodesByLen(list, 3, 30);
+        if (prjList.isEmpty() && appList.isEmpty()) {
+            return list;
+        }
+        if (prjList.isEmpty()) {
+            return appList;
+        }
+        if (appList.isEmpty()) {
+            return prjList;
+        }
+
+        return filterCodes(prjList, appList);
+    }
+
+    /**
+     * 鏍规嵁闀垮害鑾峰彇鍗曚綅缂栫爜 *
+     */
+    private static List<String> getDirCodesByLen(List<String> list, int start, int end) {
+        List<String> rs = new ArrayList<>();
+        for (String code : list) {
+            if (StringHelper.isEmpty(code)) {
                 continue;
             }
 
-            i++;
-        }
-    }
-
-    /**
-     * 鏌ユ壘瀛楃涓�
-     */
-    private static int findStr(List<String> list, int i) {
-        String source = list.get(i);
-        for (int j = i + 1, c = list.size(); j < c; j++) {
-            if (list.get(j).startsWith(source)) {
-                return j;
+            if (code.length() >= start && code.length() <= end) {
+                rs.add(code);
             }
         }
 
-        return -1;
+        return rs;
     }
 
     /**
-     * 璁剧疆 鍙砽ike
+     * 杩囨护椤圭洰缂栫爜 *
+     */
+    private static List<String> filterCodes(List<String> prjList, List<String> appList) {
+        int i = 0;
+        while (i < appList.size()) {
+            boolean flag = false;
+            for (String prj : prjList) {
+                if (appList.get(i).startsWith(prj)) {
+                    flag = true;
+                    break;
+                }
+            }
+
+            if (!flag) {
+                appList.remove(i);
+                continue;
+            }
+            i++;
+        }
+
+        return appList;
+    }
+
+    /**
+     * 璁剧疆 鍙砽ike *
      */
     private static void setRightLike(List<String> list, String field) {
+        if (list.isEmpty()) {
+            list.add("1 = 2");
+            return;
+        }
+
         for (int i = 0, c = list.size(); i < c; i++) {
             String str = String.format("%s like '%s%%'", field, list.get(i));
             list.set(i, str);
         }
     }
+
+    /**
+     * 鏍规嵁璧勬枡绫诲埆鑾峰彇鐩綍缂栫爜
+     */
+    public String getDirsByTypes(String types, String dirs, String field) {
+        if (StringUtils.isEmpty(types) && StringUtils.isEmpty(dirs)) {
+            return null;
+        }
+
+        if (!StringUtils.isEmpty(types)) {
+            List<KeyValueEntity> list = baseQueryService.selectDirsByTypes(types.split(","), copeCodes(dirs, "code"));
+            if (null == list || list.isEmpty()) {
+                return "1 = 2";
+            }
+
+            return copeDirs(list, field);
+        }
+
+        return copeCodes(dirs, field);
+    }
+
+    /**
+     * 澶勭悊鐩綍缂栫爜锛歞ircode ~ '^0A|^0E'
+     */
+    public static String copeDirs(List<KeyValueEntity> kvs, String field) {
+        List<String> list = new ArrayList<>();
+        for (KeyValueEntity kv : kvs) {
+            String[] dirs = kv.getValue().split(",");
+            for (String dir : dirs) {
+                list.add("^" + dir);
+            }
+        }
+
+        return field + " ~ '" + StringHelper.join(list, "|") + "'";
+    }
 }

--
Gitblit v1.9.3