From 796b44ea813a1133beae4f3a67f1c0263510c0c7 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期日, 17 十一月 2024 09:45:07 +0800
Subject: [PATCH] 1
---
src/main/java/com/moon/server/service/show/DataLibService.java | 117 +++++++++++++++++++++++++---------------------------------
1 files changed, 51 insertions(+), 66 deletions(-)
diff --git a/src/main/java/com/moon/server/service/show/DataLibService.java b/src/main/java/com/moon/server/service/show/DataLibService.java
index c763e10..2a99022 100644
--- a/src/main/java/com/moon/server/service/show/DataLibService.java
+++ b/src/main/java/com/moon/server/service/show/DataLibService.java
@@ -21,11 +21,8 @@
import java.io.File;
import java.util.*;
-/**
- * 璧勬枡棣�
- * @author WWW
- */
@Service
+@SuppressWarnings("ALL")
public class DataLibService {
@Autowired
PathHelper pathHelper;
@@ -41,9 +38,6 @@
private final static Log log = LogFactory.getLog(DataLibService.class);
- /**
- * 鏌ヨDB涓孩鍑虹殑鍗曚綅缂栫爜
- */
public List<String> selectDbOverflowDep(UserEntity ue, DownloadReqEntity dr) {
if (!StringHelper.isEmpty(dr.getWkt())) {
return selectDbOverflowDep4Wkt(ue, dr);
@@ -52,9 +46,6 @@
return selectDbOverflowDep4Prop(ue, dr);
}
- /**
- * 鏌ヨDB涓孩鍑虹殑鍗曚綅缂栫爜-绌洪棿鏌ヨ
- */
public List<String> selectDbOverflowDep4Wkt(UserEntity ue, DownloadReqEntity dr) {
List<String> rs = new ArrayList<>();
for (String entity : dr.getEntities()) {
@@ -74,9 +65,6 @@
return rs;
}
- /**
- * 鏌ヨDB涓孩鍑虹殑鍗曚綅缂栫爜-灞炴�ф煡璇�
- */
public List<String> selectDbOverflowDep4Prop(UserEntity ue, DownloadReqEntity dr) {
List<String> rs = new ArrayList<>();
BasicMapper<?> baseMapper = ClassHelper.getBasicMapper(dr.getEntities().get(0));
@@ -91,9 +79,6 @@
return rs;
}
- /**
- * 鏌ヨDB涓孩鍑哄崟浣嶇紪鐮佺殑鏌ヨ鍖呰鍣�
- */
private <T> QueryWrapper<T> getWrapper4DbOverflow(UserEntity ue, DownloadReqEntity dr) {
QueryWrapper<T> wrapper = new QueryWrapper<T>();
wrapper.select("depid");
@@ -117,9 +102,6 @@
return wrapper;
}
- /**
- * 娣诲姞鍗曚綅缂栫爜
- */
private void addDepCodes(List<String> rs, List<String> ids) {
if (null == ids || ids.isEmpty()) {
return;
@@ -135,9 +117,6 @@
}
}
- /**
- * 璇锋眰DB鏁版嵁涓嬭浇
- */
public String downloadDbReq(UserEntity ue, DownloadReqEntity dr) throws Exception {
Map<String, List<?>> dataMap = new HashMap<>(2);
Map<String, List<AttachEntity>> annexMap = new HashMap<>(2);
@@ -170,9 +149,6 @@
return rows > 0 ? de.getGuid() : null;
}
- /**
- * 鏌ヨ鏁版嵁+闄勪欢
- */
private void queryData(DownloadReqEntity dr, Map<String, List<?>> dataMap, Map<String, List<AttachEntity>> annexMap) {
for (String entity : dr.getEntities()) {
try {
@@ -189,9 +165,6 @@
}
}
- /**
- * 鍒涘缓鏌ヨ鍖呰鍣�
- */
private <T> QueryWrapper<T> createQueryWrapper(BasicMapper baseMapper, DownloadReqEntity dr) {
QueryWrapper<T> wrapper = new QueryWrapper<T>();
@@ -219,9 +192,6 @@
return wrapper;
}
- /**
- * 鍒涘缓闄勪欢 *
- */
private void createAnnex(String annexPath, Map<String, List<AttachEntity>> annexMap) {
if (annexMap.size() == 0) {
return;
@@ -255,9 +225,6 @@
}
}
- /**
- * 鑾峰彇涓嬭浇瀹炰綋绫�
- */
private DownloadEntity getDownloadEntity(UserEntity ue, String file, String pwd) {
DownloadEntity de = new DownloadEntity();
de.setName(FileHelper.getFileName(file));
@@ -276,24 +243,18 @@
return de;
}
- /**
- * 澶勭悊缂栫爜
- */
public static String copeCodes(String codes, String field) {
if (StringHelper.isEmpty(codes) || StringHelper.isSqlInjection(codes)) {
return null;
}
List<String> list = codesAsList(codes);
- removeDuplicate(list);
+ list = copeDirCodes(list);
setRightLike(list, field);
return "(" + StringHelper.join(list, " or ") + ")";
}
- /**
- * 鍗曚綅缂栫爜杞泦鍚�
- */
private static List<String> codesAsList(String codes) {
List<String> list = Arrays.asList(codes.split(","));
Set set = new HashSet(list);
@@ -304,40 +265,64 @@
return newList;
}
- /**
- * 鍘婚櫎閲嶅
- */
- private static void removeDuplicate(List<String> list) {
- int i = 0;
- while (i < list.size()) {
- int j = findStr(list, i);
- if (j > -1) {
- list.remove(j);
+ private static List<String> copeDirCodes(List<String> list) {
+ List<String> prjList = getDirCodesByLen(list, 0, 2);
+ List<String> appList = getDirCodesByLen(list, 3, 30);
+ if (prjList.isEmpty() && appList.isEmpty()) {
+ return list;
+ }
+ if (prjList.isEmpty()) {
+ return appList;
+ }
+ if (appList.isEmpty()) {
+ return prjList;
+ }
+
+ return filterCodes(prjList, appList);
+ }
+
+ private static List<String> getDirCodesByLen(List<String> list, int start, int end) {
+ List<String> rs = new ArrayList<>();
+ for (String code : list) {
+ if (StringHelper.isEmpty(code)) {
continue;
}
- i++;
- }
- }
-
- /**
- * 鏌ユ壘瀛楃涓�
- */
- private static int findStr(List<String> list, int i) {
- String source = list.get(i);
- for (int j = i + 1, c = list.size(); j < c; j++) {
- if (list.get(j).startsWith(source)) {
- return j;
+ if (code.length() >= start && code.length() <= end) {
+ rs.add(code);
}
}
- return -1;
+ return rs;
}
- /**
- * 璁剧疆 鍙砽ike
- */
+ private static List<String> filterCodes(List<String> prjList, List<String> appList) {
+ int i = 0;
+ while (i < appList.size()) {
+ boolean flag = false;
+ for (String prj : prjList) {
+ if (appList.get(i).startsWith(prj)) {
+ flag = true;
+ break;
+ }
+ }
+
+ if (!flag) {
+ appList.remove(i);
+ continue;
+ }
+ i++;
+ }
+
+ return appList;
+ }
+
private static void setRightLike(List<String> list, String field) {
+ if (list.isEmpty()) {
+ list.add("1 = 2");
+ return;
+ }
+
for (int i = 0, c = list.size(); i < c; i++) {
String str = String.format("%s like '%s%%'", field, list.get(i));
list.set(i, str);
--
Gitblit v1.9.3