From 16aff7930d4a2b8e1034f8f3d6caafb5f422b363 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期六, 29 十月 2022 17:03:44 +0800
Subject: [PATCH] 1
---
src/main/java/com/lf/server/service/all/BaseQueryService.java | 37 ++++++++++++++++++++++++++++++++++---
1 files changed, 34 insertions(+), 3 deletions(-)
diff --git a/src/main/java/com/lf/server/service/all/BaseQueryService.java b/src/main/java/com/lf/server/service/all/BaseQueryService.java
index 66d0589..588fc9b 100644
--- a/src/main/java/com/lf/server/service/all/BaseQueryService.java
+++ b/src/main/java/com/lf/server/service/all/BaseQueryService.java
@@ -1,5 +1,6 @@
package com.lf.server.service.all;
+import cn.hutool.db.meta.Table;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.lf.server.helper.AesHelper;
import com.lf.server.helper.ClassHelper;
@@ -8,12 +9,20 @@
import com.lf.server.mapper.all.GeomBaseMapper;
import org.springframework.stereotype.Service;
+import java.util.HashMap;
+import java.util.Map;
+
/**
* 鐖舵煡璇㈡湇鍔＄被
* @author WWW
*/
@Service
public class BaseQueryService {
+ /**
+ * 琛ㄥ悕Map
+ */
+ private static Map<String, String> tabMap = new HashMap<String, String>(3);
+
/**
* 鑾峰彇鐖禡apper
*
@@ -159,11 +168,33 @@
* @return 琛ㄥ悕
*/
public String getTabName(BasicMapper basicMapper) {
- Object obj = ClassHelper.createEntityByMapper(basicMapper);
- if (obj == null) {
+ String className = ClassHelper.getClassName(basicMapper);
+ if (tabMap.containsKey(className)) {
+ return tabMap.get(className);
+ }
+
+ return getTabName(className);
+ }
+
+ /**
+ * 鏍规嵁Mapper鑾峰彇琛ㄥ悕
+ *
+ * @param className Mapper绫诲悕
+ * @return 琛ㄥ悕
+ */
+ private String getTabName(String className) {
+ Class clazz = ClassHelper.getEntityClass(className);
+ if (clazz == null) {
return null;
}
- return null;
+ Table annotation = (Table) clazz.getAnnotation(Table.class);
+
+ String tabName = annotation.getTableName();
+ if (tabName != null && !tabMap.containsKey(className)) {
+ tabMap.put(className, tabName);
+ }
+
+ return tabName;
}
}
--
Gitblit v1.9.3