博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
MyBatis源码解析 ---- MyBatis动态SQL底层原理分析
阅读量:2288 次
发布时间:2019-05-09

本文共 9547 字,大约阅读时间需要 31 分钟。

转自 http://format-blog-image.qiniudn.com/dynamicsql2.png

前言

废话不多说,直接进入文章。

我们在使用mybatis的时候,会在xml中编写sql语句。
比如这段动态sql代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
<
update
id
=
"update"
parameterType
=
"org.format.dynamicproxy.mybatis.bean.User"
>
    
UPDATE users
    
<
trim
prefix
=
"SET"
prefixOverrides
=
","
>
        
<
if
test
=
"name != null and name != ''"
>
            
name = #{name}
        
</
if
>
        
<
if
test
=
"age != null and age != ''"
>
            
, age = #{age}
        
</
if
>
        
<
if
test
=
"birthday != null and birthday != ''"
>
            
, birthday = #{birthday}
        
</
if
>
    
</
trim
>
    
where id = ${id}
</
update
>

mybatis底层是如何构造这段sql的?

这方面的知识网上资料不多,于是就写了这么一篇文章。
下面带着这个疑问,我们一步一步分析。

介绍MyBatis中一些关于动态SQL的接口和类

SqlNode接口,简单理解就是xml中的每个标签,比如上述sql的update,trim,if标签:

1
2
3
public
interface
SqlNode {
  
boolean
apply(DynamicContext context);
}

SqlSource Sql源接口,代表从xml文件或注解映射的sql内容,主要就是用于创建BoundSql,有实现类DynamicSqlSource(动态Sql源),StaticSqlSource(静态Sql源)等:

1
2
3
public
interface
SqlSource {
  
BoundSql getBoundSql(Object parameterObject);
}

BoundSql类,封装mybatis最终产生sql的类,包括sql语句,参数,参数源数据等参数:

XNode,一个Dom API中的Node接口的扩展类。

BaseBuilder接口及其实现类(属性,方法省略了,大家有兴趣的自己看),这些Builder的作用就是用于构造sql:

下面我们简单分析下其中4个Builder:

1 XMLConfigBuilder

解析mybatis中configLocation属性中的全局xml文件,内部会使用XMLMapperBuilder解析各个xml文件。

2 XMLMapperBuilder

遍历mybatis中mapperLocations属性中的xml文件中每个节点的Builder,比如user.xml,内部会使用XMLStatementBuilder处理xml中的每个节点。

3 XMLStatementBuilder

解析xml文件中各个节点,比如select,insert,update,delete节点,内部会使用XMLScriptBuilder处理节点的sql部分,遍历产生的数据会丢到Configuration的mappedStatements中。

4 XMLScriptBuilder

解析xml中各个节点sql部分的Builder。

LanguageDriver接口及其实现类(属性,方法省略了,大家有兴趣的自己看),该接口主要的作用就是构造sql:

简单分析下XMLLanguageDriver(处理xml中的sql,RawLanguageDriver处理静态sql):

XMLLanguageDriver内部会使用XMLScriptBuilder解析xml中的sql部分。

ok, 大部分比较重要的类我们都已经介绍了,下面源码分析走起。

源码分析走起

Spring与Mybatis整合的时候需要配置SqlSessionFactoryBean,该配置会加入数据源和mybatis xml配置文件路径等信息:

1
2
3
4
5
<
bean
id
=
"sqlSessionFactory"
class
=
"org.mybatis.spring.SqlSessionFactoryBean"
>
    
<
property
name
=
"dataSource"
ref
=
"dataSource"
/>
    
<
property
name
=
"configLocation"
value
=
"classpath:mybatisConfig.xml"
/>
    
<
property
name
=
"mapperLocations"
value
=
"classpath*:org/format/dao/*.xml"
/>
</
bean
>

我们就分析这一段配置背后的细节:

SqlSessionFactoryBean实现了Spring的InitializingBean接口,InitializingBean接口的afterPropertiesSet方法中会调用buildSqlSessionFactory方法

