Mybatis之SqlNodeSqlSource

SqlNode

SqlNode接口

apply()是SqlNode 接口中定义的唯一方法，该方法会根据用户传入的实参，参数解析该SqlNode所记录的动态SQL节点，并调用DynamicContext.appendSql()方法将解析后的SQL片段追加到DynamicContext.sqlBuilder中保存。当SQL节点下的所有SqlNode 完成解析后,我们就可以从DynamicContext中获取一条动态生成的、完整的SQL语句

public interface SqlNode {

    boolean apply(DynamicContext context);
}

SqlNode子类实现

StaticTextSqlNode
MixedSqlNode
TextSqlNode
ForeachSqlNode
VarDeclSqlNode
IfSqlNode
ChooseSqlNode
TrimSqlNode
- WhereSqlNode
- SetSqlNode

MixedSqlNode

MixedSqlNode 中使用contents 字段(List<SqlNode>类型)记录其子节点对应的SqINode对象集合，其apply()方法会循环调用contents集合中所有SqlNode 对象的apply()方法

StaticTextSqlNode

StaticTextSqlNode中使用text字段(String类型)记录了对应的非动态SQL语句节点，其apply()方法直接将text字段追加到DynamicContext.sqlBuilder字段中

TextSqlNode

TextSqlNode表示的是包含“${}”占位符的动态SQL节点。TextSqlNode.apply()方法会使用GenericTokenParser解析“${}”占位符，并直接替换成用户给定的实际参数值

IfSqlNode

SqlNode对应的动态SQL 节点是<If>节点

public class IfSqlNode implements SqlNode {
    // 对象用于解析<if>节点的test表达式的值
    private final ExpressionEvaluator evaluator;
            
    // 记录了<if>节点中的test表达式
    private final String test;
    
    // 记录了<if>节点的子节点
    private final SqlNode contents;

}

TrimSqlNode & WhereSqlNode & SetSqlNode

TrimSqlNode 会根据子节点的解析结果，添加或删除相应的前缀或后缀。

public class TrimSqlNode implements SqlNode {
    // 该<trim>节点的子节点
    private final SqlNode contents;

    // 记录了前缀字符串(为<trim>节点包裹的SQL语句添加的前级)
    private final String prefix;

    // 记录了后缀字符串(为<trim>节点包裹的SQL语句添加的后缀)
    private final String suffix;

    // 如果<trim>节点包裹的 SQL语句是空语句(经常出现在if判断为否的情况下),删除指定的前辍
    private final List<String> prefixesToOverride;

    // 如果<trim>节点包裹的 SQL语句是空语句(经常出现在if判断为否的情况下),删除指定的后缀
    private final List<String> suffixesToOverride;

    private final Configuration configuration;
}

ChooseSqlNode

如果在编写动态SQL语句时需要类似Java中的switch语句的功能,可以考虑使用<choose>、<when>和<otherwise>三个标签的组合。MyBatis会将<choose>标签解析成ChooseSqlNode, <when>标签解析成 IfSqlNode，将<otherwise>标签解析成MixedSqlNode。

public class ChooseSqlNode implements SqlNode {
            
    // <otherwise>节点对应的SqlNode
    private final SqlNode defaultSqlNode;
     
    // <when>节点对应的IfSqlNode 集合
    private final List<SqlNode> ifSqlNodes;

}

VarDeclSqlNode

VarDeclSqlNode 表示的是动态SQL语句中的<bind>节点,该节点可以从OGNL表达式中创建一个变量,并将其记录到上下文中。在VarDeclSqlNode中通过name字段记录<bind>节点的name属性值，expression字段记录<bind>节点的value属性值。

public class VarDeclSqlNode implements SqlNode {

    // <bind>节点的name属性值
    private final String name;

    // <bind>节点的value属性值
    private final String expression;
}

ForEachSqlNode

在动态SQL语句中构建IN条件语句的时候，常需要对一个集合进行迭代，MyBatis提供了<foreach>标签实现该功能。在使用<foreach>标签迭代集合时，不仅可以使用集合的元素和索引值，还可以在循环开始之前或结束之后添加指定的字符串，也允许在迭代过程中添加指定的分隔符。

public class ForEachSqlNode implements SqlNode {
    public static final String ITEM_PREFIX = "__frch_";

    // 用于判断循环的终止条件
    private final ExpressionEvaluator evaluator;

    // 迭代的集合表达式
    private final String collectionExpression;

    // 记录了该ForeachSqlNode 节点的子节点
    private final SqlNode contents;

    // 在循环开始前要添加的字符串
    private final String open;

    // 在循环结束后要添加的字符串
    private final String close;

    // 循环过程中，每项之间的分隔符
    private final String separator;

    // index是当前迭代的次数，item的值是本次迭代的元素。若迭代集合是Map，则index是键,item是值
    private final String item;
    private final String index;

    // 配置对象
    private final Configuration configuration;
}

SqlNode的解析流程

SqlNode的解析流程，主要是由XMLScriptBuilder这个类来完成的，其构造方法会调用initNodeHandlerMap这个方法，这个方法会注册很多handler，即不同的标签将会由不同的handler处理。方法明细如下：

