From dd6183027575e9a335a7d4f2d145aec947969a34 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期六, 24 十二月 2022 10:44:43 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/service/show/DataLibService.java |   62 +++++++++++++++++++++++++++---
 1 files changed, 55 insertions(+), 7 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 9835765..2f4e354 100644
--- a/src/main/java/com/lf/server/service/show/DataLibService.java
+++ b/src/main/java/com/lf/server/service/show/DataLibService.java
@@ -16,10 +16,7 @@
 import org.springframework.stereotype.Service;
 
 import java.io.File;
-import java.util.Date;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 /**
  * 璧勬枡棣�
@@ -40,6 +37,54 @@
     BaseQueryService baseQueryService;
 
     private final static Log log = LogFactory.getLog(DataLibService.class);
+
+    /**
+     * 鏌ヨDB涓孩鍑虹殑鍗曚綅ID
+     */
+    public List<Integer> selectDbOverflowDep(UserEntity ue, List<String> entities, String wkt) {
+        List<Integer> rs = new ArrayList<>();
+        for (String enity : entities) {
+            try {
+                GeomBaseMapper<?> baseMapper = ClassHelper.getGeoBaseMapper(enity);
+                if (null == baseMapper) {
+                    continue;
+                }
+
+                QueryWrapper wrapper = new QueryWrapper();
+                wrapper.select("depid");
+                wrapper.gt("depid", 0);
+                wrapper.apply(String.format("depid != ALL(fn_rec_array(%d, 'dep')", ue.getDepid()));
+
+                Integer srid = baseQueryService.getSrid(baseMapper);
+                if (null != srid) {
+                    wrapper.apply(String.format("ST_Intersects(ST_PolygonFromText('%s', %d), geom)", wkt, srid));
+                }
+
+                List<Integer> ids = baseMapper.selectList(wrapper);
+
+                addDepIds(rs, ids);
+            } catch (Exception ex) {
+                log.error(ex.getMessage(), ex);
+            }
+        }
+
+        return rs;
+    }
+
+    /**
+     * 娣诲姞鍗曚綅ID
+     */
+    private void addDepIds(List<Integer> rs, List<Integer> ids) {
+        if (null == ids || ids.isEmpty()) {
+            return;
+        }
+
+        for (Integer id : ids) {
+            if (!rs.contains(id)) {
+                rs.add(id);
+            }
+        }
+    }
 
     /**
      * 鍒涘缓Zip鍖�
@@ -82,7 +127,7 @@
         Map<String, List<?>> map = new HashMap<>(5);
         for (String enity : entities) {
             try {
-                GeomBaseMapper baseMapper = ClassHelper.getGeoBaseMapper(enity);
+                GeomBaseMapper<?> baseMapper = ClassHelper.getGeoBaseMapper(enity);
                 if (null == baseMapper) {
                     continue;
                 }
@@ -97,7 +142,7 @@
                     map.put(enity, list);
                 }
             } catch (Exception ex) {
-                //
+                log.error(ex.getMessage(), ex);
             }
         }
 
@@ -110,8 +155,11 @@
     private QueryWrapper createWrapper(GeomBaseMapper baseMapper, String wkt) {
         QueryWrapper wrapper = new QueryWrapper();
         wrapper.select("ST_AsText(geom) as geom, *");
+
         Integer srid = baseQueryService.getSrid(baseMapper);
-        wrapper.apply(String.format("ST_Intersects(ST_PolygonFromText('%s', %d), geom)", wkt, srid));
+        if (null != srid) {
+            wrapper.apply(String.format("ST_Intersects(ST_PolygonFromText('%s', %d), geom)", wkt, srid));
+        }
 
         return wrapper;
     }

--
Gitblit v1.9.3