关于mybatis拦截器

已于 2024-12-12 11:58:48 修改
阅读量406 收藏
点赞数 2
CC 4.0 BY-SA版权
文章标签： mybatis
于 2024-12-10 17:05:05 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_37253610/article/details/144377737

【背景】

希望select语句执行时，都能自动拼接del_flag和dept_id查询条件。
但现表结构混乱，并不是所有表都存在这两个字段，且并不是都叫这两个名字。

【执行逻辑】

1.项目启动时，把当前项目所连表中，需要字段读出来，

2.select语句过来时，查当前表是否存在要拼接字段，存在则做拼接

【代码实现】

第一步：

import com.yushu.base.common.utils.RedisUtils;
import com.yushu.base.framework.constant.ColumnConstant;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
import org.springframework.boot.autoconfigure.SpringBootApplication;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.List;

@Slf4j
@SpringBootApplication
public class DatabaseMetadataLoaderApplication implements CommandLineRunner {

    @Autowired
    private DataSource dataSource;

    private final static List<String> COLUMN_LIST = ColumnConstant.LIST;

    @Override
    public void run(String... args) throws Exception {
        long start = System.currentTimeMillis();
        // 加载所有表和字段信息
        loadDatabaseMetadata();
        long end = System.currentTimeMillis();
        System.out.println("[DatabaseMetadataLoaderApplication]处理时长：" + (end - start));
    }

    /**
     * 获取所有表和字段信息
     *
     * @throws SQLException
     */
    public void loadDatabaseMetadata() throws SQLException {

        // 获取数据库连接
        try (Connection connection = dataSource.getConnection()) {
            DatabaseMetaData metaData = connection.getMetaData();

            // 获取所有表
            try (ResultSet tables = metaData.getTables(connection.getCatalog(), null, "%", new String[]{"TABLE"})) {
                while (tables.next()) {
                    String tableName = tables.getString("TABLE_NAME");
                    StringBuilder value = new StringBuilder();
                    // 获取当前表的所有字段
                    try (ResultSet columnsResultSet = metaData.getColumns(connection.getCatalog(), null, tableName, "%")) {
                        while (columnsResultSet.next()) {
                            String columnName = columnsResultSet.getString("COLUMN_NAME");
                            if (COLUMN_LIST.contains(columnName)) {
                                value.append(columnName).append(",");
                            }
                        }
                    }
                    log.info("[DatabaseMetadataLoaderApplication]TABLE_NAME:{},COLUMN_NAME:{}", tableName, value);
                    RedisUtils.set(tableName, value.toString().endsWith(",") ? value.substring(0, value.length() - 1) : value);
                }
            }
        }
    }
}

第二步：


import cn.hutool.core.util.ReflectUtil;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.yushu.base.common.utils.CollectionUtil;
import com.yushu.base.common.utils.RedisUtils;
import com.yushu.base.common.utils.SecurityUtils;
import com.yushu.base.framework.constant.ColumnConstant;
import lombok.extern.slf4j.Slf4j;
import net.sf.jsqlparser.JSQLParserException;
import net.sf.jsqlparser.expression.operators.conditional.AndExpression;
import net.sf.jsqlparser.parser.CCJSqlParserUtil;
import net.sf.jsqlparser.schema.Table;
import net.sf.jsqlparser.statement.Statement;
import net.sf.jsqlparser.statement.select.FromItem;
import net.sf.jsqlparser.statement.select.PlainSelect;
import net.sf.jsqlparser.statement.select.Select;
import net.sf.jsqlparser.statement.select.SubSelect;
import org.apache.ibatis.executor.statement.StatementHandler;
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.springframework.stereotype.Component;

import java.sql.Connection;
import java.util.List;
import java.util.Properties;

@Slf4j
@Component
@Intercepts({
        @Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class})})
public class MybatisSqlLogInterceptor implements Interceptor {


    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        StatementHandler statementHandler = (StatementHandler) invocation.getTarget();

        StatementHandler delegate = (StatementHandler) ReflectUtil.getFieldValue(statementHandler, "delegate");
        MappedStatement mappedStatement = (MappedStatement) ReflectUtil.getFieldValue(delegate, "mappedStatement");
        SqlCommandType commandType = mappedStatement.getSqlCommandType();

