來源| OSCHINA 社區(qū)

作者 |華為云開發(fā)者聯(lián)盟

摘要：使用動(dòng)態(tài) SQL 并非一件易事，但借助可用于任何 SQL 映射語句中的強(qiáng)大的動(dòng)態(tài) SQL 語言，MyBatis 顯著地提升了這一特性的易用性。

本文分享自華為云社區(qū)《MyBatis 詳解 - 動(dòng)態(tài) SQL 使用與原理》，作者：龍哥手記 。 動(dòng)態(tài) SQL 是 MyBatis 的強(qiáng)大特性之一。如果你使用過 JDBC 或其它類似的框架，你應(yīng)該能理解根據(jù)不同條件拼接 SQL 語句有多痛苦，例如拼接時(shí)要確保不能忘記添加必要的空格，還要注意去掉列表最后一個(gè)列名的逗號(hào)。利用動(dòng)態(tài) SQL，可以徹底擺脫這種痛苦。 使用動(dòng)態(tài) SQL 并非一件易事，但借助可用于任何 SQL 映射語句中的強(qiáng)大的動(dòng)態(tài) SQL 語言，MyBatis 顯著地提升了這一特性的易用性。 如果你之前用過 JSTL 或任何基于類 XML 語言的文本處理器，你對(duì)動(dòng)態(tài) SQL 元素可能會(huì)感覺似曾相識(shí)。在 MyBatis 之前的版本中，需要花時(shí)間了解大量的元素。借助功能強(qiáng)大的基于 OGNL 的表達(dá)式，MyBatis 3 替換了之前的大部分元素，大大精簡(jiǎn)了元素種類，現(xiàn)在要學(xué)習(xí)的元素種類比原來的一半還要少。

if

choose (when, otherwise)

trim (where, set)

foreach

if

使用動(dòng)態(tài) SQL 最常見情景是根據(jù)條件包含 where 子句的一部分。比如：

  SELECT * FROM BLOG
  WHERE state = ‘ACTIVE’
 
    AND title like #{title}
 

這條語句提供了可選的查找文本功能。如果不傳入 “title”，那么所有處于 “ACTIVE” 狀態(tài)的 BLOG 都會(huì)返回；如果傳入了 “title” 參數(shù)，那么就會(huì)對(duì) “title” 一列進(jìn)行模糊查找并返回對(duì)應(yīng)的 BLOG 結(jié)果（細(xì)心的讀者可能會(huì)發(fā)現(xiàn)，“title” 的參數(shù)值需要包含查找掩碼或通配符字符）。

如果希望通過 “title” 和 “author” 兩個(gè)參數(shù)進(jìn)行可選搜索該怎么辦呢？首先，我想先將語句名稱修改成更名副其實(shí)的名稱；接下來，只需要加入另一個(gè)條件即可。

  SELECT * FROM BLOG WHERE state = ‘ACTIVE’
 
    AND title like #{title}
 
 
    AND author_name like #{author.name}
 

choose、when、otherwise

有時(shí)候，我們不想使用所有的條件，而只是想從多個(gè)條件中選擇一個(gè)使用。針對(duì)這種情況，MyBatis 提供了 choose 元素，它有點(diǎn)像 Java 中的 switch 語句。

還是上面的例子，但是策略變?yōu)椋簜魅肓?“title” 就按 “title” 查找，傳入了 “author” 就按 “author” 查找的情形。若兩者都沒有傳入，就返回標(biāo)記為 featured 的 BLOG（這可能是管理員認(rèn)為，與其返回大量的無意義隨機(jī) Blog，還不如返回一些由管理員挑選的 Blog）。

  SELECT * FROM BLOG WHERE state = ‘ACTIVE’
 
 
      AND title like #{title}
 
 
      AND author_name like #{author.name}
 
 
      AND featured = 1
 
 

trim、where、set

前面幾個(gè)例子已經(jīng)合宜地解決了一個(gè)臭名昭著的動(dòng)態(tài) SQL 問題?，F(xiàn)在回到之前的 “if” 示例，這次我們將 “state = ‘ACTIVE’” 設(shè)置成動(dòng)態(tài)條件，看看會(huì)發(fā)生什么。

  SELECT * FROM BLOG
  WHERE
 
    state = #{state}
 
 
    AND title like #{title}
 
 
    AND author_name like #{author.name}
 

