数据库[分库分表]中间件 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

1. UPDATE [LOW_PRIORITY] [IGNORE] table_reference

2.    SET col_name1={expr1|DEFAULT} [, col_name2={expr2|DEFAULT}] ...

3.    [WHERE where_condition]

4.    [ORDER BY ...]

5.    [LIMIT row_count]

• 第二种：Multiple-table syntax

1. UPDATE [LOW_PRIORITY] [IGNORE] table_references

2.    SET col_name1={expr1|DEFAULT} [, col_name2={expr2|DEFAULT}] ...

3.    [WHERE where_condition]

Sharding-JDBC 目前仅支持第一种。业务场景上使用第二种的很少很少。

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

1. // AbstractUpdateParser.java

2. @Override

3. public UpdateStatement parse() {

4.   sqlParser.getLexer().nextToken(); // 跳过 UPDATE

5.   skipBetweenUpdateAndTable(); // 跳过关键字，例如：MYSQL 里的 LOW_PRIORITY、IGNORE

6.   sqlParser.parseSingleTable(updateStatement); // 解析表

7.   parseSetItems(); // 解析 SET

8.   sqlParser.skipUntil(DefaultKeyword.WHERE);

9.   sqlParser.setParametersIndex(parametersIndex);

10.   sqlParser.parseWhere(updateStatement);

11.   return updateStatement; // 解析 WHERE

12. }

Sharding-JDBC 正在收集使用公司名单：传送门。
🙂 你的登记，会让更多人参与和使用 Sharding-JDBC。传送门
Sharding-JDBC 也会因此，能够覆盖更多的业务场景。传送门
登记吧，骚年！传送门

2. UpdateStatement

更新SQL 解析结果。

1. public final class UpdateStatement extends AbstractSQLStatement {

2. }

😈 对，没有其他属性。

我们来看下 UPDATE t_user SET nickname=?,age=?WHERE user_id=? 的解析结果：

3. #parse()

3.1 #skipBetweenUpdateAndTable()

在 UPDATE 和 表名 之间有些词法，对 SQL 路由和改写无影响，进行跳过。

1. // MySQLUpdateParser.java

2. @Override

3. protected void skipBetweenUpdateAndTable() {

4.   getSqlParser().skipAll(MySQLKeyword.LOW_PRIORITY, MySQLKeyword.IGNORE);

5. }

6. // OracleUpdateParser.java

7. @Override

8. protected void skipBetweenUpdateAndTable() {

9.   getSqlParser().skipIfEqual(OracleKeyword.ONLY);

10. }

3.2 #parseSingleTable()

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

3.3 #parseSetItems()

解析 SET后语句。

1. // AbstractUpdateParser.java

2. /**

3. * 解析多个 SET 项

4. */

5. private void parseSetItems() {

6.   sqlParser.accept(DefaultKeyword.SET);

7.   do {

8.       parseSetItem();

9.   } while (sqlParser.skipIfEqual(Symbol.COMMA)); // 以 "," 分隔

10. }

11. /**

12. * 解析单个 SET 项

13. */

14. private void parseSetItem() {

15.   parseSetColumn();

16.   sqlParser.skipIfEqual(Symbol.EQ, Symbol.COLON_EQ);

17.   parseSetValue();

18. }

19. /**

20. * 解析单个 SET 项

21. */

22. private void parseSetColumn() {

23.   if (sqlParser.equalAny(Symbol.LEFT_PAREN)) {

24.       sqlParser.skipParentheses();

25.       return;

26.   }

27.   int beginPosition = sqlParser.getLexer().getCurrentToken().getEndPosition();

28.   String literals = sqlParser.getLexer().getCurrentToken().getLiterals();

29.   sqlParser.getLexer().nextToken();

30.   if (sqlParser.skipIfEqual(Symbol.DOT)) { // 字段有别名

31.       // TableToken

32.       if (updateStatement.getTables().getSingleTableName().equalsIgnoreCase(SQLUtil.getExactlyValue(literals))) {

33.           updateStatement.getSqlTokens().add(new TableToken(beginPosition - literals.length(), literals));

34.       }

35.       sqlParser.getLexer().nextToken();

36.   }

37. }

38. /**

39. * 解析单个 SET 值

40. */

41. private void parseSetValue() {

42.   sqlParser.parseExpression(updateStatement);

43.   parametersIndex = sqlParser.getParametersIndex();

44. }

3.4 #parseWhere()

解析 WHERE 条件。解析代码：《SQL 解析（二）之SQL解析》的#parseWhere()小节。

666. 彩蛋

😝 比更新SQL解析是不是简单，更不用对比查询SQL解析。😳有一种在水更的感觉。嘿嘿，下一篇（《删除SQL解析》）会更加容易。

道友，帮我分享一波怎么样！