private void initNodeHandlerMap() {
    nodeHandlerMap.put("trim", new XMLScriptBuilder.TrimHandler());
    nodeHandlerMap.put("where", new XMLScriptBuilder.WhereHandler());
    nodeHandlerMap.put("set", new XMLScriptBuilder.SetHandler());
    nodeHandlerMap.put("foreach", new XMLScriptBuilder.ForEachHandler());
    nodeHandlerMap.put("if", new XMLScriptBuilder.IfHandler());
    nodeHandlerMap.put("choose", new XMLScriptBuilder.ChooseHandler());
    nodeHandlerMap.put("when", new XMLScriptBuilder.IfHandler());
    nodeHandlerMap.put("otherwise", new XMLScriptBuilder.OtherwiseHandler());
    nodeHandlerMap.put("bind", new XMLScriptBuilder.BindHandler());
}

除了BindHandler，上述所有的handler的handleNode方法，都会调用parseDynamicTags()方法。即sql的解析过程，我们可以看做是parseDynamicTags()方法的递归调用过程。一个子节点解析完成，会被封装成MixedSqlNode对象。parseDynamicTags源码如下：

  protected MixedSqlNode parseDynamicTags(XNode node) {
    List<SqlNode> contents = new ArrayList<>();
    NodeList children = node.getNode().getChildNodes();
    for (int i = 0; i < children.getLength(); i++) {
      XNode child = node.newXNode(children.item(i));
      if (child.getNode().getNodeType() == Node.CDATA_SECTION_NODE || child.getNode().getNodeType() == Node.TEXT_NODE) {
        String data = child.getStringBody("");
        TextSqlNode textSqlNode = new TextSqlNode(data);
        if (textSqlNode.isDynamic()) {
          contents.add(textSqlNode);
          isDynamic = true;
        } else {
          contents.add(new StaticTextSqlNode(data));
        }
      } else if (child.getNode().getNodeType() == Node.ELEMENT_NODE) { // issue #628
        String nodeName = child.getNode().getNodeName();
        NodeHandler handler = nodeHandlerMap.get(nodeName);
        if (handler == null) {
          throw new BuilderException("Unknown element <" + nodeName + "> in SQL statement.");
        }
        handler.handleNode(child, contents);
        isDynamic = true;
      }
    }
    return new MixedSqlNode(contents);
  }

图示一个复杂SQL的解析结果树

<select id="listDataByCondition" resultType="map">
    select *
    from ${tableName}
    <where>
        and 1 = 1
        <if test="id != null or ids != null">
            <choose>
                <when test="id != null">
                    and id = #{id}
                </when>
                <otherwise>
                    and id in
                    <foreach collection="ids" item="id" open="(" separator="," close=")">
                        #{id}
                    </foreach>
                </otherwise>
            </choose>
        </if>
        <if test="search != null and fieldName != null">
            <bind name="search" value="'%'+ search + '%' "/>
            and ${fieldName} like #{search}
        </if>
    </where>
</select>

@MapKey("id")
List<Map<String, Object>> listDataByCondition(Map<String, Object> map);

xml解析结果树,如下所示

演示不同查询条件,SQL的拼接结果

查询1

@Test
public void listDataByCondition() {
    SqlSession sqlSession = sqlSessionFactory.openSession();
    CommentMapper mapper = sqlSession.getMapper(CommentMapper.class);
    Map<String, Object> map = new HashMap<>();
    map.put("tableName", "`comment`");
    map.put("id", 1);
    List<Map<String, Object>> data = mapper.listDataByCondition(map);
    System.out.println(data);
}

根据上述查询传入的条件,执行相关Node的apply方法,会动态拼接上图所示①、②、③处，最终sql如下：

select * from `comment` where 1 = 1 and id = #{id}

查询2

@Test
public void listDataByCondition() {
    SqlSession sqlSession = sqlSessionFactory.openSession();
    CommentMapper mapper = sqlSession.getMapper(CommentMapper.class);
    Map<String, Object> map = new HashMap<>();
    map.put("tableName", "`comment`");
    map.put("ids", Arrays.asList(1, 2, 3, 4));
    map.put("fieldName", "content");
    map.put("search", "百");
    List<Map<String, Object>> data = mapper.listDataByCondition(map);
    System.out.println(data);
}

根据上述查询传入的条件,执行相关Node的apply方法,会动态拼接上图所示①、②、④、⑤、⑥、⑦处，最终sql如下：

select * from `comment` 
WHERE 1 = 1 
and id in (#{__frch_id_0},#{__frch_id_1},#{__frch_id_2},#{__frch_id_3}) 
and content like #{search}

SqlSource

SqlSource接口

public interface SqlSource {

  BoundSql getBoundSql(Object parameterObject);

}

getBoundSql

RawSqlSource

RawSqlSource会在构造方法中,直接解析原始sql。解析流程会将原始sql中占位符的名称封装成ParameterMapping对象，然后再将占位符替换成'?'。最后将解析结果赋值给内部属性sqlSource，这个sqlSource的类型是StaticSqlSource

RawSqlSource的getBoundSql()方法,交由这个内部sqlSource获取,即最终会调用StaticSqlSource的getBoundSql()方法。

DynamicSqlSource

DynamicSqlSource的getBoundSql()方法与RawSqlSource的getBoundSql()方法大体一致。只是DynamicSqlSource的getBoundSql()方法，会在解析之前调用rootSqlNode的apply()方法。该方法会依次调用子节点的apply()方法，动态拼接、修剪sql。