From 3cdd0dd684052517ee0e90daf4b03ad41b8f2771 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 03 四月 2023 15:04:47 +0800
Subject: [PATCH] 1
---
src/main/java/com/lf/server/service/data/DownloadService.java | 65 ++++++++++++++++++++++++++++++++
1 files changed, 64 insertions(+), 1 deletions(-)
diff --git a/src/main/java/com/lf/server/service/data/DownloadService.java b/src/main/java/com/lf/server/service/data/DownloadService.java
index 9a0a4b1..3050e91 100644
--- a/src/main/java/com/lf/server/service/data/DownloadService.java
+++ b/src/main/java/com/lf/server/service/data/DownloadService.java
@@ -1,5 +1,6 @@
package com.lf.server.service.data;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.lf.server.entity.all.StaticData;
import com.lf.server.entity.ctrl.DownloadReqEntity;
import com.lf.server.entity.data.DownloadEntity;
@@ -9,7 +10,9 @@
import com.lf.server.entity.sys.MetaDownEntity;
import com.lf.server.entity.sys.UserEntity;
import com.lf.server.helper.*;
+import com.lf.server.mapper.all.BasicMapper;
import com.lf.server.mapper.data.DownloadMapper;
+import com.lf.server.service.all.BaseQueryService;
import com.lf.server.service.sys.MetaDownService;
import net.lingala.zip4j.ZipFile;
import net.lingala.zip4j.model.FileHeader;
@@ -273,7 +276,64 @@
* 鏌ヨ鏁版嵁
*/
private void queryData(Map<String, List<String>> tabs, Map<String, List<?>> dataMap, Map<String, List<AttachEntity>> annexMap) {
- //
+ for (String tab : tabs.keySet()) {
+ try {
+ String entity = tab.split(StaticData.POINT)[1].replace("_", "");
+ BasicMapper baseMapper = ClassHelper.getBasicMapper(entity);
+ if (null == baseMapper) {
+ continue;
+ }
+
+ QueryWrapper wrapper = createQueryWrapper(baseMapper, tabs.get(tab));
+ addData(entity, baseMapper, wrapper, dataMap, annexMap);
+ } catch (Exception ex) {
+ log.error(ex.getMessage(), ex);
+ }
+ }
+ }
+
+ /**
+ * 娣诲姞鏁版嵁
+ */
+ public void addData(String entity, BasicMapper baseMapper, QueryWrapper wrapper, Map<String, List<?>> dataMap, Map<String, List<AttachEntity>> annexMap) {
+ List list = baseMapper.selectList(wrapper);
+ if (null == list || list.size() == 0) {
+ return;
+ }
+
+ if (!dataMap.containsKey(entity)) {
+ dataMap.put(entity, list);
+ } else {
+ dataMap.get(entity).addAll(list);
+ }
+ if (wrapper.isEmptyOfWhere()) {
+ wrapper.apply("1 = 1");
+ }
+ if (StaticData.BBOREHOLE.equals(entity)) {
+ wrapper.last("limit 100");
+ }
+
+ String tab = BaseQueryService.getTabName(baseMapper);
+ List<AttachEntity> annex = baseMapper.selectAnnex(tab, wrapper);
+ if (null == annex || annex.isEmpty()) {
+ return;
+ }
+
+ if (!annexMap.containsKey(entity)) {
+ annexMap.put(tab.replace(StaticData.POINT, "_"), annex);
+ } else {
+ annexMap.get(tab.replace(StaticData.POINT, "_")).addAll(annex);
+ }
+ }
+
+ /**
+ * 鍒涘缓鏌ヨ鍖呰鍣�
+ */
+ private <T> QueryWrapper<T> createQueryWrapper(BasicMapper baseMapper, List<String> ids) {
+ QueryWrapper<T> wrapper = new QueryWrapper<T>();
+
+
+ return wrapper;
}
/**
@@ -389,6 +449,9 @@
*/
private void addSingleFile(String uploadPath, MetaEntity mf, ZipFile zip, ZipParameters params) throws Exception {
File file = new File(uploadPath + File.separator + mf.getPath());
+ if (!file.exists() || file.isDirectory()) {
+ return;
+ }
zip.addFile(file, params);
String fileName = FileHelper.getFileName(file.getPath());
--
Gitblit v1.9.3