mybatis的插件机制

2019-11-03 来源: 不懂是非 发布在  https://www.cnblogs.com/qm-article/p/11785350.html

一、mybatis的插件介绍

关于mybatis的插件,我想大家也都用过,就比如最常用的逆向工程,根据表结构生成model,dao,xml文件,还有分页插件,那这些插件的工作原理是怎么样的呢,就比如分页插件,它为什么能改变我们在xml文件中写的sql语句,本文将带大家一起来了解mybatis的插件机制。(由于本人也是在不断学习,文中难免有错误或不足之处,还望指正,本文基于mybatis3.3.0版本),下面将围绕这几个方面

1、插件入口,即怎么把插件注入到mybatis代码里面

2、插件能拦截哪些类或哪些方法

3、举例简易分表插件

二、插件入口

在先了解前,我们来一段自定义mybatis插件的代码

 import org.apache.ibatis.executor.statement.StatementHandler;
 import org.apache.ibatis.plugin.*;

 import java.sql.Connection;
 import java.util.Properties;

 @Intercepts(
         value = {
                 @Signature(
                         type = StatementHandler.class,
                         method = "prepare",
                         args = {Connection.class} // 不同版本的prepare方法参数不一样,高版本的还有一个Integer参数
                 )
         }
 )
 public class PluginDemo implements Interceptor {
     @Override
     public Object intercept(Invocation invocation) throws Throwable {
         return invocation.proceed();
     }

     @Override
     public Object plugin(Object target) {
         return Plugin.wrap(target,this);
     }

     @Override
     public void setProperties(Properties properties) {

     }

要自定义mybatis插件,必须得实现Interceptor接口,这个接口有三个抽象方法

1、intercept,这个方法是mybatis的核心方法,要实现自定义逻辑,基本都是改造这个方法,其中invocation参数可以通过反射要获取原始方法和对应参数信息

2、plugin,它的作用是用来生成一个拦截对方,也就是代理对象,使得被代理的对象一定会经过intercept方法,通常都会使用mybatis提供的工具类Plugin来获取代理对象,如果有自己独特需求,可以自定义

3、setProperties,这个方法就是用来设置插件的一些属性

其中@intercepts注解就是用来标明拦截哪些类,哪些方法。

当我们脱离spring容器来使用mybatis的时候,我们通常是这样写代码的

而在mybatisConfig.xml文件中,我们有配置数据库连接信息,插件,别名,mapper文件映射地址等信息,(温馨提示,这些配置有一定顺序,如果不按照顺序配置,则mybatis解析时会抛出异常,详细配置信息可以参考mbatis的dtd文件配置)既然在配置文件中配置了,那肯定也会被解析掉,如下图,在XmlConfigBuild中进行解析

解析后,插件信息会被存储到configuation中InterceptorChain集合里面,这个configuation在mybatis运行周期是一个单例的,它负责存储所有的配置信息

最终,mybatis的插件信息完整的注入到了configuation里面。

二、mybaits插件能拦截哪些类或哪些方法

在正常开发流程中,我们基本都是通过先获取sqlSession,如果不采用自定义配置,在默认的sqlsession实现就是defaultSqlSession,在该类的一个方法里就隐含了融合插件,

其中会获取到executor,这个类是通过configuation获取的,

最终融合插件的方法就是图中红框的代码。它会为原始的executor类生成代理类。从而你在执行executor类的一些方法时,比如query,update方法,会先生成对应的代理对象,myabtis采用的是jdk的动态代理,代理后,你执行executor类的query,update方法时会自动转接到你自定义的插件intercept方法里面,也可以理解为覆盖原来的方法。

通过这,我们大概也可以猜测的出,mybatis插件要拦截的类,很大原因在configuation类中有生成,也不带大家绕弯子了,来看截图

拦截statementHandler接口的实现类,它通过routingStatementHandler来实现路由,最终定向到prepareStatementHandler类

拦截parameterHandler。这个接口是用来封装参数用的,也就是最终将参数设置到sql里面

拦截ResultSetHandler接口,它主要用于处理sql运行返回的结果

总结,mybatis可以拦截的类分四大类

一、executor,executor类可以说是执行sql的全过程,如组装参数,sql改造,结果处理,比较广泛,但实际用的不多

二、StatementHandler,这个是执行sql的过程,可以获取到待执行的sql,可用来改造sql,如分页,分表,最常拦截的类

三、paremeterHandler,这个用来拦截sql的参数,可以自定义参数组装规则

四、resultHandler,这个用来处理结果

三、简易版分表插件

 import org.apache.ibatis.executor.statement.StatementHandler;
 import org.apache.ibatis.mapping.BoundSql;
 import org.apache.ibatis.plugin.*;
 import org.apache.ibatis.reflection.MetaObject;
 import org.apache.ibatis.reflection.SystemMetaObject;

 import java.sql.Connection;
 import java.util.Properties;

 @Intercepts(
         value = {
                 @Signature(
                         type = StatementHandler.class,
                         method = "prepare",
                         args = {Connection.class}
                 )
         }
 )
 public class PluginDemo implements Interceptor {
     @Override
     public Object intercept(Invocation invocation) throws Throwable {

         /**
          * 获取被拦截的目前类,在这里是拦截了statementHandler,所有目前类也就是它
          * 通过这个类我们可以拿到待执行的sql语句,通常使用mataObject工具类来获取
          * 关于这个工具类,大家可自行了解,个人认为这个工具类很强大
           */
         StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
         MetaObject metaObject = SystemMetaObject.forObject(statementHandler);
         /**
          * 先解释下为什么写成delegate.boundSql就可以拿到boundSql类
          * 从前面也可以得知,statementHandler的默认实现是routingStatementHandler。
          * 这个类有一个属性statementHandler,属性名就叫delegate,而这个属性的默认实现又是preparedStatementHandler
          * 后面这个类又有属性boundSql,所以,最终形成的写法就是delegate.boundSql。
          * 所以这也体现了MetaObject工具类的强大,可以通过实例传参,就可以根据属性名获取对应属性值
          */
         BoundSql boundSql = (BoundSql) metaObject.getValue("delegate.boundSql");

         // 待执行的sql,在这里也就是预编译后的sql,即参数位都是?号
         String sql = boundSql.getSql();
         /**
          * 既然拿到了预编译后的sql,那就可以按照你自己的想法为所欲为,如分页,按年分表等等
          * 分表的话,个人推荐druid的sql解析器,我认为还是不错的,大家可以自行了解
          * 最后改造完sql,别忘了把它设置回去
          * metaObject.setValue("delegate.boundSql.sql",sql);
          *  invocation.proceed,即原始方法的执行
          *  注意点就是,因为mybatis插件采用的是代理模式,所以如果存在多个插件,会形成多个代理
          *  你如果要拿到最原始的对象,还得进一步进行分解
          *  如:while(metaObject.getValue(""h) != null){
          *      Object obj = metaObject.getValue("h");
          *       ....
          *  }
          */
         return invocation.proceed();
     }

     @Override
     public Object plugin(Object target) {
         return Plugin.wrap(target,this);
     }

     @Override
     public void setProperties(Properties properties) {

     }
 }

-----------------------------------------------------------------------------------------------------------------------------分界线-------------------------------------------------------------------------------------

以上就是全部内容

相关文章