Albert's Blog
所有文章 标签 分类 Wiki 收藏夹 关于
Albert's Blog

MyBatis 如何执行一条 SQL 语句 - 今日头条

本文由 简悦 SimpRead 转码， 原文地址 www.toutiao.com

前言 Mybatis 是 Java 开发中比较常用的 ORM 框架。在日常工作中，我们都是直接通过 Spring Boot 自动配置，并直接使用，

Mybatis 是 Java 开发中比较常用的 ORM 框架。在日常工作中，我们都是直接通过 Spring Boot 自动配置，并直接使用，但是却不知道 Mybatis 是如何执行一条 SQL 语句的，而这篇文章就是来揭开 Mybatis 的神秘面纱。

我们要理解 Mybatis 的执行过程，就必须先了解 Mybatis 中都有哪一些重要的类，这些类的职责都是什么？

SqlSession

我们都很熟悉，它对外提供用户和数据库之间交互需要使用的方法，隐藏了底层的细节。它默认是实现类是 DefaultSqlSession

Executor

这个是执行器，SqlSession 中对数据库的操作都是委托给它。它有多个实现类，可以使用不同的功能。

https://p3.toutiaoimg.com/origin/pgc-image/4318dcc01c6545858a9a20367dab1160?from=pc

Configuration

它是一个很重要的配置类，它包含了 Mybatis 的所有有用信息，包括 xml 配置，动态 sql 语句等等，我们到处都可以看到这个类的身影。

MapperProxy

这是一个很重要的代理类，它代理的就是 Mybatis 中映射 SQL 的接口。也就是我们常写的 Dao 接口。

首先，我们需要得到一个 SqlSessionFactory 对象，该对象的作用是可以获取 SqlSession 对象。

1
2
3
4
5
6

// 读取配置
InputStream resourceAsStream = Resources.getResourceAsStream("config.xml");
SqlSessionFactoryBuilder sqlSessionFactoryBuilder = new SqlSessionFactoryBuilder();
// 创建一个 SqlSessionFactory 对象
SqlSessionFactory sqlSessionFactory = sqlSessionFactoryBuilder.build(resourceAsStream);
复制代码

当我们得到一个 SqlSessionFactory 对象之后，就可以通过它的 openSession 方法得到一个 SqlSession 对象。

1
2

SqlSession sqlSession = sqlSessionFactory.openSession(true);
复制代码

最后，我们通过 SqlSession 对象获取 Mapper ，从而可以从数据库获取数据。

1
2
3
4
5

// 获取 Mapper 对象
HeroMapper mapper = sqlSession.getMapper(HeroMapper.class);
// 执行方法，从数据库中获取数据
Hero hero = mapper.selectById(1);
复制代码

我们现在主要关注的就是 getMapper 方法，该方法为我们创建一个代理对象，该代理对象为我们执行 SQL 语句提供了重要的支持。

1
2
3
4
5
6

// SqlSession 对象
@Override
public <T> T getMapper(Class<T> type) {
    return configuration.getMapper(type, this);
}
复制代码

getMapper 方法里面委托 Configuration 对象去获取对应的 Mapper 代理对象，之前说过 Configuration 对象里面包含了 Mybatis 中所有重要的信息，其中就包括我们需要的 Mapper 代理对象，而这些信息都是在读取配置信息的时候完成的，也就是执行 sqlSessionFactoryBuilder.build 方法。

1
2
3
4
5

// Configuration 对象
public <T> T getMapper(Class<T> type, SqlSession sqlSession) {
    return mapperRegistry.getMapper(type, sqlSession);
}
复制代码

我们可以看到它又将获取 Mapper 代理对象的操作委托给了 MapperRegistry 对象（搁着俄罗斯套娃呢？），这个 MapperRegistry 对象里面就存放了我们想要的 Mapper 代理对象，如果你这么想，就错了，实际上，它存放的并不是我们想要的 Mapper 代理对象，而是 Mapper 代理对象的工厂，Mybatis 这里使用到了工厂模式。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

public class MapperRegistry {

