源码级别解读 mybatis 插件

简介:

MyBatis 是一款优秀的持久层框架,它支持定制化 SQL、存储过程以及高级映射。MyBatis 避免了几乎所有的 JDBC 代码和手动设置参数以及获取结果集。MyBatis 可以使用简单的 XML 或注解来配置和映射原生信息,将接口和 Java 的 POJOs(Plain Old Java Objects,普通的 Java对象)映射成数据库中的记录。

摘自官网。。

为什么要使用mybatis,相信看这篇文章的童鞋都有所了解,俺这里就多废话几句。在orm框架中,有轻量级的dbutils与mybatis,重量级的有hibernate。

为什么要选择mybatis呢?原因如下:

1.简单,这个无需我多废话。肯定的啊,使用过的童鞋都知道,在dao层定义一个接口,然后定义一个对应的xml(xml中namespace的值对应接口的全限定名就ok。因为mybatis是通过namespaceid去和接口类进行映射,然后使用动态代理创建接口类的实例方法,参见动态代理).

2.结果集映射,我认为这是mybatis做的最牛的一点,也是选择他最大的一个理由

3.完善的缓存机制(mybatis有1级缓存和2级缓存),1级缓存是基于会话的(Session),2级缓存是基于应用级别。这里不做过多阐述,有兴趣的可以参考文章 http://www.iteye.com/topic/1112327/

4.强大的插件机制。它可以在运行过程中动态的执行你自己的业务逻辑,比如防止sql注入,分页,sql日志打印,sql执行耗时等都可以在插件中做。真正实现了业务与功能分离。让你随行所欲的在飞dao层中增加任何你想做的事情。

好了。上面说了那么多好处,本文中会有点穿插,但重点是插件机制。废话不多说,我们开始缕一缕mybatis的流程

当然,最古老的方式就不去讲了(就是去调用mybatis select, selectOne这种方式),我们直接从接口的玩法中入手

这是mybatis的一张架构图,从中我们可以看得出。应用程序去调用对应的db操作时会经历 配置->打开会话->在会话中执行相应的操作,同时会话中包含了jdbc的事务(对,没错。就是jdbc的事务)。那么我们可以看得出来,会话是核心。而配置是关键。

我们可以打开配置类看看,中间有很多属性。我们绝大多数都可以忽略,但是

这个就是mybatis内部维护的拦截器链。我们在解析配置,初始化的时候会给xml中写入一个拦截器的类,他是从

Configuration.addInterceptor(Interceptor interceptor)

这个方法中写入。这一点对于本文来讲很重要,请大家牢记。

接下来说会话。有心急的小伙伴就会提问,插件和会话有关系吗?

必然的事情,要不然我也不会罗里吧嗦的说这么多关于会话的东西。我们可以看看sqlsession的结构

可以看得出来,他有2个实现类(用过spring的童鞋应该知道,sqlsessionTemplate是spring对sqlsession的简单封装,这里不再赘述), 那么就剩下了DefaultSqlSession和SqlSessionManager这两个类。

那么我们可以做一个大胆的假设,mybatis他默认打开的是DefaultSqlSession(其实在源码中是的)。我们要构建一个sqlsession的时候会使用 SqlSessionFactoryBuilder 这个类,通过传递一个抽象的Reader对象(是不是很眼熟?这玩意就是IO流里的字节流基类), 通过这个玩意传一个xml文件进来

然后去构建一个SqlSessionFactory对象。

可以看得出,它是new的defaultSqlSessionFactory这个类。我们跟进去看看

这里没什么稀奇的,无非就是把我们文件中的那些属性转移给这个Configuration类中,至于这个类里面具体有什么我就不带大家看了。我们继续看打开会话的流程

到这里他的会话构建就出来了,有心的童鞋这里应该能看到了吧,我上面说的jdbc事务和DefaultSqlSession

ok,我们打开这个类看看,里面全是一些模板方法。俺这里就不一一赘述

我们的重点是从

执行器(我暂且叫他执行器,也比较形象)开始

他这里就有好多个实现类了。老规矩。我就不带着大家找了,直接上图。默认的执行器类型是SimpleExecutor

找到这里,我们就会发现,这就是它的增删改查了。老规矩,不赘述

继续深入,接下来是插件部分的重点了:

在开始之前我们需要介绍一下,它的拦截器可以作用于

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

这些范围。

