MyBatis框架原理4:插件

插件的定义和作用

首先引用MyBatis文档对插件(plugins)的定义:

MyBatis 允许你在已映射语句执行过程中的某一点进行拦截调用。默认情况下,MyBatis 允许使用插件来拦截的方法调用包括:

Executor (update, query, flushStatements, commit, rollback, getTransaction, close, isClosed)

ParameterHandler (getParameterObject, setParameters)

ResultSetHandler (handleResultSets, handleOutputParameters)

StatementHandler (prepare, parameterize, batch, update, query)

这些类中方法的细节可以通过查看每个方法的签名来发现,或者直接查看 MyBatis 发行包中的源代码。 如果你想做的不仅仅是监控方法的调用,那么你最好相当了解要重写的方法的行为。 因为如果在试图修改或重写已有方法的行为的时候,你很可能在破坏 MyBatis 的核心模块。 这些都是更低层的类和方法,所以使用插件的时候要特别当心。

Mybatis插件所拦截的4个对象正是在之前的MyBatis框架原理2:SqlSession运行过程文章中介绍的4个实现核心功能的接口。那么插件拦截这4个接口能做什么呢?根据之前文章对4个接口的介绍,可以猜测到:

Executor是SqlSession整个执行过程的总指挥,同时还对缓存进行操作,通过插件可以使用自定义的缓存,比如mybatis-enhanced-cache插件。

StatementHandler负责SQL的编译和执行,通过插件可以改写SQL语句。

ParameterHandler负责SQL的参数设置,通过插件可以改变参数设置。

ResultSetHandler负责结果集映射和存储过程输出参数的组装,通过插件可以对结果集映射规则改写。

插件的原理

在理解插件原理之前,得先搞清楚以下三个概念:

动态代理

代理模式是一种给真实对象提供一个代理对象,并由代理对象控制对真实对象的引用的一种设计模式,动态代理是在程序运行时动态生成代理类的模式,JDK动态代理对象是由java提供的一个Proxy类和InvocationHandler接口以及一个真实对象的接口生成的。通常InvocationHandler的实现类持有一个真实对象字段和定义一个invoke方法,通过Proxy类的newProxyInstance方法就可以生成这个真实对象的代理对象,通过代理对象调度方法实际就是调用InvocationHandler实现类的invoke方法,在invoke方法中可以通过反射实现调用真实对象的方法。

拦截器(Interceptor)

动态代理对象可以对真实对象方法引用,是因为InvocationHandler实现类持有了一个真实对象的字段,通过反射就可以实现这个功能。如果InvocationHandler实现类再持有一个Interceptor接口的实现类,Interceptor接口定义了一个入参为真实对象的intercept方法,Interceptor接口的实现类通过重写intercept方法可以对真实对象的方法引用或者实现增强功能等等,也就是当我们再次使用这个动态代理对象调度方法时,可以根据需求对真实对象的方法做出改变。

从这个Interceptor接口实现类的功能上来看,可以叫做真实对象方法的拦截器。于是我们再想一下,如果前面讲到MyBatis的4个核心功能接口的实现类(比如PreparedStatementHandler)是一个真实对象,我们通过JDK动态代理技术生成一个代理对象,并且生成代理类所需的InvocationHandler实现类同时还持有了一个Interceptor接口实现类,通过使用代理对象调度方法,我们就可以根据需求对PreparedStatementHandler的功能进行增强。

实际上MyBatis确实提供了这样一个Interceptor接口和intercept方法,也提供了这样的一个InvocationHandler接口的实现类,类名叫Plugin,它们都位于MyBatis的org.apache.ibatis.plugin包下。等等,那么MyBatis的插件不就是拦截器吗?拦截器的原理都讲完了,等下还怎么讲什么插件原理?


责任链模式

我们通过JDK动态代理技术生成一个代理对象,代理的真实对象是个StatementHandler,并且持有StatementHandler的拦截器(插件)。如果我们把这个代理对象视为一个target对象,再利用动态代理生成一个代理类,并且持有对这个target对象的拦截器(插件),如果再把新生成的代理视为一个新的target类,同样持有对新target类的拦截器(插件),那么我们就得到了一个像是被包裹了三层拦截器(插件)的StatementHandler的代理对象:

当MyBatis每一次SqlSession会话需要引用到StatementHandler的方法时,如过符合上图中拦截器3的拦截逻辑,则按拦截器3的定义的方法执行;如果不符合拦截逻辑,则将执行责任交给拦截器2处理,以此类推,这样的模式叫做责任链模式。MyBatis全局配置文件里可以配置多个插件,多个插件的运行就是按照这样的责任链模式执行的。

