提交处理sql自动拼接业务线逻辑

王富生
1 parent 04ee424b
Showing 5 changed files with 249 additions and 2 deletions
urbanops-framework/urbanops-spring-boot-starter-mybatis/src/main/java/com/zteits/urbanops/framework/mybatis/config/BusinessLineSqlInterceptor.java
urbanops-framework/urbanops-spring-boot-starter-mybatis/src/main/java/com/zteits/urbanops/framework/mybatis/config/UrbanopsMybatisAutoConfiguration.java
urbanops-module-garden/src/main/java/com/zteits/urbanops/module/garden/dal/mysql/inspectionplan/InspectionPlanInterceptMapper.java
urbanops-module-garden/src/main/java/com/zteits/urbanops/module/garden/dal/mysql/inspectionplan/InspectionPlanMapper.java
urbanops-module-garden/src/main/java/com/zteits/urbanops/module/garden/service/inspectionplan/InspectionPlanServiceImpl.java
+package com.zteits.urbanops.framework.mybatis.config;
+
+import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
+import com.zteits.urbanops.framework.annotation.ProcessBusinessLine;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.ibatis.cache.CacheKey;
+import org.apache.ibatis.executor.Executor;
+import org.apache.ibatis.mapping.BoundSql;
+import org.apache.ibatis.mapping.MappedStatement;
+import org.apache.ibatis.mapping.SqlCommandType;
+import org.apache.ibatis.plugin.*;
+import org.apache.ibatis.reflection.MetaObject;
+import org.apache.ibatis.reflection.SystemMetaObject;
+import org.apache.ibatis.session.ResultHandler;
+import org.apache.ibatis.session.RowBounds;
+import org.springframework.stereotype.Component;
+
+import java.lang.reflect.Method;
+import java.util.HashSet;
+import java.util.Properties;
+import java.util.Set;
+import java.util.stream.Collectors;
+
+/**
+ * MyBatis 插件：拦截 SQL，自动拼接业务线条件
+ */
+@Slf4j
+@Component
+//@Intercepts({
+//        @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class})
+//        //@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class})
+//})
+// 关键：同时拦截两个query重载方法
+@Intercepts({
+        // 重载1：基础query（你原有的签名）
+        @Signature(
+                type = Executor.class,
+                method = "query",
+                args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}
+        ),
+        // 重载2：带CacheKey的query（MP/分页常用）
+        @Signature(
+                type = Executor.class,
+                method = "query",
+                args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class, CacheKey.class, BoundSql.class}
+        )
+})
+public class BusinessLineSqlInterceptor implements Interceptor {
+    /**
+     * 缓存：Mapper方法全限定名 → 注解（避免重复反射，提升性能）
+     */
+    private static final java.util.Map<String, ProcessBusinessLine> ANNOTATION_CACHE = new java.util.concurrent.ConcurrentHashMap<>();
+
+    @Override
+    public Object intercept(Invocation invocation) throws Throwable {
+        // 1. 跳过过滤则直接执行原 SQL
+        if (!isSkipAnnotation(invocation)) {
+            return invocation.proceed();
+        }
+
+        // 1. 解析拦截参数
+        Object[] args = invocation.getArgs();
+        MappedStatement ms = (MappedStatement) args[0];
+        Object parameter = args[1];
+        BoundSql boundSql = null;
+
+        // 2. 仅处理 SELECT 类型的 SQL
+        if (ms.getSqlCommandType() != SqlCommandType.SELECT) {
+            return invocation.proceed();
+        }
+
+        // 3. 适配两种重载方法，获取 BoundSql
+        if (args.length == 4) {
+            boundSql = ms.getBoundSql(parameter);
+        } else if (args.length == 6) {
+            boundSql = (BoundSql) args[5];
+        }
+        if (boundSql == null) {
+            return invocation.proceed();
+        }
+        String originalSql = boundSql.getSql().trim();
+        SqlCommandType sqlCommandType = ms.getSqlCommandType();
+        // 6. 替换 BoundSql 中的 SQL 语句（关键：反射修改）
+        MetaObject metaBoundSql = SystemMetaObject.forObject(boundSql);
+
+        // 3. 仅处理 SELECT/UPDATE/DELETE（INSERT 由自动填充处理）
+        if (SqlCommandType.SELECT.equals(sqlCommandType)) {
+            Set<String> businessLineSet = new HashSet<>();//UserContextHolder.getBusinessLine();
+            businessLineSet.add("yl");
+            if (!CollectionUtils.isEmpty(businessLineSet)) {
+                // 拼接业务线条件
+                String newSql = buildSqlWithBusinessLine(originalSql, businessLineSet, sqlCommandType);
+                log.debug("原始 SQL：{}，拼接后 SQL：{}", originalSql, newSql);
+                // 替换原始 SQL
+                metaBoundSql.setValue("sql", newSql);
+            }
+        }
+
+        // 4. 执行原方法
+        return invocation.proceed();
+    }
+
+    /**
+     * 拼接业务线条件到 SQL 中
+     */
+    private String buildSqlWithBusinessLine(String originalSql, Set<String> businessLineSet, SqlCommandType sqlCommandType) {
+        String upperSql = originalSql.toUpperCase();
+        // 核心转换：每个元素加单引号，逗号分隔
+        String result = businessLineSet.stream()
+                .map(s -> "'" + s + "'") // 为每个元素包裹单引号
+                .collect(Collectors.joining(",")); // 用逗号拼接所有元素
+        String condition = "busi_line in( " + result + ")";
+
+        // SELECT 语句：处理 WHERE/ORDER BY/GROUP BY 等
+        if (SqlCommandType.SELECT.equals(sqlCommandType)) {
+            int whereIndex = upperSql.indexOf("WHERE");
+            int orderIndex = upperSql.indexOf("ORDER BY");
+            int groupIndex = upperSql.indexOf("GROUP BY");
+            int havingIndex = upperSql.indexOf("HAVING");
+            int insertIndex = -1;
+            if (whereIndex > 0) {
+                insertIndex = whereIndex + 5; // WHERE 后开始拼接
+            } else {
+                // 无 WHERE，找 ORDER/GROUP/HAVING 的位置，在其前加 WHERE
+                insertIndex = originalSql.length();
+                if (orderIndex > 0) insertIndex = orderIndex;
+                else if (groupIndex > 0) insertIndex = groupIndex;
+                else if (havingIndex > 0) insertIndex = havingIndex;
+            }
+
+            // 拼接条件
+            String prefix = whereIndex > 0 ? " " +condition +" AND" : " WHERE "+ condition;
+            return new StringBuilder(originalSql)
+                    .insert(insertIndex, prefix)
+                    .toString();
+        }
+
+        // UPDATE/DELETE 语句：同理拼接 WHERE 条件
+        if (SqlCommandType.UPDATE.equals(sqlCommandType) || SqlCommandType.DELETE.equals(sqlCommandType)) {
+            int whereIndex = upperSql.indexOf("WHERE");
+            if (whereIndex > 0) {
+                return new StringBuilder(originalSql)
+                        .insert(whereIndex + 5, " AND " + condition)
+                        .toString();
+            } else {
+                return originalSql + " WHERE " + condition;
+            }
+        }
+
+        return originalSql;
+    }
+
+    /**
+     * 插件代理（固定写法）
+     */
+    @Override
+    public Object plugin(Object target) {
+        return Plugin.wrap(target, this);
+    }
+
+    /**
+     * 设置插件属性（无需配置则空实现）
+     */
+    @Override
+    public void setProperties(Properties properties) {
+    }
+
+    /**
+     * 判断massper 方式上是否存在注解 @SkipBusinessLine
+     * 有注解则拼接业务线；没有则不拼接
+     * @param invocation 对象
+     * @return true:拼接业务线；false不拼接业务线
+     */
+    public boolean isSkipAnnotation(Invocation invocation){
+        // 1. 获取MappedStatement（MP封装的Mapper方法信息）
+        MappedStatement ms = (MappedStatement) invocation.getArgs()[0];
+        String mapperMethodId = ms.getId(); // 格式：com.xxx.mapper.UserMapper.selectByAge
+        // 2. 解析Mapper全类名和方法名
+        String[] idParts = mapperMethodId.split("\\.");
+        String methodName = idParts[idParts.length - 1]; // 方法名：selectByAge
+        String mapperClassName = mapperMethodId.substring(0, mapperMethodId.lastIndexOf("."));
+        Class<?> mapperClass = null; // 加载Mapper接口
+        try {
+            mapperClass = Class.forName(mapperClassName);
+        } catch (ClassNotFoundException e) {
+            throw new RuntimeException(e);
+        }
+        ProcessBusinessLine skipAnnotation = ANNOTATION_CACHE.get(mapperClassName+methodName);
+        if (skipAnnotation == null) {
+            if(mapperClass.isAnnotationPresent(ProcessBusinessLine.class)){
+                ProcessBusinessLine processBusinessLine =  mapperClass.getAnnotation(ProcessBusinessLine.class);
+                ANNOTATION_CACHE.put(mapperClassName+methodName, processBusinessLine);
+                return true;
+            }
+        }
+
+        return false;
+    }
+
+}
 \ No newline at end of file
