SpringBoot+Mybatis实现Mysql分表

基本思路

  • 根据创建时间字段按年进行分表,比如日志表log可以分为log_2024、log_2025

  • 在需要进行插入、更新操作的地方利用threadlocal将数据表对应的Entity.class创建时间放入当前的线程中,利用mybatis提供的拦截器在sql执行前进行拦截,将threadlocal中的Class类取出,根据类上标注的注解获取要操作的表名,再利用创建时间得到最终要操作的实际表名,最后更换sql中的表名让拦截器继续执行

定义注解

定义注解@ShardedTable,将该注解标注在数据表对应的Entity类上,比如User类上

java 复制代码
/**
 * 分表注解
 */
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
public @interface ShardedTable {

    // 表名前缀
    String prefix();

}
java 复制代码
@ShardedTable(prefix = "user")
@TableName("user")
public class User {

    @TableId(type = IdType.AUTO)
    private Integer id;

    private String name;

    private Integer age;

    public User(String name, Integer age) {
        this.name = name;
        this.age = age;
    }
}

创建ThreadLocal

java 复制代码
public class ShardingContext{
    private static final ThreadLocal<ShardingContext> CONTEXT = new ThreadLocal<>();

    private Class<?> entityClass; // 数据表对应的实体类
    private Date date;

    public static void setContext(Class<?> entityClass, Date date) {
        ShardingContext context = new ShardingContext();
        context.entityClass = entityClass;
        context.date = date;
        CONTEXT.set(context);
    }

    public static ShardingContext getContext() {
        return CONTEXT.get();
    }

    public static void clearContext() {
        CONTEXT.remove();
    }

    public Class<?> getEntityClass() {
        return entityClass;
    }

    public Date getDate() {
        return date;
    }
}

创建拦截器

java 复制代码
@Component
@Intercepts({@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class,Integer.class})})
public class ShardingInterceptor implements Interceptor {

    @Autowired
    private ShardingStrategy shardingStrategy;

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        // 获取原始SQL
        StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
        BoundSql boundSql = statementHandler.getBoundSql();
        String originalSql =  boundSql.getSql();

        // 获取当前操作的实体类
        ShardingContext context = ShardingContext.getContext();
        if (context != null){
            Class<?> entityClass = context.getEntityClass();
            Date date = context.getDate();
            ShardedTable annotation = entityClass.getAnnotation(ShardedTable.class);
            if (annotation != null) {
                // 设置新的sql,替换表名
                String baseTableName = annotation.prefix();
                String actualTableName = shardingStrategy.getTableName(User.class, date);
                String modifiedSql = originalSql.replace(baseTableName, actualTableName);
                setSql(boundSql, modifiedSql);

                // 将数据保存到原表,作为备份
                executeBackupInsert(statementHandler,originalSql);
            }
        }

        return invocation.proceed();
    }

    private void setSql(BoundSql boundSql, String sql) throws Exception {
        Field field = BoundSql.class.getDeclaredField("sql");
        field.setAccessible(true);
        field.set(boundSql, sql);
    }

    // 同时将数据保存到原表,作为备份
    private void executeBackupInsert(StatementHandler statementHandler, String backupSql) throws SQLException {
        Connection connection = null;
        PreparedStatement preparedStatement = null;
        try {
            // 通过反射获取 MappedStatement
            MetaObject metaObject = SystemMetaObject.forObject(statementHandler);
            MappedStatement mappedStatement = (MappedStatement) metaObject.getValue("delegate.mappedStatement");

            connection = mappedStatement.getConfiguration().getEnvironment().getDataSource().getConnection();
            preparedStatement = connection.prepareStatement(backupSql);

            // 设置参数
            ParameterHandler parameterHandler = statementHandler.getParameterHandler();
            parameterHandler.setParameters(preparedStatement);

            preparedStatement.executeUpdate();
        } finally {
            if (preparedStatement != null) {
                preparedStatement.close();
            }
            if (connection != null) {
                connection.close();
            }
        }
    }

    @Override
    public Object plugin(Object target) {
        // 判断是否为StatementHandler类型
        if (target instanceof StatementHandler){
            return Plugin.wrap(target, this);
        }else {
            return target;
        }
    }

    @Override
    public void setProperties(Properties properties) {
    }
}

获取表名

java 复制代码
@Component
public class ShardingStrategy {

    public String getTableName(Class<?> entityClass, Date date) {
        ShardedTable annotation = entityClass.getAnnotation(ShardedTable.class);
        if (annotation == null) {
            throw new RuntimeException("实体类必须使用@ShardedTable注解");
        }
        // 获取分表前缀
        String tablePrefix = annotation.prefix();
        if (tablePrefix == null || tablePrefix.isEmpty()) {
            throw new RuntimeException("分表前缀不能为空");
        }
        // 获取当前日期所在的年份
        int year = DateUtil.year(date);
        return tablePrefix + "_" + year;
    }
}

业务处理

在需要进行业务处理的地方,将数据表对应的Entity.class创建时间通过threadlocal放入当前线程中,后面要根据这些信息获取实际要操作的表名

java 复制代码
public void insert(ServiceOrderLogEntity serviceOrderLogEntity) {
    ShardingContext.setContext(ServiceOrderLogEntity.class, serviceOrderLogEntity.getTime() == null ? new Date() : serviceOrderLogEntity.getTime());
    
    int result = serviceOrderLogMapper.insert(serviceOrderLogEntity);
    
    ShardingContext.clearContext();
}
相关推荐
爱海贼的无处不在4 分钟前
一个需求竟然要开14个会:程序员的日常到底有多“会”?
后端·程序员
IT_陈寒1 小时前
Java 性能优化:5个被低估的JVM参数让你的应用吞吐量提升50%
前端·人工智能·后端
南囝coding1 小时前
《独立开发者精选工具》第 018 期
前端·后端
绝无仅有1 小时前
数据库MySQL 面试之死锁与排查经验总结
后端·面试·github
用户384958730692 小时前
Spring Boot 集成 Redis 的完整流程
后端
昨日的风2 小时前
springboot 多数据源切换
后端
绝无仅有2 小时前
mysql性能优化实战与总结
后端·面试·github
用户8356290780512 小时前
从手动编辑到代码生成:Python 助你高效创建 Word 文档
后端·python
德育处主任3 小时前
玩转 Strands:AI Agent 开发,原来可以这么简单!
后端·aigc
Undoom3 小时前
大模型选型“炼狱”与终结:一份来自普通开发者的AI Ping深度评测报告
后端