您好,欢迎来到三六零分类信息网!老站,搜索引擎当天收录,欢迎发信息
三六零分类信息网 > 平顶山分类信息网,免费分类信息发布

MyBatis插件的详细介绍

2025/4/17 4:25:25发布11次查看
plugins
摘一段来自mybatis官方文档的文字。
mybatis允许你在某一点拦截已映射语句执行的调用。默认情况下,mybatis允许使用插件来拦截方法调用
executor(update、query、flushstatements、commint、rollback、gettransaction、close、isclosed)
parameterhandler(getparameterobject、setparameters)
resultsethandler(handleresultsets、handleoutputparameters)
statementhandler(prepare、parameterize、batch、update、query)
这些类中方法的详情可以通过查看每个方法的签名来发现,而且它们的源代码存在于mybatis发行包中。你应该理解你所覆盖方法的行为,假设你所做的要比监视调用要多。如果你尝试修改或覆盖一个给定的方法,你可能会打破mybatis的核心。这是低层次的类和方法,要谨慎使用插件。
插件示例:打印每条sql语句及其执行时间
以下通过代码来演示一下如何使用mybatis的插件,要演示的场景是:打印每条真正执行的sql语句及其执行的时间。这是一个非常有用的需求,mybatis本身的日志可以记录sql,但是有以下几个问题:
mybatis日志打印出来的sql日志,参数都被占位符?替换,无法知道真正执行的sql语句中的参数是什么
mybatis日志打印出来的sql日志,有大量的换行符,通常一句sql语句要通过十几行显示,阅读体验非常差
无法记录sql执行时间,有sql执行时间就可以精准定位到执行时间比较慢的sql
写mybatis插件非常简单,只需要实现interceptor接口即可,我这里将我的interceptor命名为sqlcostinterceptor:
  1 /**  2  * sql执行时间记录拦截器    3  */  4 @intercepts({@signature(type = statementhandler.class, method = query, args = {statement.class, resulthandler.class}),  5     @signature(type = statementhandler.class, method = update, args = {statement.class}),  6     @signature(type = statementhandler.class, method = batch, args = { statement.class })})  7 public class sqlcostinterceptor implements interceptor {  8   9     @override 10     public object intercept(invocation invocation) throws throwable { 11         object target = invocation.gettarget(); 12          13         long starttime = system.currenttimemillis(); 14         statementhandler statementhandler = (statementhandler)target; 15         try { 16             return invocation.proceed(); 17         } finally { 18             long endtime = system.currenttimemillis(); 19             long sqlcost = endtime - starttime; 20              21             boundsql boundsql = statementhandler.getboundsql(); 22             string sql = boundsql.getsql(); 23             object parameterobject = boundsql.getparameterobject(); 24             list<parametermapping> parametermappinglist = boundsql.getparametermappings(); 25              26             // 格式化sql语句,去除换行符,替换参数 27             sql = formatsql(sql, parameterobject, parametermappinglist); 28              29             system.out.println(sql:[ + sql + ]执行耗时[ + sqlcost + ms]); 30         } 31     } 32  33     @override 34     public object plugin(object target) { 35         return plugin.wrap(target, this); 36     } 37  38     @override 39     public void setproperties(properties properties) { 40          41     } 42      43     @suppresswarnings(unchecked) 44     private string formatsql(string sql, object parameterobject, list<parametermapping> parametermappinglist) { 45         // 输入sql字符串空判断 46         if (sql == null || sql.length() == 0) { 47             return ; 48         } 49          50         // 美化sql 51         sql = beautifysql(sql); 52          53         // 不传参数的场景,直接把sql美化一下返回出去 54         if (parameterobject == null || parametermappinglist == null || parametermappinglist.size() == 0) { 55             return sql; 56         } 57          58         // 定义一个没有替换过占位符的sql,用于出异常时返回 59         string sqlwithoutreplaceplaceholder = sql; 60          61         try { 62             if (parametermappinglist != null) { 63                 class<?> parameterobjectclass = parameterobject.getclass(); 64  65                 // 如果参数是strictmap且value类型为collection,获取key=list的属性,这里主要是为了处理<foreach>循环时传入list这种参数的占位符替换 66                 // 例如select * from xxx where id in <foreach collection="list">...</foreach> 67                 if (isstrictmap(parameterobjectclass)) { 68                     strictmap strictmap = (strictmap)parameterobject; 69                      70                     if (islist(strictmap.get(list).getclass())) { 71                         sql = handlelistparameter(sql, strictmap.get(list)); 72                     } 73                 } else if (ismap(parameterobjectclass)) { 74                     // 如果参数是map则直接强转,通过map.get(key)方法获取真正的属性值 75                     // 这里主要是为了处理<insert>、<delete>、<update>、<select>时传入parametertype为map的场景 76                     map<?, ?> parammap = (map<?, ?>) parameterobject; 77                     sql = handlemapparameter(sql, parammap, parametermappinglist); 78                 } else { 79                     // 通用场景,比如传的是一个自定义的对象或者八种基本数据类型之一或者string 80                     sql = handlecommonparameter(sql, parametermappinglist, parameterobjectclass, parameterobject); 81                 } 82             } 83         } catch (exception e) { 84             // 占位符替换过程中出现异常,则返回没有替换过占位符但是格式美化过的sql,这样至少保证sql语句比boundsql中的sql更好看 85             return sqlwithoutreplaceplaceholder; 86         } 87          88         return sql; 89     } 90      91     /** 92      * 美化sql 93      */ 94     private string beautifysql(string sql) { 95         sql = sql.replace(\n, ).replace(\t, ).replace(  ,  ).replace(( , ().replace( ), )).replace( ,, ,); 96          97         return sql; 98     } 99     100     /**101      * 处理参数为list的场景102      */103     private string handlelistparameter(string sql, collection<?> col) {104         if (col != null && col.size() != 0) {105             for (object obj : col) {106                 string value = null;107                 class<?> objclass = obj.getclass();108                 109                 // 只处理基本数据类型、基本数据类型的包装类、string这三种110                 // 如果是复合类型也是可以的,不过复杂点且这种场景较少,写代码的时候要判断一下要拿到的是复合类型中的哪个属性111                 if (isprimitiveorprimitivewrapper(objclass)) {112                     value = obj.tostring();113                 } else if (objclass.isassignablefrom(string.class)) {114                     value = \ + obj.tostring() + \;  115                 }116                 117                 sql = sql.replacefirst(\\?, value);118             }119         }120         121         return sql;122     }123     124     /**125      * 处理参数为map的场景126      */127     private string handlemapparameter(string sql, map<?, ?> parammap, list<parametermapping> parametermappinglist) {128         for (parametermapping parametermapping : parametermappinglist) {129             object propertyname = parametermapping.getproperty();130             object propertyvalue = parammap.get(propertyname);131             if (propertyvalue != null) {132                 if (propertyvalue.getclass().isassignablefrom(string.class)) {133                     propertyvalue = \ + propertyvalue + \;134                 }135 136                 sql = sql.replacefirst(\\?, propertyvalue.tostring());137             }138         }139         140         return sql;141     }142     143     /**144      * 处理通用的场景145      */146     private string handlecommonparameter(string sql, list<parametermapping> parametermappinglist, class<?> parameterobjectclass,  147             object parameterobject) throws exception {148         for (parametermapping parametermapping : parametermappinglist) {149             string propertyvalue = null;150             // 基本数据类型或者基本数据类型的包装类,直接tostring即可获取其真正的参数值,其余直接取paramtermapping中的property属性即可151             if (isprimitiveorprimitivewrapper(parameterobjectclass)) {152                 propertyvalue = parameterobject.tostring();153             } else {154                 string propertyname = parametermapping.getproperty();155                 156                 field field = parameterobjectclass.getdeclaredfield(propertyname);157                 // 要获取field中的属性值,这里必须将私有属性的accessible设置为true158                 field.setaccessible(true);159                 propertyvalue = string.valueof(field.get(parameterobject));160                 if (parametermapping.getjavatype().isassignablefrom(string.class)) {161                     propertyvalue = \ + propertyvalue + \;162                 }163             }164 165             sql = sql.replacefirst(\\?, propertyvalue);166         }167         168         return sql;169     }170     171     /**172      * 是否基本数据类型或者基本数据类型的包装类173      */174     private boolean isprimitiveorprimitivewrapper(class<?> parameterobjectclass) {175         return parameterobjectclass.isprimitive() ||  176                 (parameterobjectclass.isassignablefrom(byte.class) || parameterobjectclass.isassignablefrom(short.class) ||177                         parameterobjectclass.isassignablefrom(integer.class) || parameterobjectclass.isassignablefrom(long.class) ||178                         parameterobjectclass.isassignablefrom(double.class) || parameterobjectclass.isassignablefrom(float.class) ||179                         parameterobjectclass.isassignablefrom(character.class) || parameterobjectclass.isassignablefrom(boolean.class));180     }181     182     /**183      * 是否defaultsqlsession的内部类strictmap184      */185     private boolean isstrictmap(class<?> parameterobjectclass) {186         return parameterobjectclass.isassignablefrom(strictmap.class);187     }188     189     /**190      * 是否list的实现类191      */192     private boolean islist(class<?> clazz) {193         class<?>[] interfaceclasses = clazz.getinterfaces();194         for (class<?> interfaceclass : interfaceclasses) {195             if (interfaceclass.isassignablefrom(list.class)) {196                 return true;197             }198         }199         200         return false;201     }202     203     /**204      * 是否map的实现类205      */206     private boolean ismap(class<?> parameterobjectclass) {207         class<?>[] interfaceclasses = parameterobjectclass.getinterfaces();208         for (class<?> interfaceclass : interfaceclasses) {209             if (interfaceclass.isassignablefrom(map.class)) {210                 return true;211             }212         }213         214         return false;215     }216     217 }
分析一下这段代码(这个是改良过的版本,主要是增加了对select * from xxx where id in <foreach collection="list">...</foreach>这种写法占位符替换为真正参数的支持)。
首先是注解@intercepts与@signature,这两个注解是必须的,因为plugin的wrap方法会取这两个注解里面参数。@intercepts中可以定义多个@signature,一个@signature表示符合如下条件的方法才会被拦截:
接口必须是type定义的类型
方法名必须和method一致
方法形参的class类型必须和args定义class类型顺序一致
接着的一个问题是:有四个接口可以拦截,为什么使用statementhandler去拦截?根据名字来看parameterhandler和resultsethandler,前者处理参数,后者处理结果是不可能使用的,剩下的就是executor和statementhandler了。拦截statementhandler的原因是而不是用executor的原因是:
executor的update与query方法可能用到mybatis的一二级缓存从而导致统计的并不是真正的sql执行时间
statementhandler的update与query方法无论如何都会统计到preparedstatement的execute方法执行时间,尽管也有一定误差(误差主要来自会将处理结果的时间也算上),但是相差不大
接着讲一下setproperties方法,可以将一些配置属性配置在<plugin></plugin>的子标签<property />中,所有的配置属性会在形参properties中,setproperties方法可以拿到配置的属性进行需要的处理。
接着讲一下plugin方法,这里是为目标接口生成代理,不需要也没必要自己去写生成代理的方法,mybatis的plugin类已经为我们提供了wrap方法(当然如果自己有自己的逻辑也可以在plugin.wrap方法前后加入,但是最终一定要使用plugin.wrap方法生成代理),看一下该方法的实现:
 1 public static object wrap(object target, interceptor interceptor) { 2     map signaturemap = getsignaturemap(interceptor); 3     class<?> type = target.getclass(); 4     class<?>[] interfaces = getallinterfaces(type, signaturemap); 5     if (interfaces.length > 0) { 6       return proxy.newproxyinstance( 7           type.getclassloader(), 8           interfaces, 9           new plugin(target, interceptor, signaturemap));10     }11     return target;12 }
因为这里的target一定是一个接口,因此可以放心使用jdk本身提供的proxy类,这里相当于就是如果该接口满足方法签名那么就为之生成一个代理。
最后就是intercept方法了,这里就是拦截器的核心代码了,方法的逻辑我就不解释了,可以自己看一下,唯一要注意的一点就是无论如何最终一定要返回invocation.proceed(),保证拦截器的层层调用。
xml文件配置即效果演示
写完了插件,只需要在config.xml文件中进行一次配置即可,非常简单:
 1 <plugins> 2     <plugin interceptor="org.xrq.mybatis.plugin.sqlcostinterceptor" /> 3 </plugins>
这里每个<plugin>子标签代表一个插件,interceptor表示拦截器的完整路径,每个人的不同。
有了类和这段配置,就可以使用sqlcostinterceptor了,sqlcostinterceptor是通用的,但是每个人的crud是不同的,我打印一下我这里crud执行的结果:
sql:[insert into mail(id, create_time, modify_time, web_id, mail, use_for) values(null, now(), now(), 1, 123@sina.com, 个人使用);]执行耗时[1ms] sql:[insert into mail(id, create_time, modify_time, web_id, mail, use_for) values(null, now(), now(), 2, 123@qq.com, 企业使用);]执行耗时[1ms] sql:[insert into mail(id, create_time, modify_time, web_id, mail, use_for) values(null, now(), now(), 3, 123@sohu.com, 注册账号使用);]执行耗时[0ms]

看到打印了完整的sql语句以及sql语句执行时间。
不过要说明一点,这个插件只是一个简单的demo,我并没有完整测试过,应该是无法覆盖所有场景的,所以如果想用这段代码片段打印真正的sql及其执行时间的朋友,还需要在这个基础上做修改,不过即使不改代码,这个插件起到美化sql的作用,去除一些换行符还是没问题的。
至于mybatis插件的实现原理,会在我【mybatis源码分析】系列文章中详细解读,文章地址为【mybatis源码分析】插件实现原理。
后记
mybatis插件机制非常有用,用得好可以解决很多问题,不只是这里的打印sql语句以及记录sql语句执行时间,分页、分表都可以通过插件来实现。用好插件的关键是我开头就列举的,这里再列一次:
executor(update、query、flushstatements、commint、rollback、gettransaction、close、isclosed)
parameterhandler(getparameterobject、setparameters)
resultsethandler(handleresultsets、handleoutputparameters)
statementhandler(prepare、parameterize、batch、update、query)
只有理解这四个接口及相关方法是干什么的,才能写出好的拦截器,开发出符合预期的功能。
以上就是mybatis插件的详细介绍的详细内容。
平顶山分类信息网,免费分类信息发布

VIP推荐

免费发布信息,免费发布B2B信息网站平台 - 三六零分类信息网 沪ICP备09012988号-2
企业名录