  private final Configuration config;
  private final Map<Class<?>, MapperProxyFactory<?>> knownMappers = new HashMap<>();

  public MapperRegistry(Configuration config) {
    this.config = config;
  }

  @SuppressWarnings("unchecked")
  public <T> T getMapper(Class<T> type, SqlSession sqlSession) {
    final MapperProxyFactory<T> mapperProxyFactory = (MapperProxyFactory<T>) knownMappers.get(type);
    if (mapperProxyFactory == null) {
      throw new BindingException("Type " + type + " is not known to the MapperRegistry.");
    }
    try {
      return mapperProxyFactory.newInstance(sqlSession);
    } catch (Exception e) {
      throw new BindingException("Error getting mapper instance. Cause: " + e, e);
    }
  }

  public <T> void addMapper(Class<T> type) {
    if (type.isInterface()) {
      if (hasMapper(type)) {
        throw new BindingException("Type " + type + " is already known to the MapperRegistry.");
      }
      boolean loadCompleted = false;
      try {
        knownMappers.put(type, new MapperProxyFactory<>(type));
        // It's important that the type is added before the parser is run
        // otherwise the binding may automatically be attempted by the
        // mapper parser. If the type is already known, it won't try.
        MapperAnnotationBuilder parser = new MapperAnnotationBuilder(config, type);
        parser.parse();
        loadCompleted = true;
      } finally {
        if (!loadCompleted) {
          knownMappers.remove(type);
        }
      }
    }
  }
}
复制代码

我只保留了 getMapper 方法和 addMapper 方法。

在 getMapper 方法中，它获取的是 MapperProxyFactory 对象，我们通过名称可以得出这是一个 Mapper 代理对象工厂，但是我们是要得到一个 MapperProxy 对象，而不是一个工厂对象，我们再来看 getMapper 方法，它通过 mapperProxyFactory.newInstance 来创建代理对象。

1
2
3
4
5
6
7
8
9

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<>(sqlSession, mapperInterface, methodCache);
    return newInstance(mapperProxy);
}
复制代码

创建了一个 MapperProxy 对象，并且通过 Proxy.newProxyInstance 方法（不会还有人不知道这是 JDK 动态代理吧），创建一个代理对象处理，这个代理对象就是我们想要的结果。这里没有体现出来代理了哪个对象啊？其实 mapperInterface 这是一个成员变量，它引用了需要被代理的对象。而这个成员变量实在创建 MapperProxyFactory 对象的时候赋值的，所以我们每一个需要被代理的接口，在 Mybatis 中都会为它生成一个 MapperProxyFactory 对象，该对象的作用就是为了创建所需要的代理对象。

https://p3.toutiaoimg.com/origin/pgc-image/2644347c1d354be0a8b7065540dba0d4?from=pc

​

当我们获取到代理对象 mapper 之后，就可以执行它里面的方法。

这里使用一个例子：

1
2
3
4
5

// Myabtis 所需要的接口
public interface HeroMapper {
    Hero selectById(Integer id);
}
复制代码

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11

// HeroMapper 接口所对应的 xml 文件
<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper
    PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
    "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="test.HeroMapper">
    <select id="selectById" resultType="test.Hero">
        select * from hero where id = #{id}
    </select>
</mapper>
复制代码

我们执行 selectById 方法，获取一个用户的信息。

1
2
3
4
5

// 获取 Mapper 对象
HeroMapper mapper = sqlSession.getMapper(HeroMapper.class);
// 执行方法，从数据库中获取数据
Hero hero = mapper.selectById(1);
复制代码

通过上面的解析已经知道，这里的 mapper 是一个代理对象的引用，而这个代理类则是 MapperProxy，所以我们主要是去了解 MapperProxy 这个代理类做了什么事情。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

public class MapperProxy<T> implements InvocationHandler, Serializable {
    
  private final SqlSession sqlSession;
  private final Class<T> mapperInterface;
  private final Map<Method, MapperMethodInvoker> methodCache;

  public MapperProxy(SqlSession sqlSession, Class<T> mapperInterface, Map<Method, MapperMethodInvoker> methodCache) {
    this.sqlSession = sqlSession;
    this.mapperInterface = mapperInterface;
    this.methodCache = methodCache;
  }

