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 的返回值类型返回操作结果。