基于SpringBoot+Mybatis实现Mysql分表

2025-04-07 16:50

本文主要是介绍基于SpringBoot+Mybatis实现Mysql分表,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

《基于SpringBoot+Mybatis实现Mysql分表》这篇文章主要为大家详细介绍了基于SpringBoot+Mybatis实现Mysql分表的相关知识,文中的示例代码讲解详细,感兴趣的小伙伴可...

基本思路

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

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

定义注解

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

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

    // 表名前缀
    String prefix();

}
@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

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 = pythonentityClass;
        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;
    }
}

创建拦截器

@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<?> epythonntityClass = 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.getParameterHHYHMOandler();
            parameterHandler.setParameters(preparedStatement);

            preparedStatement.executeUpdate();
        } finally {
            if (preparedStatement != null) {
                preparedStatement.close()China编程;
            }
            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) {
    }
}

获取表名

@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放入当前线程中,后面要根据这些信息获取实际要操作的表名

public void insert(ServiceOrderLogEntity serviceOrderLogEntity) {
    ShardingContext.setContext(ServiceOrderLogEntity.class, serviceOrderLogEntity.getTime() == null ? new Date() : serviceOrderLogEntity.getTime());
    
    int result = serviceOrderLogMapper.insert(serviceOrderLogEntity);
    
    ShardingContext.clearContext();
}

到此这篇关于基于SpringBoot+Mybatis实现mysql分表的文章就介绍到这了,更多相关SpringBoot Mysql分表内容请搜索China编程(www.chinasem.cn)以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程China编程(www.chinasem.cn)!

这篇关于基于SpringBoot+Mybatis实现Mysql分表的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1154111

相关文章

java使用protobuf-maven-plugin的插件编译proto文件详解

《java使用protobuf-maven-plugin的插件编译proto文件详解》:本文主要介绍java使用protobuf-maven-plugin的插件编译proto文件,具有很好的参考价... 目录protobuf文件作为数据传输和存储的协议主要介绍在Java使用maven编译proto文件的插件

c++ 类成员变量默认初始值的实现

《c++类成员变量默认初始值的实现》本文主要介绍了c++类成员变量默认初始值,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧... 目录C++类成员变量初始化c++类的变量的初始化在C++中,如果使用类成员变量时未给定其初始值,那么它将被

Java中的数组与集合基本用法详解

《Java中的数组与集合基本用法详解》本文介绍了Java数组和集合框架的基础知识,数组部分涵盖了一维、二维及多维数组的声明、初始化、访问与遍历方法,以及Arrays类的常用操作,对Java数组与集合相... 目录一、Java数组基础1.1 数组结构概述1.2 一维数组1.2.1 声明与初始化1.2.2 访问

Javaee多线程之进程和线程之间的区别和联系(最新整理)

《Javaee多线程之进程和线程之间的区别和联系(最新整理)》进程是资源分配单位,线程是调度执行单位,共享资源更高效,创建线程五种方式:继承Thread、Runnable接口、匿名类、lambda,r... 目录进程和线程进程线程进程和线程的区别创建线程的五种写法继承Thread,重写run实现Runnab

Java 方法重载Overload常见误区及注意事项

《Java方法重载Overload常见误区及注意事项》Java方法重载允许同一类中同名方法通过参数类型、数量、顺序差异实现功能扩展,提升代码灵活性,核心条件为参数列表不同,不涉及返回类型、访问修饰符... 目录Java 方法重载(Overload)详解一、方法重载的核心条件二、构成方法重载的具体情况三、不构

Java通过驱动包(jar包)连接MySQL数据库的步骤总结及验证方式

《Java通过驱动包(jar包)连接MySQL数据库的步骤总结及验证方式》本文详细介绍如何使用Java通过JDBC连接MySQL数据库,包括下载驱动、配置Eclipse环境、检测数据库连接等关键步骤,... 目录一、下载驱动包二、放jar包三、检测数据库连接JavaJava 如何使用 JDBC 连接 mys

SpringBoot线程池配置使用示例详解

《SpringBoot线程池配置使用示例详解》SpringBoot集成@Async注解,支持线程池参数配置(核心数、队列容量、拒绝策略等)及生命周期管理,结合监控与任务装饰器,提升异步处理效率与系统... 目录一、核心特性二、添加依赖三、参数详解四、配置线程池五、应用实践代码说明拒绝策略(Rejected

SQL中如何添加数据(常见方法及示例)

《SQL中如何添加数据(常见方法及示例)》SQL全称为StructuredQueryLanguage,是一种用于管理关系数据库的标准编程语言,下面给大家介绍SQL中如何添加数据,感兴趣的朋友一起看看吧... 目录在mysql中,有多种方法可以添加数据。以下是一些常见的方法及其示例。1. 使用INSERT I

Qt使用QSqlDatabase连接MySQL实现增删改查功能

《Qt使用QSqlDatabase连接MySQL实现增删改查功能》这篇文章主要为大家详细介绍了Qt如何使用QSqlDatabase连接MySQL实现增删改查功能,文中的示例代码讲解详细,感兴趣的小伙伴... 目录一、创建数据表二、连接mysql数据库三、封装成一个完整的轻量级 ORM 风格类3.1 表结构

基于Python实现一个图片拆分工具

《基于Python实现一个图片拆分工具》这篇文章主要为大家详细介绍了如何基于Python实现一个图片拆分工具,可以根据需要的行数和列数进行拆分,感兴趣的小伙伴可以跟随小编一起学习一下... 简单介绍先自己选择输入的图片,默认是输出到项目文件夹中,可以自己选择其他的文件夹,选择需要拆分的行数和列数,可以通过