如果沒有匹配的條件會(huì)怎么樣？最終這條 SQL 會(huì)變成這樣：

SELECT * FROM BLOG
WHERE

這會(huì)導(dǎo)致查詢失敗。如果匹配的只是第二個(gè)條件又會(huì)怎樣？這條 SQL 會(huì)是這樣:

SELECT * FROM BLOG
WHERE
AND title like ‘someTitle’

這個(gè)查詢也會(huì)失敗。這個(gè)問題不能簡(jiǎn)單地用條件元素來解決。這個(gè)問題是如此的難以解決，以至于解決過的人不會(huì)再想碰到這種問題。

MyBatis 有一個(gè)簡(jiǎn)單且適合大多數(shù)場(chǎng)景的解決辦法。而在其他場(chǎng)景中，可以對(duì)其進(jìn)行自定義以符合需求。而這，只需要一處簡(jiǎn)單的改動(dòng)：

  SELECT * FROM BLOG
 
 
         state = #{state}
 
 
        AND title like #{title}
 
 
        AND author_name like #{author.name}
 
 

where 元素只會(huì)在子元素返回任何內(nèi)容的情況下才插入 “WHERE” 子句。而且，若子句的開頭為 “AND” 或 “OR”，where 元素也會(huì)將它們?nèi)コ?/p>

如果 where 元素與你期望的不太一樣，你也可以通過自定義 trim 元素來定制 where 元素的功能。比如，和 where 元素等價(jià)的自定義 trim 元素為：

  ...

prefixOverrides 屬性會(huì)忽略通過管道符分隔的文本序列（注意此例中的空格是必要的）。上述例子會(huì)移除所有 prefixOverrides 屬性中指定的內(nèi)容，并且插入 prefix 屬性中指定的內(nèi)容。

用于動(dòng)態(tài)更新語句的類似解決方案叫做 set。set 元素可以用于動(dòng)態(tài)包含需要更新的列，忽略其它不更新的列。比如：

  update Author
 
 username=#{username},
 password=#{password},
 email=#{email},
 bio=#{bio}
 
  where id=#{id}

這個(gè)例子中，set 元素會(huì)動(dòng)態(tài)地在行首插入 SET 關(guān)鍵字，并會(huì)刪掉額外的逗號(hào)（這些逗號(hào)是在使用條件語句給列賦值時(shí)引入的）。

來看看與 set 元素等價(jià)的自定義 trim 元素吧：

  ...

注意，我們覆蓋了后綴值設(shè)置，并且自定義了前綴值。

foreach

動(dòng)態(tài) SQL 的另一個(gè)常見使用場(chǎng)景是對(duì)集合進(jìn)行遍歷（尤其是在構(gòu)建 IN 條件語句的時(shí)候）。比如：

  SELECT *
  FROM POST P
  WHERE ID in
 
        #{item}
 

foreach 元素的功能非常強(qiáng)大，它允許你指定一個(gè)集合，聲明可以在元素體內(nèi)使用的集合項(xiàng)（item）和索引（index）變量。它也允許你指定開頭與結(jié)尾的字符串以及集合項(xiàng)迭代之間的分隔符。這個(gè)元素也不會(huì)錯(cuò)誤地添加多余的分隔符，看它多智能！

提示：你可以將任何可迭代對(duì)象（如 List、Set 等）、Map 對(duì)象或者數(shù)組對(duì)象作為集合參數(shù)傳遞給 foreach。當(dāng)使用可迭代對(duì)象或者數(shù)組時(shí)，index 是當(dāng)前迭代的序號(hào)，item 的值是本次迭代獲取到的元素。當(dāng)使用 Map 對(duì)象（或者 Map.Entry 對(duì)象的集合）時(shí)，index 是鍵，item 是值。

至此，我們已經(jīng)完成了與 XML 配置及映射文件相關(guān)的討論。下一章將詳細(xì)探討 Java API，以便你能充分利用已經(jīng)創(chuàng)建的映射配置。