buildSqlSessionFactory方法内部会使用XMLConfigBuilder解析属性configLocation中配置的路径,还会使用XMLMapperBuilder属性解析mapperLocations属性中的各个xml文件。

部分源码如下:

由于XMLConfigBuilder内部也是使用XMLMapperBuilder,我们就看看XMLMapperBuilder的解析细节。

我们关注一下,增删改查节点的解析。

XMLStatementBuilder的解析:

默认会使用XMLLanguageDriver创建SqlSource(Configuration构造函数中设置)。

XMLLanguageDriver创建SqlSource:

XMLScriptBuilder解析sql:

得到SqlSource之后,会放到Configuration中,有了SqlSource,就能拿BoundSql了,BoundSql可以得到最终的sql。

实例分析

我以以下xml的解析大概说下parseDynamicTags的解析过程:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
<
update
id
=
"update"
parameterType
=
"org.format.dynamicproxy.mybatis.bean.User"
>
    
UPDATE users
    
<
trim
prefix
=
"SET"
prefixOverrides
=
","
>
        
<
if
test
=
"name != null and name != ''"
>
            
name = #{name}
        
</
if
>
        
<
if
test
=
"age != null and age != ''"
>
            
, age = #{age}
        
</
if
>
        
<
if
test
=
"birthday != null and birthday != ''"
>
            
, birthday = #{birthday}
        
</
if
>
    
</
trim
>
    
where id = ${id}
</
update
>

parseDynamicTags方法的返回值是一个List,也就是一个Sql节点集合。SqlNode本文一开始已经介绍,分析完解析过程之后会说一下各个SqlNode类型的作用。

1 首先根据update节点(Node)得到所有的子节点,分别是3个子节点

(1)文本节点 \n UPDATE users

(2)trim子节点 …

(3)文本节点 \n where id = #{id}

2 遍历各个子节点

(1) 如果节点类型是文本或者CDATA,构造一个TextSqlNode或StaticTextSqlNode

(2) 如果节点类型是元素,说明该update节点是个动态sql,然后会使用NodeHandler处理各个类型的子节点。这里的NodeHandler是XMLScriptBuilder的一个内部接口,其实现类包括TrimHandler、WhereHandler、SetHandler、IfHandler、ChooseHandler等。看类名也就明白了这个Handler的作用,比如我们分析的trim节点,对应的是TrimHandler;if节点,对应的是IfHandler…

这里子节点trim被TrimHandler处理,TrimHandler内部也使用parseDynamicTags方法解析节点

3 遇到子节点是元素的话,重复以上步骤

trim子节点内部有7个子节点,分别是文本节点、if节点、是文本节点、if节点、是文本节点、if节点、文本节点。文本节点跟之前一样处理,if节点使用IfHandler处理

遍历步骤如上所示,下面我们看下几个Handler的实现细节。

IfHandler处理方法也是使用parseDynamicTags方法,然后加上if标签必要的属性。

1
2
3
4
5
6
7
8
9
private
class
IfHandler
implements
NodeHandler {
    
public
void
handleNode(XNode nodeToHandle, List<SqlNode> targetContents) {
      
List<SqlNode> contents = parseDynamicTags(nodeToHandle);
      
MixedSqlNode mixedSqlNode =
new
MixedSqlNode(contents);
      
String test = nodeToHandle.getStringAttribute(
"test"
);
      
IfSqlNode ifSqlNode =
new
IfSqlNode(mixedSqlNode, test);
      
targetContents.add(ifSqlNode);
    
}
}

TrimHandler处理方法也是使用parseDynamicTags方法,然后加上trim标签必要的属性。

1
2
3
4
5
6
7
8
9
10
11
12
private
class
TrimHandler
implements
NodeHandler {
    
public
void
handleNode(XNode nodeToHandle, List<SqlNode> targetContents) {
      
List<SqlNode> contents = parseDynamicTags(nodeToHandle);
      
MixedSqlNode mixedSqlNode =
new
MixedSqlNode(contents);
      
String prefix = nodeToHandle.getStringAttribute(
"prefix"
);
      
String prefixOverrides = nodeToHandle.getStringAttribute(
"prefixOverrides"
);
      
String suffix = nodeToHandle.getStringAttribute(
"suffix"
);
      
String suffixOverrides = nodeToHandle.getStringAttribute(
"suffixOverrides"
);
      
TrimSqlNode trim =
new
TrimSqlNode(configuration, mixedSqlNode, prefix, prefixOverrides, suffix, suffixOverrides);
      
targetContents.add(trim);
    
}
}

