Spring+Mybatis学习之Plugin使用

这里的Plugin不是开发工具中集成的插件,而是Mybatis中额外扩展功能使用到的插件,这个类所在的包都在plugin包下。这个原理比较简单,总共有7个类在该包下,其中一个PluginException异常类,抛出相关插件异常的。两个注解InterceptsSignature,其余四个类实现相关功能。插件实现相关功能的主要是其中的接口Interceptor,实现了该接口,然后将其注入到容器中,就能修改每条sql执行时的动作,如果操作不好,容易出现意想不到的结果,所以操作需要慎重。

  • Interceptor实现该接口即可拦截所有的sql操作
  • Intercepts 注解,用于注解到插件类上,参数决定拦截哪些类哪些方法
  • Signature 注解,与Intercepts搭配使用
  • Plugin 辅助实现一些识别,判断及生成代理对象相关功能

注入容器

 public MapperAutoConfiguration(MybatisProperties properties,
                                    ObjectProvider<Interceptor[]> interceptorsProvider,
                                    ResourceLoader resourceLoader,
                                    ObjectProvider<DatabaseIdProvider> databaseIdProvider,
                                    ObjectProvider<List<ConfigurationCustomizer>> configurationCustomizersProvider) {
        this.properties = properties;
        this.interceptors = interceptorsProvider.getIfAvailable();
        this.resourceLoader = resourceLoader;
        this.databaseIdProvider = databaseIdProvider.getIfAvailable();
        this.configurationCustomizers = configurationCustomizersProvider.getIfAvailable();
    }

这里还是上篇介绍的MapperAutoConfiguration类,通过配置注解,该类随着spring boot启动而注入,而容器中实现了Interceptor的所有类都会作为一个列表注入到容器中。
MapperAutoConfiguration生成SqlSessionFactory的Bean时调用

 if (!ObjectUtils.isEmpty(this.interceptors)) {
            factory.setPlugins(this.interceptors);
 }

SqlSessionFactory在构建时调用了如下代码保存相关插件,其实最终是保存在了配置类Configuration

if (!isEmpty(this.plugins)) {
      for (Interceptor plugin : this.plugins) {
        configuration.addInterceptor(plugin);
        if (LOGGER.isDebugEnabled()) {
          LOGGER.debug("Registered plugin: '" + plugin + "'");
        }
      }

该类中保存了几乎绝大部分mybatis需要用到的东西,在spring容器中,Configuration是单例,相关插件会保存到如下属性中。

  protected final InterceptorChain interceptorChain = new InterceptorChain();

InterceptorChain内部很简单,一个List用来保存相关插件,而该类也在plugin包下,是刚才提到的7个类之一。

插件实现Interceptor

接下来看如果实现相关接口,制作一个插件。该接口也很简单,内部代码如下,没有过多说明。

public interface Interceptor {

  Object intercept(Invocation invocation) throws Throwable;

  Object plugin(Object target);

  void setProperties(Properties properties);

}

实现类

@Intercepts({@Signature(type= Executor.class,method = "update",args ={MappedStatement.class,Object.class})})
public class ExamplePlugin implements Interceptor {
  private Properties properties = new Properties();

  public Object intercept(Invocation invocation) throws Throwable {
    // implement pre processing if need
    Object returnObject = invocation.proceed();
    // implement post processing if need
    return returnObject;
  }
  
    @Override
    public Object plugin(Object target) {
        return Plugin.wrap(target, this);
    }

  public void setProperties(Properties properties) {
    this.properties = properties;
  }
}

将该类注入到容器中

<!-- mybatis-config.xml -->
<plugins>
  <plugin interceptor="org.mybatis.example.ExamplePlugin">
    <property name="someProperty" value="100"/>
  </plugin>
</plugins>

spring boot中可以通过添加@Component注解,则该插件会自动注入,但是不能使用@Configuration注入,否则不能正常使用。

插件拦截过程

上边的代码会在Executor类执行所有的update操作时都经过此插件。而可以拦截的type类型有四种:

  • Executor (update, query, flushStatements, commit, rollback, getTransaction, close, isClosed)
  • ParameterHandler (getParameterObject, setParameters)
  • ResultSetHandler (handleResultSets, handleOutputParameters)
  • StatementHandler (prepare, parameterize, batch, update, query)

具体调用过程可以查看Configuration在创建这些对象是如何执行。这四个类中都执行了如下代码
executor = (Executor) interceptorChain.pluginAll(executor);
具体的入参和返回值是这四个类中的之一。然后在InterceptorChain中的pluginAll方法实际是循环调用了所有插件的plugin方法。也就是说以上四个类的所有方法都会经过我们插件中的plugin方法。而Plugin.wrap(target, this)这个方法才是识别相关注解中的类和方法。该方法会判断当前target是否为注解需要拦截的对象类型,如果是,为其生成相关代理对象返回。
生成的代理对象调用的处理程序代码即为Plugin类中的invoke方法

public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    try {
      Set<Method> methods = signatureMap.get(method.getDeclaringClass());
      if (methods != null && methods.contains(method)) {
        return interceptor.intercept(new Invocation(target, method, args));
      }
      return method.invoke(target, args);
    } catch (Exception e) {
      throw ExceptionUtil.unwrapThrowable(e);
    }
  }

如果执行的方法是编写插件类注解中需要拦截的方法,则调用插件中的intercept方法,否则按原方法执行。
这里Interceptor接口中的两个方法intercept和plugin,plugin用于替换拦截到的类,而intercept方法则是替换原有执行方法。这里代理执行有点绕,别的还好理解。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,542评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,596评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,021评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,682评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,792评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,985评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,107评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,845评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,299评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,612评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,747评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,441评论 4 333
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,072评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,828评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,069评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,545评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,658评论 2 350

推荐阅读更多精彩内容