@@ -16,6 +16,7 @@ import com.baomidou.mybatisplus.extension.parser.cache.JdkSerialCaffeineJsqlPars
 import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
 import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import jakarta.annotation.Resource;
 import org.apache.ibatis.annotations.Mapper;
 import org.mybatis.spring.annotation.MapperScan;
 import org.springframework.boot.autoconfigure.AutoConfiguration;
@@ -36,6 +37,14 @@ import java.util.concurrent.TimeUnit;
         lazyInitialization = "${mybatis.lazy-initialization:false}") // Mapper 懒加载，目前仅用于单元测试
 public class UrbanopsMybatisAutoConfiguration {
+    /**
+     * 注册自定义业务线 SQL 拦截器
+     */
+    @Bean
+    public BusinessLineSqlInterceptor businessLineSqlInterceptor() {
+        return new BusinessLineSqlInterceptor();
+    }
+
     static {
         // 动态 SQL 智能优化支持本地缓存加速解析，更完善的租户复杂 XML 动态 SQL 支持，静态注入缓存
         JsqlParserGlobal.setJsqlParseCache(new JdkSerialCaffeineJsqlParseCache(
+package com.zteits.urbanops.module.garden.dal.mysql.inspectionplan;
+
+
+import com.zteits.urbanops.framework.annotation.ProcessBusinessLine;
+import com.zteits.urbanops.framework.common.pojo.PageResult;
+import com.zteits.urbanops.framework.mybatis.core.mapper.BaseMapperX;
+import com.zteits.urbanops.framework.mybatis.core.query.LambdaQueryWrapperX;
+import com.zteits.urbanops.module.garden.controller.admin.inspectionplan.vo.InspectionPlanPageReqVO;
+import com.zteits.urbanops.module.garden.dal.dataobject.inspectionplan.InspectionPlanDO;
+import org.apache.ibatis.annotations.Mapper;
+
+/**
+ * 巡检计划汇总 Mapper
+ *
+ * @author 超级管理员
+ */
+@ProcessBusinessLine
+@Mapper
+public interface InspectionPlanInterceptMapper extends BaseMapperX<InspectionPlanDO> {
+    default PageResult<InspectionPlanDO> selectPage(InspectionPlanPageReqVO reqVO) {
+        return selectPage(reqVO, new LambdaQueryWrapperX<InspectionPlanDO>()
+                .eqIfPresent(InspectionPlanDO::getBatchNo, reqVO.getBatchNo())
+                .likeIfPresent(InspectionPlanDO::getPlanName, reqVO.getPlanName())
+                .eqIfPresent(InspectionPlanDO::getPlanAttr, reqVO.getPlanAttr())
+                .eqIfPresent(InspectionPlanDO::getPlanTypeId, reqVO.getPlanTypeId())
+                .eqIfPresent(InspectionPlanDO::getCompanyId, reqVO.getCompanyId())
+                .likeIfPresent(InspectionPlanDO::getRoadName, reqVO.getRoadName())
+                .eqIfPresent(InspectionPlanDO::getLevelId, reqVO.getLevelId())
+                .betweenIfPresent(InspectionPlanDO::getBeginTime, reqVO.getBeginTime())
+                .betweenIfPresent(InspectionPlanDO::getEndTime, reqVO.getEndTime())
+                .orderByDesc(InspectionPlanDO::getId));
+    }
+
+}
 \ No newline at end of file
 package com.zteits.urbanops.module.garden.dal.mysql.inspectionplan;
+import com.zteits.urbanops.framework.annotation.ProcessBusinessLine;
 import com.zteits.urbanops.framework.common.pojo.PageResult;
 import com.zteits.urbanops.framework.mybatis.core.query.LambdaQueryWrapperX;
 import com.zteits.urbanops.framework.mybatis.core.mapper.BaseMapperX;
@@ -16,7 +17,6 @@ import org.apache.ibatis.annotations.Param;
  */
 @Mapper
 public interface InspectionPlanMapper extends BaseMapperX<InspectionPlanDO> {
-
     default PageResult<InspectionPlanDO> selectPage(InspectionPlanPageReqVO reqVO) {
         return selectPage(reqVO, new LambdaQueryWrapperX<InspectionPlanDO>()
                 .eqIfPresent(InspectionPlanDO::getBatchNo, reqVO.getBatchNo())
@@ -10,6 +10,7 @@ import com.zteits.urbanops.framework.security.core.util.SecurityFrameworkUtils;
 import com.zteits.urbanops.module.garden.dal.dataobject.inspectionplan.InspectionPlanDetailDO;
 import com.zteits.urbanops.module.garden.dal.dataobject.inspectionplan.InspectionPlanRoleDO;
 import com.zteits.urbanops.module.garden.dal.mysql.inspectionplan.InspectionPlanDetailMapper;
+import com.zteits.urbanops.module.garden.dal.mysql.inspectionplan.InspectionPlanInterceptMapper;
 import com.zteits.urbanops.module.garden.dal.mysql.inspectionplan.InspectionPlanRoleMapper;
 import com.zteits.urbanops.module.garden.enums.FinishStateEnum;
 import com.zteits.urbanops.module.garden.enums.PlanConstants;
@@ -51,6 +52,9 @@ public class InspectionPlanServiceImpl implements InspectionPlanService {
     private InspectionPlanMapper inspectionPlanMapper;
     @Resource
+    private InspectionPlanInterceptMapper inspectionPlanInterceptMapper;
+
+    @Resource
     private InspectionPlanDetailMapper inspectionPlanDetailMapper;
     @Resource
@@ -218,7 +222,7 @@ public class InspectionPlanServiceImpl implements InspectionPlanService {
     @Override
     public PageResult<InspectionPlanDO> getInspectionPlanPage(InspectionPlanPageReqVO pageReqVO) {
-        return inspectionPlanMapper.selectPage(pageReqVO);
+        return inspectionPlanInterceptMapper.selectPage(pageReqVO);
     }
     @Override