【Mybatis源码分析】解析语句标签_Select|Update|Insert|Delete

一、前言

在阐述解析语句标签之前，得先知道我们的语句标签内容最后被封装到Configuration哪？（都应该知道 Mybatis 通过的是 XMLConfigBuilder 去解析 xml 然后封装到Configuration对象中传递给 SqlSessionFactory 去往下执行）。

而语句标签的解析内容，即被封装到了 Configuration 中的 mappedStatements Map对象中，即如下属性：

protected final Map<String, MappedStatement> mappedStatements = new StrictMap<MappedStatement>(
      "Mapped Statements collection")
          .conflictMessageProducer((savedValue, targetValue) -> ". please check " + savedValue.getResource() + " and "
              + targetValue.getResource());

其 key 对应的是标签 id，而 MappedStatement 是对语句标签内容的封装实例类（它是一个 final 类，最终是通过内部的 Builder 类 build 构建出来的，未向外提供构造），你可以理解为一条 Select|Update|Insert|Delete 语句标签映射着一个 MappedStatement，就如同一个ResultMap 字段对应着一个 ResultMapping 一样。内部属性如下：

  private String resource;// 源自于哪个mapper文件
  private Configuration configuration;// 配置文件
  private String id;// id
  private Integer fetchSize;// 每次从服务器获取的数量
  private Integer timeout;// 最长访问数据库的时间
  private StatementType statementType;// STATEMENT、PREPARED、CALLABLE
  private ResultSetType resultSetType;
  private SqlSource sqlSource;// 对SQL的包装
  private Cache cache;// 缓存
  private ParameterMap parameterMap;
  private List<ResultMap> resultMaps;// 结果映射
  private boolean flushCacheRequired;// 是否需要缓存刷新
  private boolean useCache;// 是否使用二级缓存
  private boolean resultOrdered;
  private SqlCommandType sqlCommandType;// UNKNOWN、INSERT、UPDATE、SELECT
  private KeyGenerator keyGenerator;// 主键生成器
  private String[] keyProperties;
  private String[] keyColumns;
  private boolean hasNestedResultMaps;// 是否存在嵌套查询结果集
  private String databaseId;
  private Log statementLog;
  private LanguageDriver lang;// 语言驱动，永磊解析动态或静态SQL
  private String[] resultSets;// resultset
  private boolean dirtySelect;

很多属性都是大伙熟知的，像 SqlSource、LanguageDriver 在我之前的博客中也有对其的源码解析，后者是对前者的获取，前者是获取最终要执行的 sql 的封装（boundsql）。【Mybatis源码分析】动态标签的底层原理，DynamicSqlSource源码分析

大概流程就如下图这样：

前言先到这，开始真正的源码分析。

二、语句标签的源码分析

这边的话由于是博客，不会从头到尾对源码进行分析，只对核心部分进行解析，在进行之前，先对 Mybatis 对 xml 解析时，使用到的构造器总览构个图，了解了整体对源码分析更有帮助（自身观看源码后得出来的，如有问题，评论区可以指出）。（图虽然直观，但想出来感觉太酷了，也不知道如果我来设计的话要吃几个核桃才能设计出来）

解析语句标签的核心内容在 XMLStatementBuilder.parseStatementNode 方法中，解析完标签内容然后将其通过 MapperBuilderAssistant 对象映射成 MapperStatement 然后封装到 Configuration 中的 mappedStatements 中。

核心源码如下-有删减（可以大致看看，细节或者说主要部分，下面有图片详细解释）：

public void parseStatementNode() {
    String id = context.getStringAttribute("id");

    String nodeName = context.getNode().getNodeName();
    SqlCommandType sqlCommandType = SqlCommandType.valueOf(nodeName.toUpperCase(Locale.ENGLISH));
    boolean isSelect = sqlCommandType == SqlCommandType.SELECT;
    boolean flushCache = context.getBooleanAttribute("flushCache", !isSelect);
    boolean useCache = context.getBooleanAttribute("useCache", isSelect);
    boolean resultOrdered = context.getBooleanAttribute("resultOrdered", false);

    // Include Fragments before parsing
    // 这是对 sql 片段的引用，在下面对 sql 标签进行了源码分析
    // 这里是通过 include 标签对 sql 标签内容的引用
    // 可以说是替换内容吧
    XMLIncludeTransformer includeParser = new XMLIncludeTransformer(configuration, builderAssistant);
    includeParser.applyIncludes(context.getNode());

    String parameterType = context.getStringAttribute("parameterType");
    Class<?> parameterTypeClass = resolveClass(parameterType);
		
	// 这里的话可以自定义 LanguageDriver 对象然后去使用然后获取对应的 SqlSource对象
	// 这里的话默认是 XMLLanguageDriver
    String lang = context.getStringAttribute("lang");
    LanguageDriver langDriver = getLanguageDriver(lang);

    // Parse selectKey after includes and remove them.
    processSelectKeyNodes(id, parameterTypeClass, langDriver);

    // Parse the SQL (pre:  and  were parsed and removed)
    KeyGenerator keyGenerator;
    String keyStatementId = id + SelectKeyGenerator.SELECT_KEY_SUFFIX;
    keyStatementId = builderAssistant.applyCurrentNamespace(keyStatementId, true);
    if (configuration.hasKeyGenerator(keyStatementId)) {
      keyGenerator = configuration.getKeyGenerator(keyStatementId);
    } else {
      keyGenerator = context.getBooleanAttribute("useGeneratedKeys",
          configuration.isUseGeneratedKeys() && SqlCommandType.INSERT.equals(sqlCommandType))
              ? Jdbc3KeyGenerator.INSTANCE : NoKeyGenerator.INSTANCE;
    }

    SqlSource sqlSource = langDriver.createSqlSource(configuration, context, parameterTypeClass);

	// 去得到最后执行语句准备使用的语句类型，默认是 PreparedStatement
    StatementType statementType = StatementType
        .valueOf(context.getStringAttribute("statementType", StatementType.PREPARED.toString()));

    String parameterMap = context.getStringAttribute("parameterMap");
    
    // 获取返回值类型
    String resultType = context.getStringAttribute("resultType");
    Class<?> resultTypeClass = resolveClass(resultType);
    String resultMap = context.getStringAttribute("resultMap");
    if (resultTypeClass == null && resultMap == null) {
      resultTypeClass = MapperAnnotationBuilder.getMethodReturnType(builderAssistant.getCurrentNamespace(), id);
    }
   
   // 通过助手去构建 MappedStatement，
   // 并且封装进 configuration 中的 mappedStatements中
    builderAssistant.addMappedStatement(id, sqlSource, statementType, sqlCommandType, fetchSize, timeout, parameterMap,
        parameterTypeClass, resultMap, resultTypeClass, resultSetTypeEnum, flushCache, useCache, resultOrdered,
        keyGenerator, keyProperty, keyColumn, databaseId, langDriver, resultSets, dirtySelect);
  }