script

要在帶注解的映射器接口類中使用動(dòng)態(tài) SQL，可以使用 script 元素。比如:

 @Update({""})
 void updateAuthorValues(Author author);

bind

bind 元素允許你在 OGNL 表達(dá)式以外創(chuàng)建一個(gè)變量，并將其綁定到當(dāng)前的上下文。比如：

 
  SELECT * FROM BLOG
  WHERE title LIKE #{pattern}

多數(shù)據(jù)庫(kù)支持

如果配置了 databaseIdProvider，你就可以在動(dòng)態(tài)代碼中使用名為 “_databaseId” 的變量來為不同的數(shù)據(jù)庫(kù)構(gòu)建特定的語句。比如下面的例子：

 
 
      select seq_users.nextval from dual
 
 
      select nextval for seq_users from sysibm.sysdummy1"
 
 
  insert into users values (#{id}, #{name})

動(dòng)態(tài) SQL 中的插入腳本語言

MyBatis 從 3.2 版本開始支持插入腳本語言，這允許你插入一種語言驅(qū)動(dòng)，并基于這種語言來編寫動(dòng)態(tài) SQL 查詢語句。

可以通過實(shí)現(xiàn)以下接口來插入一種語言：

public interface LanguageDriver {
 ParameterHandler createParameterHandler(MappedStatement mappedStatement, Object parameterObject, BoundSql boundSql);
 SqlSource createSqlSource(Configuration configuration, XNode script, Class parameterType);
 SqlSource createSqlSource(Configuration configuration, String script, Class parameterType);
}

實(shí)現(xiàn)自定義語言驅(qū)動(dòng)后，你就可以在 mybatis-config.xml 文件中將它設(shè)置為默認(rèn)語言：

或者，你也可以使用 lang 屬性為特定的語句指定語言：

  SELECT * FROM BLOG

或者，在你的 mapper 接口上添加@Lang 注解：

public interface Mapper {
 @Lang(MyLanguageDriver.class)
 @Select("SELECT * FROM BLOG")
 List selectBlog();
}

提示：可以使用 Apache Velocity 作為動(dòng)態(tài)語言，更多細(xì)節(jié)請(qǐng)參考 MyBatis-Velocity 項(xiàng)目。

你前面看到的所有 xml 標(biāo)簽都由默認(rèn) MyBatis 語言提供，而它由語言驅(qū)動(dòng) org.apache.ibatis.scripting.xmltags.XmlLanguageDriver（別名為 xml）所提供。

動(dòng)態(tài) SQL 解析原理

我們?cè)谑褂?mybatis 的時(shí)候，會(huì)在 xml 中編寫 sql 語句。比如這段動(dòng)態(tài) sql 代碼：

    UPDATE users
 
 
            name = #{name}
 
 
            , age = #{age}
 
 
            , birthday = #{birthday}
 
 
    where id = ${id}

mybatis 底層是如何構(gòu)造這段 sql 的？下面帶著這個(gè)疑問，我們一步一步分析。

關(guān)于動(dòng)態(tài) SQL 的接口和類

SqlNode 接口，簡(jiǎn)單理解就是 xml 中的每個(gè)標(biāo)簽，比如上述 sql 的 update,trim,if 標(biāo)簽：

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

SqlSource Sql 源接口，代表從 xml 文件或注解映射的 sql 內(nèi)容，主要就是用于創(chuàng)建 BoundSql，有實(shí)現(xiàn)類 DynamicSqlSource (動(dòng)態(tài) Sql 源)，StaticSqlSource (靜態(tài) Sql 源) 等：

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

BoundSql 類，封裝 mybatis 最終產(chǎn)生 sql 的類，包括 sql 語句，參數(shù)，參數(shù)源數(shù)據(jù)等參數(shù)：

XNode，一個(gè) Dom API 中的 Node 接口的擴(kuò)展類：

BaseBuilder 接口及其實(shí)現(xiàn)類 (屬性，方法省略了，大家有興趣的自己看), 這些 Builder 的作用就是用于構(gòu)造 sql：