通过对以上三点的理解,我们已经对MyBatis插件原理已经有了初步认识,下面就通过源码看看MyBatis插件是如何运行起来的。

插件的运行过程

插件的接口

MyBatis提供了一个Interceptor接口,插件必须实现这个接口,接口定义了3个方法如下:

publicinterfaceInterceptor{// 执行插件实现的方法,Invocation对象持有真实对象,可通过反射调用真实对象的方法Objectintercept(Invocation invocation)throwsThrowable;// 设置插件拦截的对象target,通常调用Pulgin类的wrap方法生成一个代理类Objectplugin(Object target);// 根据配置文件初始化插件voidsetProperties(Properties properties);}

插件的初始化

在MyBatis初始化时XMLConfigBuilderder的pluginElement方法对插件配置文件解析:

privatevoidpluginElement(XNode parent)throwsException{if(parent !=null) {for(XNode child : parent.getChildren()) {      String interceptor = child.getStringAttribute("interceptor");      Properties properties = child.getChildrenAsProperties();// 通过反射生成插件的实例Interceptor interceptorInstance = (Interceptor) resolveClass(interceptor).newInstance();// 调用插件配置参数interceptorInstance.setProperties(properties);// 将插件实例保存到Configuration对象中configuration.addInterceptor(interceptorInstance);    }  }}

Configuration对象最终将解析出的插件配置保存在持有InterceptorChain对象中,InterceptorChain对象又是通过一个ArrayList来保存所有插件,可见在MyBatis初始化的时候插件配置就已经加载好了,运行时就会根据插件编写的规则执行拦截逻辑。

publicclassInterceptorChain{// 通过集合来保存插件privatefinalList interceptors =newArrayList();// 通过责任链模式调用插件plugin方法生成代理对象publicObjectpluginAll(Object target){for(Interceptor interceptor : interceptors) {            target = interceptor.plugin(target);          }returntarget;  }//  Configuration对象调用的添加插件的方法publicvoidaddInterceptor(Interceptor interceptor){      interceptors.add(interceptor);  }publicListgetInterceptors(){returnCollections.unmodifiableList(interceptors);  }}

插件的运行

如果我们需要拦截MyBatis的Executor接口,Configuration在初始化Executor时就会通过责任链模式将初始化的Executor作为真实对象,调用InterceptorChain的pluginAll放法生成代理对象:

publicExecutornewExecutor(Transaction transaction, ExecutorType executorType){executorType = executorType ==null? defaultExecutorType : executorType;executorType = executorType ==null? ExecutorType.SIMPLE : executorType;Executor executor;// 根据配置文件生成相应Executorif(ExecutorType.BATCH == executorType) {  executor =newBatchExecutor(this, transaction);}elseif(ExecutorType.REUSE == executorType) {  executor =newReuseExecutor(this, transaction);}else{  executor =newSimpleExecutor(this, transaction);}if(cacheEnabled) {  executor =newCachingExecutor(executor);}// 调用InterceptorChain的pluginAll放法生成代理对象executor = (Executor) interceptorChain.pluginAll(executor);returnexecutor;}

InterceptorChain的 pluginAll方法调用插件的plugin方法,plugin方法可以调用MyBatis提供的工具类Plugin类来生成代理对象,Plugin类实现了InvocationHandler,在Plugin类中定义invoke方法来实现拦截逻辑和执行插件方法:

publicclassPluginimplementsInvocationHandler{// target为需要拦截的真实对象 privatefinalObject target;// interceptor为插件privatefinalInterceptor interceptor;privatefinalMap, Set> signatureMap;privatePlugin(Object target, Interceptor interceptor, Map<Class<?>, Set<Method>> signatureMap){this.target = target;this.interceptor = interceptor;this.signatureMap = signatureMap;}publicstaticObjectwrap(Object target, Interceptor interceptor){  Map, Set> signatureMap = getSignatureMap(interceptor);  Class type = target.getClass();  Class[] interfaces = getAllInterfaces(type, signatureMap);// 动态代理生成代理对象并返回if(interfaces.length >0) {returnProxy.newProxyInstance(        type.getClassLoader(),        interfaces,newPlugin(target, interceptor, signatureMap));  }returntarget;}@OverridepublicObjectinvoke(Object proxy, Method method, Object[] args)throwsThrowable{try{    Set methods = signatureMap.get(method.getDeclaringClass());// 根据拦截逻辑执行插件的方法if(methods !=null&& methods.contains(method)) {returninterceptor.intercept(newInvocation(target, method, args));    }// 直接调用真是对象的方法returnmethod.invoke(target, args);  }catch(Exception e) {throwExceptionUtil.unwrapThrowable(e);  }}...

插件的intercept方法参数为Invocation对象,Invocation对象持有真实对象和一个proceed方法,proceed方法通过反射调用真实对象的方法。于是多个插件生成的责任链模式的代理对象,就可以通过一层一层执行proceed方法来调用真实对象的方法。

插件的开发

自己编写插件必须继承MyBatis的Interceptor接口

publicinterfaceInterceptor{// 执行插件实现的方法,Invocation对象持有真实对象,可通过反射调用真实对象的方法Objectintercept(Invocation invocation)throwsThrowable;// 设置插件拦截的对象target,通常调用Pulgin类的wrap方法生成一个代理类Objectplugin(Object target);// 根据配置文件初始化插件voidsetProperties(Properties properties);  }

使用@Intercepts和@Signature注解

@Intercepts({@Signature(type = StatementHandler.class, method ="prepare", args = {Connection.class ,Integet.class})})publicclassMyPluginimplementsInterceptor{...}

用@Intercepts注解申明是一个插件,@Signature注解申明拦截的对象,方法和参数。上面的写法表明了拦截了StatementHandler对象的prepare方法,参数是一个Connection对象和一个Integet。

编写拦截方法

MyBatis提供了一个Invocation工具类,通常我们将需要拦截的真实对象,方法及参数封装在里面作为一个参数传给插件的intercept方法,在插件intercept方法里可以编写拦截逻辑和执行拦截方法,方法参数invocation可以通过反射调用被代理对象的方法:

@Intercepts({@Signature(type = StatementHandler.class, method ="prepare", args = {Connection.class ,Integet.class})})publicclassMyPluginimplementsInterceptor{@overridepublicObjectintercept(Invocation invocation)throwsThrowable{// do something ...// 调用被代理对象的方法invocation.proceed();// do something ...@override调用Plugin工具类生成代理对象publicObjectplugin(Object target){returnPlugin.wrap(target,this);      }      ...  }

生成代理对象

MyBatis还提供了一个Plugin工具类,其中wrap方法用于生成代理类,invoke方法验证拦截类型和方法,并选择是否按拦截器的方法,代码如下:

publicclassPluginimplementsInvocationHandler{privatefinalObject target;// 真实对象privatefinalInterceptor interceptor;// 拦截器(插件)privatefinalMap, Set> signatureMap;// Map保存签名的类型,方法和参数信息privatePlugin(Object target, Interceptor interceptor, Map<Class<?>, Set<Method>> signatureMap){this.target = target;this.interceptor = interceptor;this.signatureMap = signatureMap;}publicstaticObjectwrap(Object target, Interceptor interceptor){// getSignatureMap方法通过反射获取插件里@Intercepts和@Signature注解声明的拦截类型,方法和参数信息Map, Set> signatureMap = getSignatureMap(interceptor);  Class type = target.getClass();  从signatureMap中获取拦截对象的类型  Class[] interfaces = getAllInterfaces(type, signatureMap);// 生成代理对象,如果target的类型不是插件里注解声明的类型则直接返回target不作拦截。if(interfaces.length >0) {returnProxy.newProxyInstance(        type.getClassLoader(),        interfaces,newPlugin(target, interceptor, signatureMap));  }returntarget;}@OverridepublicObjectinvoke(Object proxy, Method method, Object[] args)throwsThrowable{try{// 验证代理对象调用的方法是否为插件里申明拦截的方法Set methods = signatureMap.get(method.getDeclaringClass());if(methods !=null&& methods.contains(method)) {// 如果是声明拦截的方法,则调用插件的intercept方法执行拦截处理returninterceptor.intercept(newInvocation(target, method, args));    }// 如果不是声明拦截的方法,则直接调用真实对象的方法returnmethod.invoke(target, args);  }catch(Exception e) {throwExceptionUtil.unwrapThrowable(e);  }}...

总结

MyBatis插件运行依靠Java动态代理技术实现,虽然原理很简单,但是编写插件涉及到修改MyBatis框架底层的接口,需要十分谨慎,做为初学者,最好使用现成的插件。

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

推荐阅读更多精彩内容