From ae3ca6829a5bae6088b6f6a005377f11fee03df8 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 31 十月 2022 17:17:37 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/service/all/BaseQueryService.java |   96 ++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 81 insertions(+), 15 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 75c238b..306d2f9 100644
--- a/src/main/java/com/lf/server/service/all/BaseQueryService.java
+++ b/src/main/java/com/lf/server/service/all/BaseQueryService.java
@@ -1,36 +1,52 @@
 package com.lf.server.service.all;
 
+import com.baomidou.mybatisplus.annotation.TableName;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
-import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import com.lf.server.entity.ctrl.IdNameEntity;
 import com.lf.server.helper.AesHelper;
 import com.lf.server.helper.ClassHelper;
 import com.lf.server.helper.StringHelper;
+import com.lf.server.mapper.all.BaseQueryMapper;
+import com.lf.server.mapper.all.BasicMapper;
 import com.lf.server.mapper.all.GeomBaseMapper;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
 
 /**
  * 鐖舵煡璇㈡湇鍔＄被
  * @author WWW
  */
 @Service
-public class BaseQueryService {
+public class BaseQueryService implements BaseQueryMapper {
+    @Autowired
+    BaseQueryMapper baseQueryMapper;
+
+    /**
+     * 琛ㄥ悕Map
+     */
+    private static Map<String, String> tabMap = new HashMap<String, String>(3);
+
     /**
      * 鑾峰彇鐖禡apper
      *
      * @param name Mapper鍚�
      * @return BaseMapper
      */
-    public BaseMapper getBaseMapper(String name) {
+    public BasicMapper getBasicMapper(String name) {
         if (StringHelper.isEmpty(name)) {
             return null;
         }
 
         Object obj = ClassHelper.getBean(name.trim() + "Mapper");
-        if (!(obj instanceof BaseMapper)) {
+        if (!(obj instanceof BasicMapper)) {
             return null;
         }
 
-        return (BaseMapper) obj;
+        return (BasicMapper) obj;
     }
 
     /**
@@ -55,11 +71,10 @@
     /**
      * 娣诲姞杩囨护鏉′欢
      *
-     * @param baseMapper 鐖禡apper
-     * @param wrapper    QueryWrapper
-     * @param filter     鍘熷杩囨护鏉′欢瀛楃涓�
+     * @param wrapper QueryWrapper
+     * @param filter  鍘熷杩囨护鏉′欢瀛楃涓�
      */
-    public void addFilterWrapper(BaseMapper baseMapper, QueryWrapper wrapper, String filter) {
+    public void addFilterWrapper(QueryWrapper wrapper, String filter) {
         if (StringHelper.isEmpty(filter)) {
             return;
         }
@@ -140,16 +155,67 @@
     /**
      * 娣诲姞绌洪棿杩囨护鏉′欢
      *
-     * @param baseMapper 鐖禡apper
-     * @param wrapper    QueryWrapper
-     * @param wkt        WKT锛堣憲鍚嶆枃鏈級
-     * @param srid       绌洪棿寮曠敤鏍囪瘑绗�
+     * @param basicMapper 鐖禡apper
+     * @param wrapper     QueryWrapper
+     * @param wkt         WKT锛堣憲鍚嶆枃鏈級
+     * @param srid        绌洪棿寮曠敤鏍囪瘑绗�
      * @throws Exception 寮傚父
      */
-    public void addGeomWrapper(BaseMapper baseMapper, QueryWrapper wrapper, String wkt, Integer srid) throws Exception {
-        if (baseMapper instanceof GeomBaseMapper && !StringHelper.isEmpty(wkt) && srid != null) {
+    public void addGeomWrapper(BasicMapper basicMapper, QueryWrapper wrapper, String wkt, Integer srid) throws Exception {
+        if (basicMapper instanceof GeomBaseMapper && !StringHelper.isEmpty(wkt) && srid != null) {
             wkt = AesHelper.decrypt(wkt);
             wrapper.apply(String.format("ST_Intersects(geom, ST_PolygonFromText('%s', %d))", wkt, srid));
         }
     }
+
+    /**
+     * 鏍规嵁Mapper鑾峰彇琛ㄥ悕
+     *
+     * @param basicMapper Mapper
+     * @return 琛ㄥ悕
+     */
+    public String getTabName(BasicMapper basicMapper) {
+        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;
+        }
+
+        TableName annotation = (TableName) clazz.getAnnotation(TableName.class);
+
+        String tabName = annotation.value();
+        if (tabName != null && !tabMap.containsKey(className)) {
+            tabMap.put(className, tabName);
+        }
+
+        return tabName;
+    }
+
+    @Override
+    public List<IdNameEntity> selectUserFuzzy(String name) {
+        name = StringHelper.getLikeStr(name);
+
+        return baseQueryMapper.selectUserFuzzy(name);
+    }
+
+    @Override
+    public List<IdNameEntity> selectDepFuzzy(String name) {
+        name = StringHelper.getLikeStr(name);
+
+        return baseQueryMapper.selectDepFuzzy(name);
+    }
 }

--
Gitblit v1.9.3