测试的 xml：

    <sql id="person_test">
        id,name,age,sex
    sql>

    <select id="hhh" resultType="com.ncpowernode.mybatis.bean.Person">
        select <include refid="person_test"/>
        from t_person
    select>

下面我认为构建过程中几个主要的属性解析特地说明一下：

你没配置 lang 属性的话，默认就是通过 XMLLanguageDriver 进行解析的。

按照测试的语句的话，我是没用使用动态SQL的，所以SQLSource解析结果应该是RawSQLSource对象。

没有去指定对应的 StatementType 语句类型的话，默认就是使用 PreparedStatement。

如果是 Select 语句标签的话，需要指定 resultType 属性或 resultMap 的原因（都没指定的话，就把 namespace 指定的类当做返回对象）：

而后就通过 Mapper 助手去构建 MappedStatement 对象，并且映射咯。

即通过 MappedStatement.Builder 进行构建，然后封装到 configuration 中。

封装的话，就直接 put(全限定id，MappedStatement对象咯）

三、sql 标签的解析

测试代码如下：

	<sql id="person_test">
        id,name,age,sex
    sql>

    <select id="hhh" resultType="com.ncpowernode.mybatis.bean.Person">
        select <include refid="person_test"/>
        from t_person
    select>    

    @Test
    public void testSqlTag(){
        SqlSessionFactoryBuilder builder = new SqlSessionFactoryBuilder();
        SqlSessionFactory sqlSessionFactory = builder.build(Thread.currentThread().getContextClassLoader().getResourceAsStream("mybatis-config.xml"));
        SqlSession sqlSession = sqlSessionFactory.openSession();
        PersonMapper mapper = sqlSession.getMapper(PersonMapper.class);
        List<Person> xx = mapper.hhh();
    }

核心解析源码如下（本质就是将 sql 片段的内容封装到 sqlFragments 这个 map 中，然后后续供语句标签里使用）

  private void sqlElement(List<XNode> list, String requiredDatabaseId) {
    for (XNode context : list) {
      String databaseId = context.getStringAttribute("databaseId");
      String id = context.getStringAttribute("id");
      // 合并成全id，namespace.id 这种形式，全限定id
      id = builderAssistant.applyCurrentNamespace(id, false);
      if (databaseIdMatchesCurrent(id, databaseId, requiredDatabaseId)) {
        sqlFragments.put(id, context);
      }
    }
  }

四、总结

• 其实通过这些源码分析也知道了：为什么默认是使用 XMLLanguageDriver 去获取 SQLSource；为什么默认是使用 PreparedStatement 去操纵最后的 SQL；查询Select语句中，不指定resultType 、resultMap 属性行不行？行，但默认就是返回你 namespace 指定的类。
• 我们可以通过配置标签的 lang 属性，来指定对应的 LanguageDriver 实现类去获取 SqlSource.当然我觉得默认的已经很棒了，这个应该用不着自己写。
• 在 mapper 文件中，使用了 sql 标签，可以使用 include 标签去使用 sql 标签中的内容。

下面我画的俩图感觉还挺形象：

这里说点题外话：在 XMLScriptBuilder 中解析动态标签时，${} 也是被解析成动态sql，对应的动态 SQLNode 是 TextSqlNode，内部 apply 方法是通过 GenericTokenParser 解析完然后封装到 DynamicContext 中的。而解析 #{} 时是不被算作为动态 sql 的，这是因为不管是 RawSqlSource 还是 DynamicSqlSource，都会通过 SqlSourceBuilder.parse 方法对 #{} 进行处理。（在上次博客中我自己也是理解的模模糊糊，这里明确后所以再指明一下）