记录Mybatis-plus动态表名的实现
实际上Mybatis-plus官方本来就有提供对动态表名的扩展支持以及给出了具体的Sample。
具体参考官方的sample github。从github的提交历史可以追溯到最开始给出动态表名实现方案的版本是3.1.1,奈何我司所用的版本刚好卡在3.1.0,差那么一个小版本,直接拿官方的例子用是会报错的。
@Configuration
@MapperScan("com.baomidou.mybatisplus.samples.dytablename.mapper")
public class MybatisPlusConfig {
@Bean
public PaginationInterceptor paginationInterceptor() {
PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
DynamicTableNameParser dynamicTableNameParser = new DynamicTableNameParser();
dynamicTableNameParser.setTableNameHandlerMap(new HashMap<String, ITableNameHandler>(2) {{
put("user", (metaObject, sql, tableName) -> {
String year = "_2018";
int random = new Random().nextInt(1);
if (random == 1) {
year = "_2019";
}
return tableName + year;
});
}});
paginationInterceptor.setSqlParserList(Collections.singletonList(dynamicTableNameParser));
return paginationInterceptor;
}
}
因为相关的支持,都是在3.1.1加进去的,所幸的是,3.1.1加进去的扩展所依赖的核心接口并没有改变,所以理论上是可以把3.1.1的扩展类代码copy到自己的项目,这样在3.1.0的版本就可与你实现动态表名了。具体涉及到的扩展类包括:
import com.baomidou.mybatisplus.extension.parsers.DynamicTableNameParser;
import com.baomidou.mybatisplus.extension.parsers.ITableNameHandler;
官方给出的只是一个随机表名,实际业务往往还有很多问题:
-
如何在调用mapper方法时传递我到底要决定使用什么表名?
最开始我琢磨了那个metaObject半天,期望从它入手,但并没有一个很通用的方式,而且在后续的版本中,这个metaObject甚至被移出了这个接口。后来才想到,这里最优的实现只能使用ThreadLocal了。
image.png -
如果使用ThreadLocal,那就意味着每次调用mapper方法之前都要去set一下ThreadLocal,代码很累赘,能不能更优雅地实现呢?
很容易想到,使用AOP或者手写InvocationHandler动态代理的方式来解决。难得有机会手写一次AOP,终于下决心认真去学习了一下AOP那些什么切面,连接点,增强各种乱七八糟的概念,好不容易学完了,发现其实我根本不太用的上。用不上AOP是因为下面的原因3导致代理的不是bean。(有空写写AOP那些概念,其实很简单就能讲清楚) -
使用动态代理,意味着代理的是mapper类,直接AOP代理mapper还是没办法拿到动态表名,怎么解决?
我想到了先用包装模式,把mapper和dynamicTableName包装起来,再进行AOP,这样包装类实例就得手动new没法用AOP了(也可能有别的办法让实例变成bean),只能自己搞InvocationHandler了。
image.png
image.png
-
因为业务上,dynamicTableName每次都是查表获得的,mapper+dynamicTableName的组合包装类实际上每次都new的话属实有点不优雅,应该搞个缓存。
这里就千万要注意我们的mapper方法调用是每个request多线程并发调用的,缓存必须使用ConcurrentHashMap。
image.png
神奇吧,一下子,把面试常用的包装模式,动态代理,并发集合,ThreadLocal都玩了一遍,所以还是很有必要记录一下的。
具体实现代码,我已经放在github上了。
使用@Scope("prototype") + AOP + @Cacheable实现
今天闲着没事,决定尝试完全使用spring提供的机制来实现。修改包括一下几个方面:
-
使用@Scope("prototype")+@Lookup注入替代手动new MapperWrapper()。
image.png -
使用@Cacheable+ConcurrentMapCacheManager替代手动使用ConcurrentHashMap。
image.png
image.png -
使用AOP的@Aspect+@Around("within(MapperWrapper)")注解自动设置threadlocal。
image.png