  @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 {
        return cachedInvoker(method).invoke(proxy, method, args, sqlSession);
      }
    } catch (Throwable t) {
      throw ExceptionUtil.unwrapThrowable(t);
    }
  }

  private MapperMethodInvoker cachedInvoker(Method method) throws Throwable {
      return methodCache.computeIfAbsent(method, m -> {
           return new PlainMethodInvoker(new MapperMethod(mapperInterface, method, sqlSession.getConfiguration()));
      }
  }
    
  private static class PlainMethodInvoker implements MapperMethodInvoker {
      private final MapperMethod mapperMethod;

      public PlainMethodInvoker(MapperMethod mapperMethod) {
          super();
          this.mapperMethod = mapperMethod;
      }

      @Override
      public Object invoke(Object proxy, Method method, Object[] args, SqlSession sqlSession) throws Throwable {
          return mapperMethod.execute(sqlSession, args);
      }
  }
}

复制代码

代理对象执行方法时都是直接执行 invoke() 方法，在这个方法中，我们主要就看一条语句 cachedInvoker(method).invoke(proxy, method, args, sqlSession);

我们首先看 cachedInvoker 方法，它的参数是 Method 类型，所以这个 method 表示的就是我们执行的方法 HeroMapper.selectById，它首先从缓存中获取是否之前已经创建过一个该方法的方法执行器 PlainMethodInvoker 对象，其实这只是一个包装类，可有可无，在工程上来说，有了这个包装类，会更加易于维护。而这个执行器里面只有一个成员对象，这个成员对象就是 MapperMethod，并且这个 MapperMethod 的构造函数中需要传递 HeroMapper、HeroMapper.selectById、Cofiguration 这三个参数。

以上步骤都执行完成之后，接下来我们可以看到执行了 PlainMethodInvoker 的 invoke 方法，而它又将真正的操作委托给了 MapperMethod，执行 MapperMethod 下的 execute 方法，这个方法就是本文章的重点所在。

https://p3.toutiaoimg.com/origin/pgc-image/85a9ae04761c404eb4e68c8a94e67947?from=pc

​

从上面的解析可以知道，最后会执行到这个方法之中。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

public Object execute(SqlSession sqlSession, Object[] args) {
    Object result;
    switch (command.getType()) {
      case INSERT: {
        Object param = method.convertArgsToSqlCommandParam(args);
        result = rowCountResult(sqlSession.insert(command.getName(), param));
        break;
      }
      case UPDATE: {
        Object param = method.convertArgsToSqlCommandParam(args);
        result = rowCountResult(sqlSession.update(command.getName(), param));
        break;
      }
      case DELETE: {
        Object param = method.convertArgsToSqlCommandParam(args);
        result = rowCountResult(sqlSession.delete(command.getName(), param));
        break;
      }
      case SELECT:
        if (method.returnsVoid() && method.hasResultHandler()) {
          executeWithResultHandler(sqlSession, args);
          result = null;
        } else if (method.returnsMany()) {
          result = executeForMany(sqlSession, args);
        } else if (method.returnsMap()) {
          result = executeForMap(sqlSession, args);
        } else if (method.returnsCursor()) {
          result = executeForCursor(sqlSession, args);
        } else {
          Object param = method.convertArgsToSqlCommandParam(args);
          result = sqlSession.selectOne(command.getName(), param);
          if (method.returnsOptional()
              && (result == null || !method.getReturnType().equals(result.getClass()))) {
            result = Optional.ofNullable(result);
          }
        }
        break;
      case FLUSH:
        result = sqlSession.flushStatements();
        break;
      default:
        throw new BindingException("Unknown execution method for: " + command.getName());
    }
    return result;
  }
复制代码

这个方法中，我们可以看到熟悉的几个关键字：select、update、delete、insert，这个就是为了找到执行方式，我们因为是 select 语句，所以分支会走向 select，并且最终会执行到 sqlSession.selectOne 方法中，所以最终饶了一大圈，依然还是回到了我们一开始就提到的 SqlSession 对象中。