以上update方法最终通过parseDynamicTags方法得到的SqlNode集合如下:

trim节点:

由于这个update方法是个动态节点,因此构造出了DynamicSqlSource。

DynamicSqlSource内部就可以构造sql了:

DynamicSqlSource内部的SqlNode属性是一个MixedSqlNode。

然后我们看看各个SqlNode实现类的apply方法

下面分析一下两个SqlNode实现类的apply方法实现:

MixedSqlNode:

1
2
3
4
5
6
public
boolean
apply(DynamicContext context) {
    
for
(SqlNode sqlNode : contents) {
      
sqlNode.apply(context);
    
}
    
return
true
;
}

MixedSqlNode会遍历调用内部各个sqlNode的apply方法。

StaticTextSqlNode:

1
2
3
4
public
boolean
apply(DynamicContext context) {
    
context.appendSql(text);
    
return
true
;
}

直接append sql文本。

IfSqlNode:

1
2
3
4
5
6
7
public
boolean
apply(DynamicContext context) {
    
if
(evaluator.evaluateBoolean(test, context.getBindings())) {
      
contents.apply(context);
      
return
true
;
    
}
    
return
false
;
}

这里的evaluator是一个ExpressionEvaluator类型的实例,内部使用了OGNL处理表达式逻辑。

TrimSqlNode:

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
public
boolean
apply(DynamicContext context) {
    
FilteredDynamicContext filteredDynamicContext =
new
FilteredDynamicContext(context);
    
boolean
result = contents.apply(filteredDynamicContext);
    
filteredDynamicContext.applyAll();
    
return
result;
}
 
public
void
applyAll() {
  
sqlBuffer =
new
StringBuilder(sqlBuffer.toString().trim());
  
String trimmedUppercaseSql = sqlBuffer.toString().toUpperCase(Locale.ENGLISH);
  
if
(trimmedUppercaseSql.length() >
0
) {
    
applyPrefix(sqlBuffer, trimmedUppercaseSql);
    
applySuffix(sqlBuffer, trimmedUppercaseSql);
  
}
  
delegate.appendSql(sqlBuffer.toString());
}
 
private
void
applyPrefix(StringBuilder sql, String trimmedUppercaseSql) {
  
if
(!prefixApplied) {
    
prefixApplied =
true
;
    
if
(prefixesToOverride !=
null
) {
      
for
(String toRemove : prefixesToOverride) {
        
if
(trimmedUppercaseSql.startsWith(toRemove)) {
          
sql.delete(
0
, toRemove.trim().length());
          
break
;
        
}
      
}
    
}
    
if
(prefix !=
null
) {
      
sql.insert(
0
,
" "
);
      
sql.insert(
0
, prefix);
    
}
  
}
}

TrimSqlNode的apply方法也是调用属性contents(一般都是MixedSqlNode)的apply方法,按照实例也就是7个SqlNode,都是StaticTextSqlNode和IfSqlNode。 最后会使用FilteredDynamicContext过滤掉prefix和suffix。

总结

大致讲解了一下mybatis对动态sql语句的解析过程,其实回过头来看看不算复杂,还算蛮简单的。 之前接触mybaits的时候遇到刚才分析的那一段动态sql的时候总是很费解。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
<
update
id
=
"update"
parameterType
=
"org.format.dynamicproxy.mybatis.bean.User"
>
    
UPDATE users
    
<
trim
prefix
=
"SET"
prefixOverrides
=
","
>
        
<
if
test
=
"name != null and name != ''"
>
            
name = #{name}
        
</
if
>
        
<
if
test
=
"age != null and age != ''"
>
            
, age = #{age}
        
</
if
>
        
<
if
test
=
"birthday != null and birthday != ''"
>
            
, birthday = #{birthday}
        
</
if
>
    
