Mybatis-plus使用TableNameHandler分表详解(附完整示例源码)

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习数据库相关编程知识。下面本篇文章就来带大家聊聊《Mybatis-plus使用TableNameHandler分表详解(附完整示例源码)》，介绍一下MySQL、Java、springboot，希望对大家的知识积累有所帮助，助力实战开发！

为什么要分表

Mysql是当前互联网系统中使用非常广泛的关系数据库，具有ACID的特性。

但是mysql的单表性能会受到表中数据量的限制，主要原因是B+树索引过大导致查询时索引无法全部加载到内存。读取磁盘的次数变多，而磁盘的每次读取对性能都有很大的影响。

这时一个简单可行的方案就是分表(当然土豪也可以堆硬件)，将一张数据量庞大的表的数据，拆分到多个表中，这同时也减少了B+树索引的大小，减少磁盘读取次数，提高性能。

两种基础分表逻辑

说完了为什么要分表，下面聊聊业务开发中常见的两种基础的分表逻辑。

按日期分表

这种方式通常会在表名的最后加上年月日，主要适用于按日期划分的统计数据或操作记录。在线实时展示的只有最近表中的数据，其他数据用于离线统计等。

按id取模分表

这种方式需要一个id生成器，例如snowflake id或分布式id服务。它保证了相同id的数据都在一张表中，主要适用于保存用户基础信息，系统中的资源信息，购买记录等。当然这种分表方式扩展性较差，后期数据持续增多后需要按id大小分库再分表处理。

下面看下这两种分表逻辑在mybatis-plus中的实现。

Mybatis-plus中的分表实现

说到java的分表中间件，可能有人会想到sharding-jdbc，作为使用很广泛的一个分表中间件，功能也比较完善，但是使用它需要引入额外的jar包和增加学习成本。

实际上mybatis-plus本身就提供了一个分表的解决方案，配置使用都很简单，适合快速开发系统。

动态表名处理器

没错，mybatis-plus提供了动态表名处理器接口

/**
 * 按天分表解析
 */
public class DaysTableNameParser implements TableNameHandler {

    @Override
    public String dynamicTableName(String sql, String tableName) {
        String dateDay = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"));
        return tableName + "_" + dateDay;
    }
}

2.创建id取模表名处理器

再来看下按id取模表名处理器的实现，这个处理器相对日期处理就要复杂一些，主要原因为需要动态传入用于分表的id值。

在之前的版本中可以在方法中通过解析

/**
 * 按id取模分表处理器
 */
public class IdModTableNameParser implements TableNameHandler {
    private Integer mod;

    //使用ThreadLocal防止多线程相互影响
    private static ThreadLocal id = new ThreadLocal();

    public static void setId(Integer idValue) {
        id.set(idValue);
    }

    IdModTableNameParser(Integer modValue) {
        mod = modValue;
    }

    @Override
    public String dynamicTableName(String sql, String tableName) {
        Integer idValue = id.get();
        if (idValue == null) {
            throw new RuntimeException("请设置id值");
        } else {
            String suffix = String.valueOf(idValue % mod);
            //这里清除ThreadLocal的值，防止线程复用出现问题
            id.set(null);
            return tableName + "_" + suffix;
        }
    }
}

3.加载表名处理器

表名处理器实际是mybatis-plus的插件，需要在初始化时创建实例并加载。因为系统中存在两种分表类型，在初始化时可以指定每张表使用的表名处理器。具体实现如下：

@Configuration
@MapperScan(basePackages = "com.yourcom.proname.repository.mapper.mainDb*", sqlSessionFactoryRef = "mainSqlSessionFactory")
public class MainDb {
    @Bean(name = "mainDataSource")
    @ConfigurationProperties(prefix = "dbconfig.maindb")
    public DataSource druidDataSource() {
        return DruidDataSourceBuilder.create().build();
    }

    @Bean(name = "mainTransactionManager")
    public DataSourceTransactionManager masterTransactionManager(@Qualifier(value = "mainDataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "mainSqlSessionFactory")
    @ConfigurationPropertiesBinding()
    public SqlSessionFactory sqlSessionFactory(@Qualifier(value = "mainDataSource") DataSource dataSource) throws Exception {
        MybatisSqlSessionFactoryBean factoryBean = new MybatisSqlSessionFactoryBean();
        factoryBean.setDataSource(dataSource);
          //加载插件
        factoryBean.setPlugins(mybatisPlusInterceptor());
        return factoryBean.getObject();
    }

    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
        HashMap map = new HashMap();

        //这里为不同的表设置对应表名处理器
        map.put("user_daily_record", new DaysTableNameParser());
        map.put("user_consume_flow", new IdModTableNameParser(10));

        dynamicTableNameInnerInterceptor.setTableNameHandlerMap(map);
        interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
        return interceptor;
    }
}

4.在controller中使用

下面通过controller中的三个接口，展示下使用方式：

@RestController
public class TableTestController {
    @Resource
    IUserDailyRecordService userDailyRecordService;

    @Resource
    IUserConsumeFlowService userConsumeFlowService;

    @GetMapping("user/record/today")
    public CommonResVo getRecordToday(Integer userId) throws Exception {
        //这里在查询时，会根据系统当前时间，自动生成当天的表名
        UserDailyRecord userDailyRecord = userDailyRecordService.getOne(new LambdaQueryWrapper().eq(UserDailyRecord::getUserId, userId));
        return CommonResVo.success(userDailyRecord);
    }

    @GetMapping("user/consume/flow")
    public CommonResVo> getConsumeFlow(Integer userId) throws Exception {
        //设置用于分表的id值
        IdModTableNameParser.setId(userId);
        List userConsumeFlowList = userConsumeFlowService.list(new LambdaQueryWrapper().eq(UserConsumeFlow::getUserId, userId));
        return CommonResVo.success(userConsumeFlowList);
    }

    /**
     * 新增数据
     */
    @PostMapping("user/consume/flow")
    public CommonResVo addConsumeFlow(@RequestBody UserConsumeFlow userConsumeFlow) throws Exception {
        Integer userId = userConsumeFlow.getUserId();
        //设置用于分表的id值
        IdModTableNameParser.setId(userId);
        userConsumeFlowService.save(userConsumeFlow);
        return CommonResVo.success(true);
    }
}

这篇对mybatis-plus动态表名处理器的介绍，通过实现

TableNameHandler

项目完整示例地址：https://gitee.com/dothetrick/...

以上内容属个人学习总结，如有不当之处，欢迎在评论中指正

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于数据库的相关知识，也可关注golang学习网公众号。