我在上面说过,interceptorChain 这个东东很重要,这就是我们的拦截器链。我们可以看到他会使用pluginAll(Object target)这个方法将所有拦截器加到拦截器链中。

而拦截器类是这样的。

如果我们需要实现一个自己的拦截器则需要实现这个接口中的方法。

其中

Object intercept(Invocation invocation) throws Throwable; // 这个方法是拦截器的业务方法
Object plugin(Object target); // 这个方法是对拦截器的包装, 如果不包装的话它是不会被加入到拦截器链中
void setProperties(Properties properties);// 这个方法是设置一些额外的属性

看明白这个之后我们自己手动去编写一个拦截器类

/**
 * @author Autorun
 * Created by Autorun on 2018/1/4.
 */@Intercepts({ @Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class}) })
public class LogInterceptor implements Interceptor {

    private final Logger logger = LoggerFactory.getLogger(this.getClass());    @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) {

    }
}

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

这一行是标注要拦截哪个类上的方法,参数等。然后使用动态代理的invoke(Object obj, Object... args)方法,执行我们的逻辑

如果有多个拦截器则按照顺序(因为他内部是使用的ArrayList容器,有序可重复,大家都懂得)

具体执行的话是使用Plugin类中的Object invoke(Object proxy, Method method, Object[] args) throws Throwable 方法来做的。可以看得出来。它里面调用了我们编写的 Object intercept(Invocation invocation) throws Throwable;

之后我们需要把我们写的拦截器类注入到拦截器链中。

在mybatis的配置文件中加入

<!-- mybatis-config.xml --><plugins>  <!-- 类的全限定名 -->
  <plugin interceptor="org.mybatis.example.ExamplePlugin"> 
  </plugin></plugins>

使我们的拦截器生效。运行程序就会发现我们的业务在对应的范围生效了。

文章编写的比较仓促,如果有问题可以留言给我,我会一一回复大家。谢谢

原文发布于微信公众号 - 芋道源码(YunaiV)

原文发表时间:2018-01-06

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏腾讯Bugly的专栏

Android 动态链接库加载原理及 HotFix 方案介绍

引言 随着项目中动态链接库越来越多,我们也遇到了很多奇怪的问题,比如只在某一种 OS 上会出现的 java.lang.UnsatisfiedLinkError,...

5487
来自专栏精讲JAVA

java面试线程必备知识点,怼死面试官,从我做起

内存屏障:限制命令操作顺序,有LoadLoad、LoadStore、LoadStore、StroreStreo四种屏障

1554
来自专栏北京马哥教育

Redis 基础、高级特性与性能调优 | 一文看全

本文将从Redis的基本特性入手,通过讲述Redis的数据结构和主要命令对Redis的基本能力进行直观介绍。之后在性能调优等方面进行更深入的介绍和指导。 概述...

8206
来自专栏Felix的技术分享

webview接入HttpDNS实践

5746
来自专栏Seebug漏洞平台

从补丁到漏洞分析——记一次joomla漏洞应急

作者:LoRexxar'@知道创宇404实验室 2018年1月30日,joomla更新了3.8.4版本,这次更新修复了4个安全漏洞,以及上百个bug修复。 ht...

35812
来自专栏玄魂工作室

看代码学PHP渗透(3) - 实例化任意对象漏洞

大家好,我们是红日安全-代码审计小组。最近我们小组正在做一个PHP代码审计的项目,供大家学习交流,我们给这个项目起了一个名字叫 PHP-Audit-Labs 。...

7211
来自专栏互联网技术栈

Java中的锁[原理、锁优化、CAS、AQS]

结论:如果volatile变量修饰符使用恰当的话,它比synchronized的使用和执行成本更低,因为它不会引起线程上下文的切换和调度。

3415
来自专栏有趣的django

Django rest framework源码分析(1)----认证

一、基础 1.1.安装 两种方式: github pip直接安装 pip install django-rest-framework 1.2.需要先了解的一...

65411
来自专栏决胜机器学习

《Redis设计与实现》读书笔记(三十) ——Redis集群节点复制与故障转移

《Redis设计与实现》读书笔记(三十) ——Redis集群节点复制与故障转移 (原创内容,转载请注明来源,谢谢) 1、概述 redis集群的...

3634
来自专栏沈唁志

ThinkPHP5框架与ThinkPHP3.2的对比区别

6442

扫码关注云+社区

领取腾讯云代金券