多元数据源无忧切换，Mybatis轻驾驾驭

2022-11-27 06:14:24

单数据源下的多数据库切换：Mybatis拦截器的威力

在数据管理的浩瀚领域中，Mybatis拦截器犹如一盏明灯，照亮了单数据源下多数据库切换的崎岖道路。对于物流分拣等数据量庞大、要求极高的行业，它无疑是应对数据挑战的利器。

多数据库切换的痛点

在传统的单数据源环境下，分拣业务需要将所有数据存储在一个巨大的表中。随着数据的不断累积，查询速度不可避免地会下降。同时，为实现分库管理，物流公司往往会创建多个具有相同表的不同数据库。切换这些数据库以进行查询或修改操作，不仅增加了开发人员的工作量，也降低了系统的整体效率。

Mybatis拦截器的妙用

Mybatis拦截器应运而生，宛如一把瑞士军刀，解决多数据库切换的难题。它是一种拦截Mybatis执行SQL语句的工具，能够根据需求修改或替换SQL语句。利用Mybatis拦截器，我们可以实现单数据源内多数据库的无缝切换。

实现步骤

第一步：引入依赖

在项目的pom.xml文件中，引入Mybatis拦截器的依赖：

<dependency>
    <groupId>org.mybatis</groupId>
    <artifactId>mybatis-spring-boot-starter</artifactId>
    <version>2.2.2</version>
</dependency>

第二步：编写拦截器

创建DatabaseSwitchInterceptor类，用于实现数据库的切换：

import org.apache.ibatis.executor.parameter.ParameterHandler;
import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.plugin.*;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.session.Configuration;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import org.apache.ibatis.type.TypeHandlerRegistry;

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

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

    private static final String DATABASE_KEY = "databaseKey";

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
        MetaObject metaObject = MetaObject.forObject(statementHandler, null);
        MappedStatement mappedStatement = (MappedStatement) metaObject.getValue("delegate.mappedStatement");
        String databaseKey = (String) mappedStatement.getSqlCommandType().name().toLowerCase();
        Connection connection = (Connection) invocation.getArgs()[0];
        connection.setCatalog(databaseKey);
        return invocation.proceed();
    }

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

    @Override
    public void setProperties(Properties properties) {

    }
}

第三步：注册拦截器

在Spring Boot的配置文件中，注册Mybatis拦截器：

@Configuration
public class MybatisConfig {

    @Bean
    public DatabaseSwitchInterceptor databaseSwitchInterceptor() {
        return new DatabaseSwitchInterceptor();
    }

    @Bean
    public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception {
        SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
        factoryBean.setDataSource(dataSource);
        factoryBean.setPlugins(new Interceptor[]{databaseSwitchInterceptor()});
        return factoryBean.getObject();
    }
}

第四步：使用拦截器

在SQL语句中，只需指定要切换的数据库即可：

@Mapper
public interface UserMapper {

    @Select("select * from user where id = #{id}")
    User getUserById(@Param("id") Long id);

    @Select("select * from user where databaseKey = #{databaseKey} and id = #{id}")
    User getUserByIdAndDatabaseKey(@Param("databaseKey") String databaseKey, @Param("id") Long id);
}