From ac20dc99bf1f463365dba071973e08fffbd294b4 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期五, 14 七月 2023 14:12:21 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/service/show/DataLibService.java |   97 ++++++++++++++++++++++++++++++++----------------
 1 files changed, 65 insertions(+), 32 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 da47e84..cb9c738 100644
--- a/src/main/java/com/lf/server/service/show/DataLibService.java
+++ b/src/main/java/com/lf/server/service/show/DataLibService.java
@@ -10,7 +10,7 @@
 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;
@@ -31,10 +31,10 @@
     PathHelper pathHelper;
 
     @Autowired
-    DownloadMapper downloadMapper;
+    MetaService metaService;
 
     @Autowired
-    DownloadService downloadService;
+    DownloadMapper downloadMapper;
 
     @Autowired
     BaseQueryService baseQueryService;
@@ -65,9 +65,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);
@@ -128,7 +126,6 @@
         }
 
         for (String id : ids) {
-            // || "00".equals(id)
             if (StringHelper.isEmpty(id)) {
                 continue;
             }
@@ -164,7 +161,7 @@
         ZipParameters params = Zip4jHelper.getZipParams(true);
         zip.addFolder(new File(gdbPath), params);
         // zip.addFolder(new File(annexPath), params)
-        downloadService.addAnnex(zip, params, annexMap);
+        metaService.addAnnex(zip, params, annexMap);
 
         String dbPwd = Md5Helper.reverse(Md5Helper.generate(dr.getPwd()));
         DownloadEntity de = getDownloadEntity(ue, zipFile, dbPwd);
@@ -185,7 +182,7 @@
                 }
 
                 QueryWrapper wrapper = createQueryWrapper(baseMapper, dr);
-                downloadService.addData(entity, baseMapper, wrapper, dataMap, annexMap);
+                metaService.addData(entity, baseMapper, wrapper, dataMap, annexMap);
             } catch (Exception ex) {
                 log.error(ex.getMessage(), ex);
             }
@@ -206,10 +203,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));
             }
         }
@@ -264,7 +261,7 @@
     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());
@@ -288,7 +285,7 @@
         }
 
         List<String> list = codesAsList(codes);
-        removeDuplicate(list);
+        list = copeDirCodes(list);
         setRightLike(list, field);
 
         return "(" + StringHelper.join(list, " or ") + ")";
@@ -308,39 +305,75 @@
     }
 
     /**
-     * 鍘婚櫎閲嶅
+     * 澶勭悊鐩綍缂栫爜
      */
-    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);
-                continue;
-            }
-
-            i++;
+    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 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;
+    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;
+            }
+
+            if (code.length() >= start && code.length() <= end) {
+                rs.add(code);
             }
         }
 
-        return -1;
+        return rs;
+    }
+
+    /**
+     * 杩囨护椤圭洰缂栫爜
+     */
+    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);

--
Gitblit v1.9.3