本博客日IP超过2000，PV 3000 左右，急需赞助商。
极客时间所有课程通过我的二维码购买后返现24元微信红包，请加博主新的微信号：xttblog2，之前的微信号好友位已满，备注：返现
受密码保护的文章请关注“业余草”公众号，回复关键字“0”获得密码
所有面试题(java、前端、数据库、springboot等)一网打尽，请关注文末小程序
腾讯云】1核2G5M轻量应用服务器50元首年，高性价比，助您轻松上云
mybatis-plus还可以这样分表
为什么要分表
Mysql是当前互联网系统中使用非常广泛的关系数据库，具有ACID的特性。
但是mysql的单表性能会受到表中数据量的限制，主要原因是B+树索引过大导致查询时索引无法全部加载到内存。读取磁盘的次数变多，而磁盘的每次读取对性能都有很大的影响。
这时一个简单可行的方案就是分表(当然土豪也可以堆硬件)，将一张数据量庞大的表的数据，拆分到多个表中，这同时也减少了B+树索引的大小，减少磁盘读取次数，提高性能。
两种基础分表逻辑
说完了为什么要分表，下面聊聊业务开发中常见的两种基础的分表逻辑。
按日期分表
这种方式通常会在表名的最后加上年月日，主要适用于按日期划分的统计数据或操作记录。在线实时展示的只有最近表中的数据，其他数据用于离线统计等。
按id取模分表
这种方式需要一个id生成器，例如snowflake id或分布式id服务。它保证了相同id的数据都在一张表中，主要适用于保存用户基础信息，系统中的资源信息，购买记录等。当然这种分表方式扩展性较差，后期数据持续增多后需要按id大小分库再分表处理。
下面看下这两种分表逻辑在mybatis-plus中的实现。
Mybatis-plus中的分表实现
说到java的分表中间件，可能有人会想到sharding-jdbc，作为使用很广泛的一个分表中间件，功能也比较完善，但是使用它需要引入额外的jar包和增加学习成本。
实际上mybatis-plus本身就提供了一个分表的解决方案，配置使用都很简单，适合快速开发系统。
动态表名处理器
没错，mybatis-plus提供了动态表名处理器接口TableNameHandler，只需要在系统中实现该接口，并作为插件加载到mybatis-plus中就可以使用，下面来看下详细的步骤。
❝
3.4版本之前的动态表名接口是ITableNameHandler，需要和分页插件配合使用。
3.4版本新增了TableNameHandler，在方法参数上取消了MetaObject。这里用最新的版本为例，使用方式差别不大。❞
假设我们的系统中有两种分表方式，按日期分表和按id取模分表。通过四个步骤来看下具体的使用示例。
创建日期表名处理器
先来看下日期处理的表名处理器，实现TableNameHandler接口后，在dynamicTableName方法中实现动态生成表名的逻辑，方法的返回值就是查询时要使用的表名。
/**
* 按天分表解析
*/
public class DaysTableNameParser implements TableNameHandler {
@Override
public String dynamicTableName(String sql, String tableName) {
String dateDay = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"));
return tableName + "_" + dateDay;
}
}创建id取模表名处理器
再来看下按id取模表名处理器的实现，这个处理器相对日期处理就要复杂一些，主要原因为需要动态传入用于分表的id值。
在之前的版本中可以在方法中通过解析MetaObject中带有的sql查询信息，获取分表使用的值。但是这种方式比较复杂，对于不同的QueryMapper分析的方式不同，比较容易出错。新版本中的方法取消了MetaObject参数，需要使用其他方式传入。
需要注意的是，表名处理器是作为mybatis-plus的插件，在项目启动时实例化的。这意味着，在运行过程中只有一个对象，多线程处理过程中，一个线程对参数的修改，会影响到其他线程。为了解决这个问题，可以使用ThreadLocal来定义参数。
由于现在的框架中大部分会使用线程池，例如springboot web项目中的tomcat。所以在每次使用后，需要手动清除本次数据，防止线程复用时的影响。
具体实现如下:
/**
* 按id取模分表处理器
*/
public class IdModTableNameParser implements TableNameHandler {
private Integer mod;
//使用ThreadLocal防止多线程相互影响
private static ThreadLocal<Integer> id = new ThreadLocal<Integer>();
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;
}
}
}加载表名处理器
表名处理器实际是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<String, TableNameHandler> map = new HashMap<String, TableNameHandler>();
//这里为不同的表设置对应表名处理器
map.put("user_daily_record", new DaysTableNameParser());
map.put("user_consume_flow", new IdModTableNameParser(10));
dynamicTableNameInnerInterceptor.setTableNameHandlerMap(map);
interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
return interceptor;
}
}在controller中使用
下面通过controller中的三个接口，展示下使用方式：
@RestController
public class TableTestController {
@Resource
IUserDailyRecordService userDailyRecordService;
@Resource
IUserConsumeFlowService userConsumeFlowService;
@GetMapping("user/record/today")
public CommonResVo<UserDailyRecord> getRecordToday(Integer userId) throws Exception {
//这里在查询时，会根据系统当前时间，自动生成当天的表名
UserDailyRecord userDailyRecord = userDailyRecordService.getOne(new LambdaQueryWrapper<UserDailyRecord>().eq(UserDailyRecord::getUserId, userId));
return CommonResVo.success(userDailyRecord);
}
@GetMapping("user/consume/flow")
public CommonResVo<List<UserConsumeFlow>> getConsumeFlow(Integer userId) throws Exception {
//设置用于分表的id值
IdModTableNameParser.setId(userId);
List<UserConsumeFlow> userConsumeFlowList = userConsumeFlowService.list(new LambdaQueryWrapper<UserConsumeFlow>().eq(UserConsumeFlow::getUserId, userId));
return CommonResVo.success(userConsumeFlowList);
}
/**
* 新增数据
*/
@PostMapping("user/consume/flow")
public CommonResVo<Boolean> addConsumeFlow(@RequestBody UserConsumeFlow userConsumeFlow) throws Exception {
Integer userId = userConsumeFlow.getUserId();
//设置用于分表的id值
IdModTableNameParser.setId(userId);
userConsumeFlowService.save(userConsumeFlow);
return CommonResVo.success(true);
}
}这篇对mybatis-plus动态表名处理器的介绍，通过实现TableNameHandler接口，可以按实际情况灵活定义表名的生成规则，希望对大家有帮助。
最后，欢迎关注我的个人微信公众号：业余草（yyucao）！可加作者微信号：xttblog2。备注：“1”，添加博主微信拉你进微信群。备注错误不会同意好友申请。再次感谢您的关注！后续有精彩内容会第一时间发给您！原创文章投稿请发送至532009913@qq.com邮箱。商务合作也可添加作者微信进行联系！
本文原文出处：业余草： » Mybatis-Plus中竟然还可以这样分表