        if (SqlCommandType.SELECT.equals(commandType)) {
            BoundSql boundSql = statementHandler.getBoundSql();
            Statement statement = CCJSqlParserUtil.parse(boundSql.getSql());
            PlainSelect selectBody = (PlainSelect) ((Select) statement).getSelectBody();
            FromItem fromItem =  selectBody.getFromItem();

            if (fromItem instanceof Table) {
                Table table = (Table) fromItem;
                String tableName = table.getName();
                String tableAlias = table.getAlias() == null ? null : table.getAlias().getName();
                System.out.println("=====" + tableName + " " + tableAlias);
                buildWhereClause(selectBody, getDataFilter(tableName, tableAlias));
            } else if (fromItem instanceof SubSelect) {
                // 子查询的先不管了，历史遗留查询逻辑保持原样，新逻辑不开发子查询
                // 正常如果是子查询，应递归处理
//                SubSelect subSelect = (SubSelect) fromItem;
//                handleSelectBody(subSelect.getSelectBody());
            }
            ReflectUtil.setFieldValue(boundSql, "sql", statement.toString());
        }

        return invocation.proceed();
    }

    private String getDataFilter(String tableName, String tableAlias) {
        // 项目启动时，读取所有表结构，记录表名与字段的关联关系，放入缓存
        // 以下校验表中是否存在需拼接字段，若存在则进行sql拼接
        String columnStr = RedisUtils.getStr(tableName);
        if (StringUtils.isEmpty(columnStr)) {
            return "";
        }

        StringBuilder stringBuilder = new StringBuilder();
        List<String> columnList = CollectionUtil.toList(columnStr.split(","));
        for (String column : columnList) {
            if (stringBuilder.length() > 0) {
                stringBuilder.append("and ");
            }
            switch (column) {
                // 3.组装sql
                case ColumnConstant.DEL_FLAG:
                    stringBuilder.append(" ").append(getTableAlias(tableAlias)).
                            append(ColumnConstant.DEL_FLAG).append(" = ").append("0").append(" ");
                    break;
                case ColumnConstant.SCHOOL_ID:
                    stringBuilder.append(" ").append(getTableAlias(tableAlias)).
                            append(ColumnConstant.SCHOOL_ID).append(" = ").append(SecurityUtils.getDeptId()).append(" ");
                    break;
                default:
            }
        }
        return stringBuilder.toString();
    }

    private String getTableAlias(String tableAlias) {
        if (StringUtils.isEmpty(tableAlias)) {
            return "";
        }
        return tableAlias + ".";
    }

    /**
     * 添加查询条件
     *
     * @param select
     * @param dataFilter
     * @throws JSQLParserException
     */
    private void buildWhereClause(PlainSelect select, String dataFilter) throws JSQLParserException {
        if (StringUtils.isBlank(dataFilter)) {
            return;
        }
        if (select.getWhere() == null) {
            select.setWhere(CCJSqlParserUtil.parseCondExpression(dataFilter));
        } else {
            AndExpression and = new AndExpression(
                    CCJSqlParserUtil.parseCondExpression(dataFilter), select.getWhere());
            select.setWhere(and);
        }
    }

    @Override
    public Object plugin(Object target) {
        if (target instanceof StatementHandler) {
            return Plugin.wrap(target, this);
        }
        return target;
    }

    @Override
    public void setProperties(Properties properties) {
    }
}
====================   遇到的问题   ====================

【问题描述】

程序执行报错   module java.base does not "opens java.lang.reflect" to unnamed module

【原因分析】

Java9及以上版本中引入的Java Platform Module System引起的，特别是强封装的实现。它仅在特定条件下允许access，对于反射导致异常的代码尝试使用相同的限制。

【解决方案】

增加如下虚拟运行参数

--add-opens java.base/java.lang=ALL-UNNAMED --add-opens java.base/java.lang.reflect=ALL-UNNAMED

