From 71726a1cb05790d6841595ef7daef5173f2cddfa Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 29 三月 2023 09:43:11 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/service/show/DataLibService.java |   22 ++++++++++++++++------
 1 files changed, 16 insertions(+), 6 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 ad49eb6..adf6c3c 100644
--- a/src/main/java/com/lf/server/service/show/DataLibService.java
+++ b/src/main/java/com/lf/server/service/show/DataLibService.java
@@ -138,6 +138,9 @@
         }
 
         for (String id : ids) {
+            if (StringHelper.isEmpty(id)) {
+                continue;
+            }
             if (!rs.contains(id)) {
                 rs.add(id);
             }
@@ -148,8 +151,8 @@
      * 璇锋眰DB鏁版嵁涓嬭浇
      */
     public String downloadDbReq(UserEntity ue, DownloadReqEntity dr) {
-        Map<String, String> annexMap = new HashMap<>(3);
-        Map<String, List<?>> dataMap = new HashMap<>(3);
+        Map<String, List<?>> dataMap = new HashMap<>(2);
+        Map<String, List<AttachEntity>> annexMap = new HashMap<>(2);
 
         queryData(dr, dataMap, annexMap);
         if (dataMap.size() == 0) {
@@ -160,9 +163,9 @@
     }
 
     /**
-     * 鏌ヨ鏁版嵁
+     * 鏌ヨ鏁版嵁+闄勪欢
      */
-    private void queryData(DownloadReqEntity dr, Map<String, List<?>> dataMap, Map<String, String> annexMap) {
+    private void queryData(DownloadReqEntity dr, Map<String, List<?>> dataMap, Map<String, List<AttachEntity>> annexMap) {
         for (String enity : dr.getEntities()) {
             try {
                 BasicMapper baseMapper = ClassHelper.getBasicMapper(enity);
@@ -176,19 +179,26 @@
                     continue;
                 }
 
+                String tab = BaseQueryService.getTabName(baseMapper);
                 if (!dataMap.containsKey(enity)) {
                     dataMap.put(enity, list);
                 } else {
                     dataMap.get(enity).addAll(list);
                 }
+                if (wrapper.isEmptyOfWhere()) {
+                    wrapper.apply("1 = 1");
+                }
 
-                String tab = BaseQueryService.getTabName(baseMapper);
                 List<AttachEntity> annex = baseMapper.selectAnnex(tab, wrapper);
                 if (null == annex || annex.isEmpty()) {
                     continue;
                 }
 
-                //
+                if (!annexMap.containsKey(enity)) {
+                    annexMap.put(enity, annex);
+                } else {
+                    annexMap.get(enity).addAll(annex);
+                }
             } catch (Exception ex) {
                 log.error(ex.getMessage(), ex);
             }

--
Gitblit v1.9.3