下面我們簡(jiǎn)單分析下其中 4 個(gè) Builder：

XMLConfigBuilder：解析 mybatis 中 configLocation 屬性中的全局 xml 文件，內(nèi)部會(huì)使用 XMLMapperBuilder 解析各個(gè) xml 文件。

XMLMapperBuilder：遍歷 mybatis 中 mapperLocations 屬性中的 xml 文件中每個(gè)節(jié)點(diǎn)的 Builder，比如 user.xml，內(nèi)部會(huì)使用 XMLStatementBuilder 處理 xml 中的每個(gè)節(jié)點(diǎn)。

XMLStatementBuilder：解析 xml 文件中各個(gè)節(jié)點(diǎn)，比如 select,insert,update,delete 節(jié)點(diǎn)，內(nèi)部會(huì)使用 XMLScriptBuilder 處理節(jié)點(diǎn)的 sql 部分，遍歷產(chǎn)生的數(shù)據(jù)會(huì)丟到 Configuration 的 mappedStatements 中。

XMLScriptBuilder：解析 xml 中各個(gè)節(jié)點(diǎn) sql 部分的 Builder。

LanguageDriver 接口及其實(shí)現(xiàn)類 (屬性，方法省略了，大家有興趣的自己看)，該接口主要的作用就是構(gòu)造 sql:

簡(jiǎn)單分析下 XMLLanguageDriver (處理 xml 中的 sql，RawLanguageDriver 處理靜態(tài) sql)：XMLLanguageDriver 內(nèi)部會(huì)使用 XMLScriptBuilder 解析 xml 中的 sql 部分。

源碼分析走起

Spring 與 Mybatis 整合的時(shí)候需要配置 SqlSessionFactoryBean，該配置會(huì)加入數(shù)據(jù)源和 mybatis xml 配置文件路徑等信息：

我們就分析這一段配置背后的細(xì)節(jié)：

SqlSessionFactoryBean 實(shí)現(xiàn)了 Spring 的 InitializingBean 接口，InitializingBean 接口的 afterPropertiesSet 方法中會(huì)調(diào)用 buildSqlSessionFactory 方法 該方法內(nèi)部會(huì)使用 XMLConfigBuilder 解析屬性 configLocation 中配置的路徑，還會(huì)使用 XMLMapperBuilder 屬性解析 mapperLocations 屬性中的各個(gè) xml 文件。部分源碼如下：

由于 XMLConfigBuilder 內(nèi)部也是使用 XMLMapperBuilder，我們就看看 XMLMapperBuilder 的解析細(xì)節(jié)：

我們關(guān)注一下，增刪改查節(jié)點(diǎn)的解析：

XMLStatementBuilder 的解析：

默認(rèn)會(huì)使用 XMLLanguageDriver 創(chuàng)建 SqlSource（Configuration 構(gòu)造函數(shù)中設(shè)置）。

XMLLanguageDriver 創(chuàng)建 SqlSource：

XMLScriptBuilder 解析 sql：

得到 SqlSource 之后，會(huì)放到 Configuration 中，有了 SqlSource，就能拿 BoundSql 了，BoundSql 可以得到最終的 sql。

實(shí)例分析

以下面的 xml 解析大概說下 parseDynamicTags 的解析過程：

    UPDATE users
 
 
            name = #{name}
 
 
            , age = #{age}
 
 
            , birthday = #{birthday}
 
 
    where id = ${id}

parseDynamicTags 方法的返回值是一個(gè) List，也就是一個(gè) Sql 節(jié)點(diǎn)集合。SqlNode 本文一開始已經(jīng)介紹，分析完解析過程之后會(huì)說一下各個(gè) SqlNode 類型的作用。

首先根據(jù) update 節(jié)點(diǎn) (Node) 得到所有的子節(jié)點(diǎn)，分別是 3 個(gè)子節(jié)點(diǎn)：

文本節(jié)點(diǎn) UPDATE users

trim 子節(jié)點(diǎn) ...

文本節(jié)點(diǎn) where id = #

遍歷各個(gè)子節(jié)點(diǎn)：