17:18:24.648 [restartedMain] WARN  o.s.b.w.s.c.AnnotationConfigServletWebServerApplicationContext - [refresh,591] - Exception encountered during context initialization - cancelling refresh attempt: org.springframework.beans.factory.BeanCreationException: Error creating bean with name 'jwTeacherInfoController': Injection of resource dependencies failed; nested exception is org.springframework.beans.factory.BeanCreationException: Error creating bean with name 'sysUserServiceImpl': Injection of resource dependencies failed; nested exception is org.springframework.beans.factory.BeanCreationException: Error creating bean with name 'sysConfigServiceImpl': Invocation of init method failed; nested exception is org.mybatis.spring.MyBatisSystemException: nested exception is org.apache.ibatis.exceptions.PersistenceException:
### Error querying database.  Cause: java.lang.reflect.InaccessibleObjectException: Unable to make field protected java.lang.reflect.InvocationHandler java.lang.reflect.Proxy.h accessible: module java.base does not "opens java.lang.reflect" to unnamed module @79defdc
### The error may exist in file [F:\Java\yushu-system\target\classes\mapper\system\SysConfigMapper.xml]
### The error may involve com.yushu.base.system.mapper.SysConfigMapper.selectConfigList
### The error occurred while executing a query
### Cause: java.lang.reflect.InaccessibleObjectException: Unable to make field protected java.lang.reflect.InvocationHandler java.lang.reflect.Proxy.h accessible: module java.base does not "opens java.lang.reflect" to unnamed module @79defdc
17:18:24.650 [restartedMain] INFO  c.a.d.p.DruidDataSource - [close,2105] - {dataSource-1} closing ...
17:18:24.655 [restartedMain] INFO  c.a.d.p.DruidDataSource - [close,2178] - {dataSource-1} closed
17:18:24.670 [restartedMain] INFO  o.a.c.c.StandardService - [log,173] - Stopping service [Tomcat]
17:18:24.704 [restartedMain] ERROR o.s.b.SpringApplication - [reportFailure,870] - Application run failed
org.springframework.beans.factory.BeanCreationException: Error creating bean with name 'jwTeacherInfoController': Injection of resource dependencies failed; nested exception is org.springframework.beans.factory.BeanCreationException: Error creating bean with name 'sysUserServiceImpl': Injection of resource dependencies failed; nested exception is org.springframework.beans.factory.BeanCreationException: Error creating bean with name 'sysConfigServiceImpl': Invocation of init method failed; nested exception is org.mybatis.spring.MyBatisSystemException: nested exception is org.apache.ibatis.exceptions.PersistenceException:
### Error querying database.  Cause: java.lang.reflect.InaccessibleObjectException: Unable to make field protected java.lang.reflect.InvocationHandler java.lang.reflect.Proxy.h accessible: module java.base does not "opens java.lang.reflect" to unnamed module @79defdc
### The error may exist in file [F:\Java\yushu-system\target\classes\mapper\system\SysConfigMapper.xml]
### The error may involve com.yushu.base.system.mapper.SysConfigMapper.selectConfigList
### The error occurred while executing a query
### Cause: java.lang.reflect.InaccessibleObjectException: Unable to make field protected java.lang.reflect.InvocationHandler java.lang.reflect.Proxy.h accessible: module java.base does not "opens java.lang.reflect" to unnamed module @79defdc
    at org.springframework.context.annotation.CommonAnnotationBeanPostProcessor.postProcessProperties(CommonAnnotationBeanPostProcessor.java:332)
    at org.springframework.beans.factory.support.AbstractAutowireCapableBeanFactory.populateBean(AbstractAutowireCapableBeanFactory.java:1431)
    at org.springframework.beans.factory.support.AbstractAutowireCapableBeanFactory.doCreateBean(AbstractAutowireCapableBeanFactory.java:619)
    at org.springframework.beans.factory.support.AbstractAutowireCapableBeanFactory.createBean(AbstractAutowireCapableBeanFactory.java:542)
    at org.springframework.beans.factory.support.AbstractBeanFactory.lambda$doGetBean$0(AbstractBeanFactory.java:335)
    at org.springframework.beans.factory.support.DefaultSingletonBeanRegistry.getSingleton(DefaultSingletonBeanRegistry.java:234)
    at org.springframework.beans.factory.support.AbstractBeanFactory.doGetBean(AbstractBeanFactory.java:333)
    at org.springframework.beans.factory.support.AbstractBeanFactory.getBean(AbstractBeanFactory.java:208)
    at org.springframework.beans.factory.support.DefaultListableBeanFactory.preInstantiateSingletons(DefaultListableBeanFactory.java:953)
    at org.springframework.context.support.AbstractApplicationContext.finishBeanFactoryInitialization(AbstractApplicationContext.java:918)
    at org.springframework.context.support.AbstractApplicationContext.refresh(AbstractApplicationContext.java:583)
    at org.springframework.boot.web.servlet.context.ServletWebServerApplicationContext.refresh(ServletWebServerApplicationContext.java:145)
    at org.springframework.boot.SpringApplication.refresh(SpringApplication.java:780)
    at org.springframework.boot.SpringApplication.refreshContext(SpringApplication.java:453)
    at org.springframework.boot.SpringApplication.run(SpringApplication.java:343)
    at org.springframework.boot.SpringApplication.run(SpringApplication.java:1370)
    at org.springframework.boot.SpringApplication.run(SpringApplication.java:1359)
    at com.yushu.base.YushuApplication.main(YushuApplication.java:22)
    at java.base/jdk.internal.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
    at java.base/jdk.internal.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:77)
    at java.base/jdk.internal.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)
    at java.base/java.lang.reflect.Method.invoke(Method.java:568)
    at org.springframework.boot.devtools.restart.RestartLauncher.run(RestartLauncher.java:49)