在这个方法中，首先会构造参数，也就是我们看到的 convertArgsToSqlCommandParam 方法，它的内部执行方式是按照如下方式来转换参数的：

使用 @param 自定义命名

amethod(@Param int a, @Param int b) 则会构造 map -> [{“a”, a_arg}, {“b”, b_arg}, {“param1”, a_arg}, {“param2”, b_arg}]，a 和 param1 是对参数 a 的命名，a_arg 是传递的实际的值。

虽然只有两个参数，但是最后却会在 Map 存在四个键值对，因为 Mybatis 最后自己会生成以 param 为前缀的参数名称，名称按照参数的位置进行命名。

不使用 @param

amethod(int a, int b)，则会构造 map -> [{“arg0”, a_arg}, {“arg1”, b_arg}, {“param1”, a_arg}, {“param2”, b_arg}]，因为没有对参数进行自定义命名，所以 Myabtis 就对参数取了一个默认的名称，以 arg 为前缀，位置为后缀进行命名。

在参数只有一个，并且参数为集合的情况下，会存放多个键值对：

amethod(Collection a)，这种情况下，会构造 map -> [{“arg0”, a_arg}, {“collection”, a_arg}]

amethod(List a)，这种情况下，会构造 map -> [{“arg0”, a_arg}, {“collection”, a_arg}, {“list”, a_arg}]

amethod(Integer[] a)，这种情况下，会构造 map -> [{“arg0”, a_arg}, {“array”, a_arg}]

但是，如果有两个参数，那么就不会这么存放，而是按照常规的方式：

amethod(List a，List b) 则会构造 map -> [{“arg0”, a_arg}, {“arg1”, b_arg}, {“param1”, a_arg}, {“param2”, b_arg}]

amethod(List a，int b) 则会构造 map -> [{“arg0”, a_arg}, {“arg1”, b_arg}, {“param1”, a_arg}, {“param2”, b_arg}]

不会作为参数的对象

在 Mybatis 中有两个特殊的对象：RowBounds、ResultHandler，这两个对象如果作为参数则不会放入到 map 中，但是会占据位置。

amethod(int a，RowBounds rb, int b)，这种情况下，会构造 map -> [{“arg0”, a_arg}, {“arg2”, b_arg}, {“param1”, a_arg}, {“param2”, b_arg}]

注意这里的 b 参数的命名分别是 arg2 和 param2，arg2 是因为它的位置在参数的第 3 位，而 param2 则是因为它是第 2 个有效参数。

参数构造完成之后，我们就需要寻找需要执行的 SQL 语句了。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

@Override
  public <T> T selectOne(String statement, Object parameter) {
    // Popular vote was to return null on 0 results and throw exception on too many.
    List<T> list = this.selectList(statement, parameter);
    if (list.size() == 1) {
      return list.get(0);
    } else if (list.size() > 1) {
      throw new TooManyResultsException("Expected one result (or null) to be returned by selectOne(), but found: " + list.size());
    } else {
      return null;
    }
  }
复制代码

这里的 statement 虽然是 String 类型的，但是它并不是真正的 SQL 语句，它是一个寻找对应 MapperStatement 对象的名称，在我们的例子中，它就是 test.HeroMapper.selectById ，Mybatis 通过这个名称可以寻找到包含了 SQL 语句的对象。

我们跟踪代码的执行，最后会来到下面这个方法，这是一个包含三个参数的重载方法。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12

@Override
  public <E> List<E> selectList(String statement, Object parameter, RowBounds rowBounds) {
    try {
      MappedStatement ms = configuration.getMappedStatement(statement);
      return executor.query(ms, wrapCollection(parameter), rowBounds, Executor.NO_RESULT_HANDLER);
    } catch (Exception e) {
      throw ExceptionFactory.wrapException("Error querying database.  Cause: " + e, e);
    } finally {
      ErrorContext.instance().reset();
    }
  }
复制代码

在第四行代码中，可以得知它通过 statement 从 Configuration 对象中获取了一个 MapperStatement 对象， MapperStatement 对象包含的信息是由