數據庫分庫分表中間件 Sharding-JDBC 源碼分析 —— SQL 解析（五）之更新SQL

🙂🙂🙂關注微信公眾號：【芋道源碼】有福利：

1. RocketMQ / MyCAT / Sharding-JDBC 所有源碼分析文章列表
2. RocketMQ / MyCAT / Sharding-JDBC 中文注釋源碼 GitHub 地址
3. 您對於源碼的疑問每條留言都將得到認真回復。甚至不知道如何讀源碼也可以請教噢。
4. 新的源碼解析文章實時收到通知。每周更新一篇左右。
5. 認真的源碼交流微信群。

---

本文主要基於 Sharding-JDBC 1.5.0 正式版

• 1. 概述
• 2. UpdateStatement
• 3. #parse()
  • 3.1 #skipBetweenUpdateAndTable()
  • 3.2 #parseSingleTable()
  • 3.3 #parseSetItems()
  • 3.4 #parseWhere()
• 666. 彩蛋

---

1. 概述

本文前置閱讀：

• 《SQL 解析（一）之詞法解析》
• 《SQL 解析（二）之SQL解析》

本文分享更新SQL解析的源碼實現。

更新SQL解析比查詢SQL解析復雜度低的多的多。不同數據庫在插入SQL語法上也統一的多。本文分享 MySQL 更新SQL解析器 MySQLUpdateParser。

MySQL UPDATE 語法一共有 2 種 ：

• 第一種：Single-table syntax

UPDATE [LOW_PRIORITY] [IGNORE] table_reference
    SET col_name1={expr1|DEFAULT} [, col_name2={expr2|DEFAULT}] ...
    [WHERE where_condition]
    [ORDER BY ...]
    [LIMIT row_count]

• 第二種：Multiple-table syntax

UPDATE [LOW_PRIORITY] [IGNORE] table_references
    SET col_name1={expr1|DEFAULT} [, col_name2={expr2|DEFAULT}] ...
    [WHERE where_condition]

Sharding-JDBC 目前僅支持第一種。業務場景上使用第二種的很少很少。

Sharding-JDBC 更新SQL解析主流程如下：

// AbstractUpdateParser.java
@Override
public UpdateStatement parse() {
   sqlParser.getLexer().nextToken(); // 跳過 UPDATE
   skipBetweenUpdateAndTable(); // 跳過關鍵字，例如：MYSQL 里的 LOW_PRIORITY、IGNORE
   sqlParser.parseSingleTable(updateStatement); // 解析表
   parseSetItems(); // 解析 SET
   sqlParser.skipUntil(DefaultKeyword.WHERE);
   sqlParser.setParametersIndex(parametersIndex);
   sqlParser.parseWhere(updateStatement);
   return updateStatement; // 解析 WHERE
}

Sharding-JDBC 正在收集使用公司名單：傳送門。
🙂 你的登記，會讓更多人參與和使用 Sharding-JDBC。傳送門
Sharding-JDBC 也會因此，能夠覆蓋更多的業務場景。傳送門
登記吧，騷年！傳送門

2. UpdateStatement

更新SQL 解析結果。

public final class UpdateStatement extends AbstractSQLStatement {
}

😈 對，沒有其他屬性。

我們來看下 UPDATE t_user SET nickname = ?, age = ? WHERE user_id = ? 的解析結果：

3. #parse()

3.1 #skipBetweenUpdateAndTable()

在 UPDATE 和 表名 之間有些詞法，對 SQL 路由和改寫無影響，進行跳過。

// MySQLUpdateParser.java
@Override
protected void skipBetweenUpdateAndTable() {
   getSqlParser().skipAll(MySQLKeyword.LOW_PRIORITY, MySQLKeyword.IGNORE);
}

// OracleUpdateParser.java
@Override
protected void skipBetweenUpdateAndTable() {
   getSqlParser().skipIfEqual(OracleKeyword.ONLY);
}

3.2 #parseSingleTable()

解析表，請看《SQL 解析（二）之SQL解析》的 #parseSingleTable() 小節。

3.3 #parseSetItems()

解析SET后語句。

// AbstractUpdateParser.java
/**
* 解析多個 SET 項
*/
private void parseSetItems() {
   sqlParser.accept(DefaultKeyword.SET);
   do {
       parseSetItem();
   } while (sqlParser.skipIfEqual(Symbol.COMMA)); // 以 "," 分隔
}
/**
* 解析單個 SET 項
*/
private void parseSetItem() {
   parseSetColumn();
   sqlParser.skipIfEqual(Symbol.EQ, Symbol.COLON_EQ);
   parseSetValue();
}
/**
* 解析單個 SET 項
*/
private void parseSetColumn() {
   if (sqlParser.equalAny(Symbol.LEFT_PAREN)) {
       sqlParser.skipParentheses();
       return;
   }
   int beginPosition = sqlParser.getLexer().getCurrentToken().getEndPosition();
   String literals = sqlParser.getLexer().getCurrentToken().getLiterals();
   sqlParser.getLexer().nextToken();
   if (sqlParser.skipIfEqual(Symbol.DOT)) { // 字段有別名
       // TableToken
       if (updateStatement.getTables().getSingleTableName().equalsIgnoreCase(SQLUtil.getExactlyValue(literals))) {
           updateStatement.getSqlTokens().add(new TableToken(beginPosition - literals.length(), literals));
       }
       sqlParser.getLexer().nextToken();
   }
}
/**
* 解析單個 SET 值
*/
private void parseSetValue() {
   sqlParser.parseExpression(updateStatement);
   parametersIndex = sqlParser.getParametersIndex();
}

3.4 #parseWhere()

解析 WHERE 條件。解析代碼：《SQL 解析（二）之SQL解析》的#parseWhere()小節。

666. 彩蛋

😝 比更新SQL解析是不是簡單，更不用對比查詢SQL解析。😳有一種在水更的感覺。嘿嘿，下一篇（《刪除SQL解析》）會更加容易。

道友，幫我分享一波怎么樣？