Mybatis自定义插件实战以及与Spring整合原理

文章目录

• 前言
• 自定义插件实战
• • 需求
  • 实现过程
  • 测试
  • 总结
• Spring与Mybatis整合
• • 整合过程
  • 实现原理
  • • @MapperScan
    • MapperScannerConfigurer
    • MapperFactoryBean
    • SqlSessionFactoryBean
  • 总结

---

前言

学习一个框架除了搞清楚核心原理外，还有很关键的需要知道如何去扩展它，而对Mybatis来说，它提供了插件的方式帮助我们进行功能扩展，所以学习如何自定义一个插件也是非常有必要的，本文介绍一个基于插件如何实现一个分表的操作，相信从这个示例的学习，读者会有些收获的。本文还介绍了Spring如何与Mybatis整合的原理，帮助读者更多的了解Mybatis。

自定义插件实战

需求

插件能够改变或者扩展Mybatis的原有的功能，像Mybatis涉及到分页操作，通常就会去引用分页插件来实现分页功能，所以搞懂插件还是非常有必要的。下面通过一个简单的自定义插件来实战一个分片功能，来让读者对插件有更深的了解。

实现过程

首先，自定义一个分片路由插件，通过@Intercepts定义插件的拦截的目标类型，当前插件主要拦截StatementHandler类型，对query、update、prepare方法进行拦截

@Intercepts(
   {
	   @Signature(type=StatementHandler.class,method="prepare",args={Connection.class,Integer.class}),
	   @Signature(type=StatementHandler.class,method="query",args={Statement.class,ResultHandler.class}),
	   @Signature(type=StatementHandler.class,method="update",args={Statement.class})
   }		
)
public class MyRoutePlugin implements Interceptor{
    @Autowired 
	private RouteConfig routeConfig;

	@Autowired
	private DataSource datasouce;
}

同时定义一个分片规则配置类，其中routeKey为分片字段，决定当前执行的SQL语句是否需要分片

@ConfigurationProperties(prefix="route")
@Component
@Data
public class RouteConfg {

	private boolean enabled=true;
	//分片字段
	private String routeKey;
	//分片表的数量
	private int tableCount;
}

接下来，重写intercept方法，定义路由插件的实现逻辑，核心逻辑就是根据StatementHandler一步步通过反射拿到当前执行的SQL语句，然后确认是否开启路由分片，开启之后就会进行后续分片处理

public Object intercept(Invocation invocation) throws Throwable {
		//先判断拦截的目标类型是否为StatementHandler
		if(invocation.getTarget() instanceof StatementHandler){
			//获得具体statementHandler实现，这里拿到的是RoutingStatementHandler
			StatementHandler statementHandler=(StatementHandler) invoca