Heim >Java >javaLernprogramm >Detaillierte Einführung des MyBatis-Plug-Ins
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<Collection<?>> strictMap = (StrictMap<Collection<?>>)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
首先是注解@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方法,这里是为目标接口生成代理,不需要也没必要自己去写生成代理的方法,MyBatis的Plugin类已经为我们提供了wrap方法(当然如果自己有自己的逻辑也可以在Plugin.wrap方法前后加入,但是最终一定要使用Plugin.wrap方法生成代理),看一下该方法的实现:
1 public static Object wrap(Object target, Interceptor interceptor) { 2 Map<Class<?>, Set<Method>> 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>
这里每个
有了类和这段配置,就可以使用SqlCostInterceptor了,SqlCostInterceptor是通用的,但是每个人的CRUD是不同的,我打印一下我这里CRUD执行的结果:
<span style="color: #000000">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]<br></span>
Sehen Sie sich die vollständige gedruckte SQL-Anweisung und die Ausführungszeit der SQL-Anweisung an.
Es ist jedoch zu beachten, dass es sich bei diesem Plug-in nur um eine einfache Demo handelt. Es sollte nicht alle Szenarien abdecken Code-Snippet zum Drucken des echten Freundes, der die SQL und ihre Ausführungszeit kennt, muss auf dieser Basis noch Änderungen vornehmen. Selbst wenn der Code nicht geändert wird, spielt dieses Plug-In jedoch die Rolle, die SQL zu verschönern und einige Zeilenumbrüche zu entfernen ist immer noch kein Problem.
Das Implementierungsprinzip des MyBatis-Plug-Ins wird in meiner Artikelserie [MyBatis-Quellcode-Analyse] ausführlich erläutert. Die Artikeladresse lautet [MyBatis-Quellcode Analyse] Plug-in-Implementierungsprinzip.
Postscript
Der MyBatis-Plug-In-Mechanismus ist sehr nützlich und kann viele Probleme lösen Bei richtiger Verwendung kann es Probleme geben, nicht nur das Drucken von SQL-Anweisungen und das Aufzeichnen der Ausführungszeit von SQL-Anweisungen, sondern auch Paging und Tabellenaufteilung können durch Plug-Ins implementiert werden. Der Schlüssel zur guten Verwendung von Plug-Ins ist das, was ich am Anfang aufgelistet habe:
Executor (Update, Query, FlushStatements, Commit, Rollback, getTransaction, close, isClosed)
ParameterHandler(getParameterObject, setParameters)
ResultSetHandler(handleResultSets, handleOutputParameters)
StatementHandler( vorbereiten, parametrisieren, stapeln, aktualisieren, abfragen)
Nur wenn wir verstehen, was diese vier Schnittstellen und zugehörigen Methoden bewirken, können wir gute Interceptoren schreiben und Funktionen entwickeln, die den Erwartungen entsprechen.
Das obige ist der detaillierte Inhalt vonDetaillierte Einführung des MyBatis-Plug-Ins. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!