From 0f58d77f1df46e82dad00b23f245abdcd6cb8a2c Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 20 二月 2023 18:49:18 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/service/show/DataLibService.java |   40 +++++++++++++++++++++++++++++-----------
 1 files changed, 29 insertions(+), 11 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 0239259..9fca96d 100644
--- a/src/main/java/com/lf/server/service/show/DataLibService.java
+++ b/src/main/java/com/lf/server/service/show/DataLibService.java
@@ -91,8 +91,8 @@
     /**
      * 鍒涘缓Zip鍖�
      */
-    public String createZipFile(UserEntity ue, List<String> entities, String wkt, String pwd) throws Exception {
-        Map<String, List<?>> map = queryData(entities, wkt);
+    public String createZipFile(UserEntity ue, List<String> entities, String depcode, String dirs, String wkt, String pwd) throws Exception {
+        Map<String, List<?>> map = queryData(entities, depcode, dirs, wkt);
         if (map.size() == 0) {
             return null;
         }
@@ -125,7 +125,7 @@
     /**
      * 鏌ヨ鏁版嵁
      */
-    private Map<String, List<?>> queryData(List<String> entities, String wkt) {
+    private Map<String, List<?>> queryData(List<String> entities, String depcode, String dirs, String wkt) {
         Map<String, List<?>> map = new HashMap<>(5);
         for (String enity : entities) {
             try {
@@ -134,7 +134,7 @@
                     continue;
                 }
 
-                QueryWrapper wrapper = createWrapper(baseMapper, wkt);
+                QueryWrapper wrapper = createWrapper(baseMapper, depcode, dirs, wkt);
                 List<?> list = baseMapper.selectList(wrapper);
                 if (null == list || list.size() == 0) {
                     continue;
@@ -154,13 +154,23 @@
     /**
      * 鍒涘缓QueryWrapper
      */
-    private QueryWrapper createWrapper(GeomBaseMapper baseMapper, String wkt) {
+    private QueryWrapper createWrapper(BasicMapper baseMapper, String depcode, String dirs, String wkt) {
         QueryWrapper wrapper = new QueryWrapper();
-        wrapper.select("ST_AsText(geom) as geom, *");
+        if (baseMapper instanceof GeomBaseMapper) {
+            wrapper.select("ST_AsText(geom) as geom, *");
 
-        Integer srid = baseQueryService.getSrid(baseMapper);
-        if (null != srid) {
-            wrapper.apply(String.format("ST_Intersects(ST_PolygonFromText('%s', %d), geom)", wkt, srid));
+            Integer srid = baseQueryService.getSrid((GeomBaseMapper) baseMapper);
+            if (null != srid) {
+                wrapper.apply(String.format("ST_Intersects(ST_PolygonFromText('%s', %d), geom)", wkt, srid));
+            }
+        }
+        if (!StringHelper.isEmpty(depcode)) {
+            wrapper.likeRight("depid", depcode);
+        }
+
+        dirs = DataLibService.copeCodes(dirs, "dirid");
+        if (!StringHelper.isEmpty(dirs)) {
+            wrapper.apply(dirs);
         }
 
         return wrapper;
@@ -189,7 +199,7 @@
     private DownloadEntity getDownloadEntity(UserEntity ue, String file, String pwd) throws Exception {
         DownloadEntity de = new DownloadEntity();
         de.setName(FileHelper.getFileName(file));
-        // 1-Shp鏂囦欢锛�2-涓撻鍥撅紝3-鍏冩暟鎹紝4-涓氬姟鏁版嵁
+        // 1-Shp鏂囦欢锛�2-涓撻鍥撅紝3-婧愭暟鎹紝4-涓氬姟鏁版嵁锛�5-绠¢亾鍒嗘瀽锛�6-缁熻鎶ュ憡
         de.setType(4);
         de.setSizes(FileHelper.sizeToMb(new File(file).length()));
         de.setDepid(ue.getDepid());
@@ -207,7 +217,7 @@
     /**
      * 鎵撳寘DB鏁版嵁
      */
-    public String zipDbData(UserEntity ue, String name, String filter, String pwd) throws Exception {
+    public String zipDbData(UserEntity ue, String name, String depcode, String dirs, String filter, String pwd) throws Exception {
         BasicMapper baseMapper = ClassHelper.getBasicMapper(name);
         if (baseMapper == null) {
             return null;
@@ -218,6 +228,14 @@
         if (baseMapper instanceof GeomBaseMapper) {
             wrapper.select("ST_AsText(geom) as geom, *");
         }
+        if (!StringHelper.isEmpty(depcode)) {
+            wrapper.likeRight("depid", depcode);
+        }
+
+        dirs = DataLibService.copeCodes(dirs, "dirid");
+        if (!StringHelper.isEmpty(dirs)) {
+            wrapper.apply(dirs);
+        }
 
         List<?> list = baseMapper.selectList(wrapper);
         if (null == list || 0 == list.size()) {

--
Gitblit v1.9.3