mybatis mysql分库分表

基础概念

MyBatis 是一个优秀的持久层框架，它支持定制化 SQL、存储过程以及高级映射。MySQL 是一种关系型数据库管理系统，广泛应用于各种规模的应用系统中。分库分表是一种数据库优化策略，用于解决单点数据库的性能瓶颈问题。

分库分表的优势

提升性能：通过将数据分散到多个数据库或表中，可以显著提高查询和写入的性能。
扩展性：分库分表使得系统更容易进行水平扩展，能够应对数据量快速增长的需求。
高可用性：通过数据冗余和负载均衡，可以提高系统的可用性和容灾能力。

分库分表的类型

垂直分库：根据业务功能将不同的表划分到不同的数据库中。
水平分表：在同一数据库内，根据某种规则（如哈希、范围等）将数据分散到多个表中。
垂直分表：在同一数据库内，根据字段的访问频率或业务逻辑将表的字段拆分到不同的表中。

应用场景

高并发场景：当系统面临大量并发请求时，分库分表可以有效减轻单点数据库的压力。
大数据量场景：随着数据量的增长，单个数据库或表可能无法满足性能需求，此时需要进行分库分表。
需要高可用性的系统：通过分库分表可以提高系统的容灾能力和可用性。

实现 MyBatis + MySQL 分库分表

配置 MyBatis

首先，在 MyBatis 的配置文件中配置多个数据源：

<environments default="development">
    <environment id="development">
        <transactionManager type="JDBC"/>
        <dataSource type="POOLED">
            <property name="driver" value="com.mysql.jdbc.Driver"/>
            <property name="url" value="jdbc:mysql://localhost:3306/db1"/>
            <property name="username" value="user1"/>
            <property name="password" value="password1"/>
        </dataSource>
    </environment>
    <environment id="development2">
        <transactionManager type="JDBC"/>
        <dataSource type="POOLED">
            <property name="driver" value="com.mysql.jdbc.Driver"/>
            <property name="url" value="jdbc:mysql://localhost:3306/db2"/>
            <property name="username" value="user2"/>
            <property name="password" value="password2"/>
        </dataSource>
    </environment>
</environments>

动态数据源切换

可以通过 AOP 或者 ThreadLocal 来实现动态数据源切换：

public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDataSource();
    }
}

public class DataSourceContextHolder {
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

    public static void setDataSource(String dataSource) {
        contextHolder.set(dataSource);
    }

    public static String getDataSource() {
        return contextHolder.get();
    }

    public static void clearDataSource() {
        contextHolder.remove();
    }
}

在需要切换数据源的地方，设置相应的数据源标识：

public void someMethod() {
    try {
        DataSourceContextHolder.setDataSource("db1");
        // 执行数据库操作
    } finally {
        DataSourceContextHolder.clearDataSource();
    }
}

分表策略

可以通过 MyBatis 的拦截器（Interceptor）来实现分表逻辑：

@Intercepts({@Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class})})
public class ShardingInterceptor implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        MappedStatement ms = (MappedStatement) invocation.getArgs()[0];
        Object parameter = invocation.getArgs()[1];
        BoundSql boundSql = ms.getBoundSql(parameter);
        String sql = boundSql.getSql();

        // 根据业务逻辑修改 SQL 中的表名
        String newSql = modifyTableName(sql);

        // 创建新的 BoundSql
        BoundSql newBoundSql = new BoundSql(ms.getConfiguration(), newSql, boundSql.getParameterMappings(), boundSql.getParameterObject());

        // 替换 MappedStatement 中的 BoundSql
        MappedStatement newMs = newMappedStatement(ms, new BoundSqlStatementHandler(newBoundSql));
        invocation.getArgs()[0] = newMs;

        return invocation.proceed();
    }

    private String modifyTableName(String sql) {
        // 根据业务逻辑修改表名
        return sql.replace("original_table", "shard_table_1");
    }

    private MappedStatement newMappedStatement(MappedStatement ms, StatementHandler newStatementHandler) {
        MappedStatement.Builder builder = new MappedStatement.Builder(ms.getConfiguration(), ms.getId(), new BoundSqlStatementHandler(newStatementHandler), ms.getSqlCommandType());
        builder.resource(ms.getResource());
        builder.fetchSize(ms.getFetchSize());
        builder.statementType(ms.getStatementType());
        builder.keyGenerator(ms.getKeyGenerator());
        if (ms.getKeyProperties() != null && ms.getKeyProperties().length != 0) {
            builder.keyProperty(String.join(",", ms.getKeyProperties()));
        }
        builder.timeout(ms.getTimeout());
        builder.parameterMap(ms.getParameterMap());
        builder.resultMaps(ms.getResultMaps());
        builder.resultSetType(ms.getResultSetType());
        builder.cache(ms.getCache());
        builder.flushCacheRequired(ms.isFlushCacheRequired());
        builder.useCache(ms.isUseCache());
        return builder.build();
    }
}