如果節(jié)點(diǎn)類型是文本或者 CDATA，構(gòu)造一個(gè) TextSqlNode 或 StaticTextSqlNode；

如果節(jié)點(diǎn)類型是元素，說明該 update 節(jié)點(diǎn)是個(gè)動(dòng)態(tài) sql，然后會(huì)使用 NodeHandler 處理各個(gè)類型的子節(jié)點(diǎn)。這里的 NodeHandler 是 XMLScriptBuilder 的一個(gè)內(nèi)部接口，其實(shí)現(xiàn)類包括 TrimHandler、WhereHandler、SetHandler、IfHandler、ChooseHandler 等?？搭惷簿兔靼琢诉@個(gè) Handler 的作用，比如我們分析的 trim 節(jié)點(diǎn)，對(duì)應(yīng)的是 TrimHandler；if 節(jié)點(diǎn)，對(duì)應(yīng)的是 IfHandler... 這里子節(jié)點(diǎn) trim 被 TrimHandler 處理，TrimHandler 內(nèi)部也使用 parseDynamicTags 方法解析節(jié)點(diǎn)。

遇到子節(jié)點(diǎn)是元素的話，重復(fù)以上步驟：

trim 子節(jié)點(diǎn)內(nèi)部有 7 個(gè)子節(jié)點(diǎn)，分別是文本節(jié)點(diǎn)、if 節(jié)點(diǎn)、是文本節(jié)點(diǎn)、if 節(jié)點(diǎn)、是文本節(jié)點(diǎn)、if 節(jié)點(diǎn)、文本節(jié)點(diǎn)。文本節(jié)點(diǎn)跟之前一樣處理，if 節(jié)點(diǎn)使用 IfHandler 處理。遍歷步驟如上所示，下面我們看下幾個(gè) Handler 的實(shí)現(xiàn)細(xì)節(jié)。

IfHandler 處理方法也是使用 parseDynamicTags 方法，然后加上 if 標(biāo)簽必要的屬性：

private class IfHandler implements NodeHandler {
 public void handleNode(XNode nodeToHandle, List targetContents) {
 List 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 標(biāo)簽必要的屬性：

private class TrimHandler implements NodeHandler {
 public void handleNode(XNode nodeToHandle, List targetContents) {
 List 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 節(jié)點(diǎn)：

由于這個(gè) update 方法是個(gè)動(dòng)態(tài)節(jié)點(diǎn)，因此構(gòu)造出了 DynamicSqlSource。DynamicSqlSource 內(nèi)部就可以構(gòu)造 sql 了:

DynamicSqlSource 內(nèi)部的 SqlNode 屬性是一個(gè) MixedSqlNode。然后我們看看各個(gè) SqlNode 實(shí)現(xiàn)類的 apply 方法。下面分析一下各個(gè) SqlNode 實(shí)現(xiàn)類的 apply 方法實(shí)現(xiàn)：

MixedSqlNode：MixedSqlNode 會(huì)遍歷調(diào)用內(nèi)部各個(gè) sqlNode 的 apply 方法。

public boolean apply(DynamicContext context) {
 for (SqlNode sqlNode : contents) {
 sqlNode.apply(context);
 }
 return true;
}

StaticTextSqlNode：直接 append sql 文本。

public boolean apply(DynamicContext context) {
 context.appendSql(text);
 return true;
}

IfSqlNode：這里的 evaluator 是一個(gè) ExpressionEvaluator 類型的實(shí)例，內(nèi)部使用了 OGNL 處理表達(dá)式邏輯。

public boolean apply(DynamicContext context) {
 if (evaluator.evaluateBoolean(test, context.getBindings())) {
 contents.apply(context);
 return true;
 }
 return false;
}

TrimSqlNode：

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 方法也是調(diào)用屬性 contents (一般都是 MixedSqlNode) 的 apply 方法，按照實(shí)例也就是 7 個(gè) SqlNode，都是 StaticTextSqlNode 和 IfSqlNode。最后會(huì)使用 FilteredDynamicContext 過濾掉 prefix 和 suffix。

審核編輯：湯梓紅