解决使用Mybatis-FLex时，制作Mybatis原生插件导致的动态SQL参数无法正常填入的问题

最新推荐文章于 2024-08-19 16:34:35 发布

Vzhangs

最新推荐文章于 2024-08-19 16:34:35 发布

阅读量387

点赞数 8

文章标签： mybatis sql 数据库

本文链接：https://blog.csdn.net/weixin_42277430/article/details/140842857

版权

背景

业务中使用Postgresql的Schema进行租户隔离，但是Mybatis-FLex的动态表名/Schema名只支持通过Mybatis-FLex的Java代码生成的SQL的动态替换，对直接写在xml里的动态SQL不支持。因此自己实现了一个可以支持xml里的动态Schema名替换插件。

import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.plugin.*;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.reflection.SystemMetaObject;

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

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

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
        MetaObject mo = SystemMetaObject.forObject(statementHandler);
        BoundSql boundSql = (BoundSql) mo.getValue("boundSql");
        String sqlStr = boundSql.getSql();

        // 获取动态Schema
        String dynamicTableName = tenantSchemaName(sqlStr);
        // 替换原始SQL中的Schema
        String modifiedSql = sqlStr.replaceAll("@tenant", "\"" + dynamicTableName + "\"");

        // 反射设置修改后的SQL
        MetaObject boundSqlMo = SystemMetaObject.forObject(boundSql);
        boundSqlMo.setValue("sql", modifiedSql);

        // 继续执行后续操作
        return invocation.proceed();
    }

    private String tenantSchemaName(String originalSql) {
        return AuthUser.current().getTenant();
    }

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

    @Override
    public void setProperties(Properties properties) {
//        properties
        // 可以接收配置的属性
    }
}

在Mybatis的Spring Configuration类中，注册这个插件

@Bean
public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception {
    SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
    sessionFactory.setDataSource(dataSource);

    // 注册自定义拦截器
    sessionFactory.addPlugins(new TenantSchemaInterceptor());

    return sessionFactory.getObject();
}

修改后，无论是通过MybatisFlex的方式，还是通过Mybatis xml的方式提交的SQL，均报错无法填入参数1 一类的错误。

一开始以为是自己的插件写的有问题，排查一番，即便不将这个插件加入sessionFactory中，也会报这个错。

于是想到，MybatisFlex是不是没有使用mybatis.spring的SqlSessionFactoryBean 查证一番，MybatisFlex确实使用的是继承自SqlSessionFactoryBean 的 FlexSqlSessionFactoryBean

将SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
替换为SqlSessionFactoryBean sessionFactory = new FlexSqlSessionFactoryBean();
问题解决