【问题描述】

部分sql不需要自动拼接上述查询条件。比如登录时查询用户信息，此时还没有dept_id；或者定时任务执行时也无需拼接dept_id字段

【解决方案】

以登录为例：

设置登录标志，为用户登录的线程上下文添加一个标志，在 MyBatis 拦截器中判断当前是否为登录场景。

步骤：

  1. 在登录请求处理时设置标志。

  2. 在拦截器中检查标志，决定是否跳过部门 ID 的拼接。

代码示例：

登录请求时：

public ResponseEntity<?> login(LoginRequest loginRequest) {
    try {
        LoginContext.setIsLoginRequest(true); // 设置为登录请求
        Authentication authentication = authenticationManager.authenticate(
            new UsernamePasswordAuthenticationToken(
                loginRequest.getUsername(), 
                loginRequest.getPassword()
            )
        );

        // 登录成功后清除标志
        LoginContext.clear();
        return ResponseEntity.ok(new JwtResponse(...));
    } catch (Exception e) {
        LoginContext.clear();
        throw e;
    }
}

线程上下文工具类：

public class LoginContext {
    private static final ThreadLocal<Boolean> IS_LOGIN_REQUEST = new ThreadLocal<>();

    public static void setIsLoginRequest(boolean isLogin) {
        IS_LOGIN_REQUEST.set(isLogin);
    }

    public static boolean isLoginRequest() {
        return IS_LOGIN_REQUEST.get() != null && IS_LOGIN_REQUEST.get();
    }

    public static void clear() {
        IS_LOGIN_REQUEST.remove();
    }
}

拦截器中判断：

@Override
public Object intercept(Invocation invocation) throws Throwable {
    if (LoginContext.isLoginRequest()) {
        return invocation.proceed(); // 登录请求，直接放行
    }

    // 正常 SQL 处理逻辑
    handleDepartmentId(...);
    return invocation.proceed();
}