</
trim
>
    
where id = ${id}
</
update
>

想搞明白这个trim节点的prefixOverrides到底是什么意思(从字面上理解就是前缀覆盖),而且官方文档上也没这方面知识的说明。我将这段xml改成如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
<
update
id
=
"update"
parameterType
=
"org.format.dynamicproxy.mybatis.bean.User"
>
    
UPDATE users
    
<
trim
prefix
=
"SET"
prefixOverrides
=
","
>
        
<
if
test
=
"name != null and name != ''"
>
            
, name = #{name}
        
</
if
>
        
<
if
test
=
"age != null and age != ''"
>
            
, age = #{age}
        
</
if
>
        
<
if
test
=
"birthday != null and birthday != ''"
>
            
, birthday = #{birthday}
        
</
if
>
    
</
trim
>
    
where id = ${id}
</
update
>

(第二段第一个if节点多了个逗号) 结果我发现这2段xml解析的结果是一样的,非常迫切地想知道这到底是为什么,然后这也促使了我去看源码的决心。最终还是看下来了。

文章有点长,而且讲的也不是非常直观,希望对有些人有帮助。

org.apache.ibatis.builder.xml.XMLMapperBuilder#buildStatementFromContext(java.util.List<org.apache.ibatis.parsing.XNode>)

判断下是否由自定义数据源

org.apache.ibatis.builder.xml.XMLMapperBuilder#buildStatementFromContext(java.util.List<org.apache.ibatis.parsing.XNode>, java.lang.String)
遍历调用org.apache.ibatis.builder.xml.XMLStatementBuilder 去解析 select|insert|update|delete  节点
方法里使用默认的org.apache.ibatis.scripting.xmltags.XMLLanguageDriver  来构建SqlSource
XMLLanguageDriver使用org.apache.ibatis.scripting.xmltags.XMLScriptBuilder.parseScriptNode 来构建SqlSource
org.apache.ibatis.scripting.xmltags.XMLScriptBuilder#parseDynamicTags解析出List<SqlNode>,封装成DynamicSqlSource(动态sql配置)或RawSqlSource(静态sql配置)
通过org.apache.ibatis.scripting.xmltags.XMLScriptBuilder#parseDynamicTags 解析节点时,把select|insert|update|delete 节点中的配置再细分为多个SqlNode,然后再遍历.遍历时,如果发现为文本类型就构建为StaticTextSqlNode或TextSqlNode,如果还是节点类型,就通过内部定义的nodehandler再进行解析.如果节点中还有子节点就递归处理.
有了sqlSource 后就可以开始构建MappedStatement了,构建好MappedStatement,增加到configuration中待使用.
查询的时候根据id获取到MappedStatement,在MappedStatement的getBoundSql中,根据sqlSource把所有sql拼装,并通过org.apache.ibatis.builder.SqlSourceBuilder#parse的处理,把sql中的 在得到BoundSql之前,通过org.apache.ibatis.scripting.xmltags.SqlNode#apply把sql都拼接起来,然org.apache.ibatis.builder.SqlSourceBuilder会把xml配置的变量处理为sql预执行中的"?",最终构建为sqlsource.进而得到BoundSql,BoundSql包含了sql和变量参数值

转载地址:http://xjbnb.baihongyu.com/

你可能感兴趣的文章
MySQL设置及修改root密码
查看>>
MySQL忘记密码重置管理员密码
查看>>
MySQL创建及授权账号
查看>>
MySQL库的基本操作
查看>>
MySQL表的基本操作
查看>>
MySQL数据类型
查看>>
MySQL SQL语句最常见的分类
查看>>
MySQL用户权限
查看>>
MySQL数据备份
查看>>
MySQL使用explain检查索引执行计划
查看>>
MySQL字符集
查看>>
MySQL存储引擎
查看>>
MySQL主从同步
查看>>
MySQL半同步复制
查看>>
MySQL主库宕机从库提权
查看>>
MySQL主主模式
查看>>
MySQL错误代码
查看>>
MySQL binlog的三种模式
查看>>
MySQL利用binlog增量恢复数据库
查看>>
Tomcat多实例多应用
查看>>