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);
}
相关子类
- RawSqlSource : 封装xml中insert、delete、update、select、selectKey标签或java文件中@Insert、@Update、@Delete、@Select、@SelectKey注解的解析结果,并且解析结果中只存在StaticTextSqlNode (MixedSqlNode除外)
- DynamicSqlSource: 封装xml中insert、delete、update、select、selectKey标签或java文件中@Insert、@Update、@Delete、@Select、@SelectKey注解的解析结果,并且解析结果中含有除StaticTextSqlNode外的其他Node(MixedSqlNode除外)
- StaticSqlSource : RawSqlSource和DynamicSqlSource的辅助类
- ProviderSqlSource : 封装java文件中@InsertProvider、@UpdateProvider、@DeleteProvider、@SelectProvider注解的解析结果
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。