From 819512a986332036d9b46ea68f8ae19d02f34353 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期二, 28 三月 2023 15:04:23 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/service/show/DataLibService.java |   17 ++++++++++++-----
 1 files changed, 12 insertions(+), 5 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..5feedcb 100644
--- a/src/main/java/com/lf/server/service/show/DataLibService.java
+++ b/src/main/java/com/lf/server/service/show/DataLibService.java
@@ -148,8 +148,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) {
@@ -162,7 +162,7 @@
     /**
      * 鏌ヨ鏁版嵁
      */
-    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 +176,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