井盖987
关注
利用mybatis 拦截器插件功能实现分片SQL
java_magicsun的专栏
07-18 1345
关于mybatis的拦截器 请移步 mybatis 拦截器实现 我们今天就根据这个插件 实现根据值hash分表功能 首先 准备一个 注解类 Share @Target({ElementType.TYPE}) @Retention(RetentionPolicy.RUNTIME) @Inherited @Documented public @interface Share { ...
mybatis通过拦截器获取sql语句，操作类型，数据库链接URL，库名，表名，处理字段名
qq_38978882的博客
02-05 2074
mybatis通过拦截器获取sql语句，操作类型，数据库链接URL，库名，表名，处理字段名
2025Mybatis最新教程（二）
最新发布
m0_74246047的博客
05-30 467
摘要： 本章介绍了MyBatis环境搭建的完整流程。首先通过Maven创建新项目，然后在pom.xml中添加必要依赖，包括Druid连接池(1.2.6)、MySQL驱动(8.0.25)、MyBatis核心(3.5.6)和Lombok(1.18.20)。接着在resources目录下创建mybatis-config.xml配置文件，详细配置了数据库环境(MySQL)、事务管理(JDBC)、数据源(PooledDataSourceFactory)以及Mapper文件位置。配置文件中包含了数据库连接所需的驱动类、
Spring Boot集成MyBatis-Plus：自定义拦截器实现动态表名切换
weixin_45626288的博客
11-25 2072
介绍动态表名的场景需求，比如多租户系统、分表分库，或者不同业务模块共用一套代码但操作不同表。说明 MyBatis-Plus 默认绑定固定表名的问题。解释 MyBatis 拦截器的核心概念，介绍接口和@Signature注解。MyBatis 拦截器为开发者提供了灵活的扩展能力，可以在 SQL 执行的多个阶段中注入自定义逻辑，从而实现多种高级功能。合理使用拦截器不仅能增强系统功能，还能提升性能和安全性。
mybatis-plus自定义拦截器统计方法中调用所有数据库表的名称
wuhao2343的博客
05-21 1280
/定义正则表达式，提取出sql语句中的关键字 private static final Pattern TABLE_NAME_PATTERN = Pattern . compile("(FROM|UPDATE)\\s+([\\w\\._]+)");// 解析SQL语句，提取表名称 Matcher matcher = TABLE_NAME_PATTERN . matcher(sql);
mybatis拦截器进行处理动态表名的问题
Linaon的博客
04-26 503
【代码】mybatis拦截器进行处理动态表名的问题。
分库分表 MyBatis的拦截器（Interceptor）在 SQL 执行前动态修改表名
2301_77615930的博客
03-05 623
分库分表 MyBatis的拦截器（Interceptor）在 SQL 执行前动态修改表名
Mybatis拦截器
qh870754310的博客
02-07 221
Mybatis核心对象 从MyBatis代码实现的角度来看，MyBatis的主要的核心部件有以下几个： Configuration 初始化基础配置，比如MyBatis的别名等，一些重要的类型对象，如，插件，映射器，ObjectFactory和typeHandler对象，MyBatis所有的配置信息都维持在Configuration对象之中 SqlSessionFactory SqlSession工厂 SqlSession 作为MyBatis工作的主要顶层API，表示和数据库交互的会话，完成必要数据库增删改
mybatis拦截器实现通用权限字段添加的方法
08-25
MyBatis拦截器实现通用权限字段添加的方法 MyBatis拦截器是一种非常实用的技术，可以用来实现各种复杂的数据库操作。本文将详细介绍如何使用MyBatis拦截器来实现通用权限字段添加，达到灵活、可靠、可维护的数据库...
通过Mybatis拦截器自动定位慢SQL并记录日志
05-30
下面我们将详细介绍如何通过Mybatis拦截器实现这一功能。 首先，了解Mybatis拦截器的基本概念。Mybatis拦截器（Interceptor）是一种插件机制，它允许我们在Mybatis执行SQL语句之前或之后进行自定义操作，比如统计...
MyBatis拦截器:给参数对象属性赋值的实例
08-30
MyBatis拦截器是MyBatis框架中的一种插件机制，允许用户自定义代码来扩展MyBatis的功能。在这个特定的实例中，我们讨论的是一个用于给参数对象属性赋值的拦截器。这个拦截器的主要目标是在执行增删改操作时，自动为...
Mybatis动态调用表名和字段名的解决方法
09-01
今天在项目开发中有个业务是需要限制各个用户对某些表里的字段查询以及某些字段是否显示，这种情况下，就需要构建sql来动态传入表名、字段名了,下面给大家介绍mybatis动态调用表名和字段名的解决方法，一起看看吧
mybatis拦截器修改执行sql语句
09-11
通过mybatis拦截器将查询语句、更新语句、删除语句、插入语句中指定表明替换为另一个表名
MyBatis拦截器分页与动态修改SQL及其参数值
10-09
在"MyBatis拦截器分页与动态修改SQL及其参数值"的主题中，我们可以深入理解以下几个关键知识点： 1. **MyBatis拦截器**：MyBatis提供了一种插件机制，即拦截器（Interceptor），它基于Java的动态代理，可以在SQL...
MyBatis拦截器修改sql表名
weixin_50545075的博客
03-28 4152
通过Mybatis拦截器、自定义注解、获取参数来修改表名
MybatisPlus动态表名
qq_44309969的博客
04-29 2500
mybatisplus动态表名
mybatis获取表名_mybatis plus 的动态表名的配置详解
weixin_33608760的博客
12-28 6133
mybatis plus简介详见mybatis plus的官网业务要求业务上要求存储数据的时候根据某个字段动态的选择数据要存储的表.如根据code字段: code->[1001,1002]来进行选择存储的表:经过下面的配置实现动态表名如 --> table1_1001,table_1002的效果.以此动态生成表名的效果.具体实现MPConfig.javaimport com.aliba...
MyBatis 获取数据库所有表名
jonathan_joestar的博客
11-04 3212
2022-11-04 12:54:26.014 INFO 26320 — [nio-8003-exec-3] c.mindskip.xzs.utility.ControllerAspect : 方法执行完毕, 共用时:94毫秒。MyBatis 获取数据库所有表名。“message”: “成功”,
Mybatis动态替换表名-拦截器
qq_35019521的博客
09-21 2723
mybatis动态替换表名-拦截器
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值