前言
MyBatis 是一个被广泛应用的持久化框架。一个简单的使用示例如下所示，先创建会话工厂，然后从会话工厂中打开会话，通过 class 类型和配置生成 Mapper 接口的代理实现，最后使用 Mapper 进行持久化操作。
本文将从 MyBatis 中的 SqlSessionFactoryBuilder、SqlSessionFactory、SqlSession 和 Mapper 几个方面入手简单分析 MyBatis 的实现原理。在后面的系列文章中会进一步具体分析核心类的细节实现。
SqlSessionFactoryBuilder
SqlSessionFactoryBuilder 使用 Builder 模式去生成 SqlSessionFactory，因此只提供了多个 build 方法。这些方法可以接受 XML 配置文件的 Reader 或 InputStream 输入流，也可以传入 environment 指定环境或传入 Properties 作为属性。
在 build 方法的实现中，首先根据传入的输入流、environment 和 Properties 构建 XMLConfigBuilder 对象，然后调用其 parse() 方法解析 XML 文件得到 Configuration 对象，最后创建 SqlSessionFactory 对象并返回。
SqlSessionFactory
SqlSessionFactory 是一个工厂接口，默认实现是 DefaultSqlSessionFactory。SqlSessionFactory 的作用是获取 SqlSession，因此提供了多个 openSession 方法，支持从 DataSource 数据源和一个给定的连接 Connection 中创建 SqlSession。
openSession 方法的底层实现可以分为 5 步：
①从 Configuration 对象中获取环境配置 Environment；
②根据环境配置得到事务工厂 TransactionFactory；
③从事务工厂得到事务 Transaction，Transaction 包装了数据库连接，处理数据库连接的创建、准备、提交、回滚和关闭；
④创建执行器 Executor；
⑤创建 SqlSession，返回 DefaultSqlSession 的实例。
其中从 DataSource 数据源创建 SqlSession 的过程如下所示：
Transaction tx = null;
try {
final Environment environment = configuration.getEnvironment();
final TransactionFactory transactionFactory = getTransactionFactoryFromEnvironment(environment);
tx = transactionFactory.newTransaction(environment.getDataSource(), level, autoCommit);
final Executor executor = configuration.newExecutor(tx, execType);
return new DefaultSqlSession(configuration, executor, autoCommit);
} catch (Exception e) {
closeTransaction(tx); // may have fetched a connection so lets call close()
throw ExceptionFactory.wrapException("Error opening session. Cause: " + e, e);
} finally {
ErrorContext.instance().reset();
}
SqlSession
SqlSession 是一个接口，默认实现是 DefaultSqlSession，提供了多种数据库操作方式，如 select、selectOne、selectList、insert、update、delete、commit、rollback 和 getMapper 等方法。getMapper 方法用于获取 Mapper 接口的代理实现。在 MyBatis 中建议使用 Mapper 接口操作数据库。
数据库的增删改查和事务的提交回滚都是通过 Executor 执行的。Executor 有 3 种类型 SIMPLE、REUSE、BATCH，默认使用简易执行器 SIMPLE，REUSE 类型执行器重用预处理语句，BATCH 类型执行器重用预处理语句和批量更新。Executor 对象的创建在 Configuration 类型的 newExecutor 方法中进行。
Executor 在执行过程中，会用到 StatementHandler、ParameterHandler 和 ResultHandler，其中 StatementHandler 封装了 java.sql.Statement 的相关操作，ParameterHandler 封装了 SQL 对参数的处理，ResultHandler 封装了对返回数据集的处理。Executor 的执行过程，就是对这 3 个对象的调度过程。更多分析在后续文章中进行。
Mapper
Mapper 是通过 JDK 动态代理实现的，在 MapperProxyFactory 中创建 MapperProxy 并进行接口代理封装。对 Mapper 接口的调用实际上是由 MapperProxy 实现的。
@SuppressWarnings("unchecked")
protected T newInstance(MapperProxy<T> mapperProxy) {
return (T) Proxy.newProxyInstance(mapperInterface.getClassLoader(), new Class[] { mapperInterface }, mapperProxy);
}
public T newInstance(SqlSession sqlSession) {
final MapperProxy<T> mapperProxy = new MapperProxy<T>(sqlSession, mapperInterface, methodCache);
return newInstance(mapperProxy);
}
在 MapperProxy 中，实现了 InvocationHandler 的 invoke 方法。methodCache 是一个 ConcurrentHashMap，其中存储了方法与 MapperMethod 的对应关系。从 methodCache 缓存中获取或创建 MapperMethod 对象，然后调用 MapperMethod 对象的 execute 方法执行数据库操作。
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
try {
if (Object.class.equals(method.getDeclaringClass())) {
return method.invoke(this, args);
} else if (isDefaultMethod(method)) {
return invokeDefaultMethod(proxy, method, args);
}
} catch (Throwable t) {
throw ExceptionUtil.unwrapThrowable(t);
}
final MapperMethod mapperMethod = cachedMapperMethod(method);
return mapperMethod.execute(sqlSession, args);
}
private MapperMethod cachedMapperMethod(Method method) {
MapperMethod mapperMethod = methodCache.get(method);
if (mapperMethod == null) {
mapperMethod = new MapperMethod(mapperInterface, method, sqlSession.getConfiguration());
methodCache.put(method, mapperMethod);
}
return mapperMethod;
}
创建 MapperMethod 对象时，会在构造函数中初始化 SqlCommand 和MethodSignature。SqlCommand 包含了数据库操作的名称，格式为 “接口名.操作名称”，以及 XML 中配置的操作类型，如 select、update等，把一个 Mapper 接口与 XML中的一个配置结合起来。MethodSignature 是方法的签名，标记了方法的返回值类型，对于使用 RowBounds（offset 和 limit 配置）、ResultHandler（结果处理回调）作为参数的方法记录参数位置并初始化参数处理器。
在 MapperMethod 的 execute 方法中，根据 SqlCommand 中的配置选择 SqlSession 的方法，根据 MethodSignature 的配置处理传入的参数，调用 SqlSession 的方法进行数据库操作，最后根据 MethodSignature 的返回值类型返回操作结果。
每周 3 篇学习笔记或技术总结，面向有一定基础的 Java 程序员，内容涉及 Java 进阶、虚拟机、MySQL、NoSQL、分布式计算、开源框架等多个领域。关注作者或微信公众号 backend-develop 第一时间获取最新内容。
