数据库【分库分表】中间件 Sharding-JDBC 源码分析 —— SQL 解析（五）之更新SQL解析

 

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

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

1. 概述

本文前置阅读：

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

本文分享更新SQL解析的源码实现。

更新SQL解析比查询SQL解析复杂度低的多的多。不同数据库在插入SQL语法上也统一的多。本文分享 MySQL 更新SQL解析器 MySQLUpdateParser。

MySQL UPDATE 语法一共有 2 种 ：

• 第一种：Single-table syntax

  12345

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

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

  123

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

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

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

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

123456789101112

// AbstractUpdateParser.java@Overridepublic 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}

// 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 解析结果。

12

public final class UpdateStatement extends AbstractSQLStatement {}

public final class UpdateStatement extends AbstractSQLStatement {
}

😈 对，没有其他属性。

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

3. #parse()

3.1 #skipBetweenUpdateAndTable()

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

1234567891011

// MySQLUpdateParser.java@Overrideprotected void skipBetweenUpdateAndTable() {  getSqlParser().skipAll(MySQLKeyword.LOW_PRIORITY, MySQLKeyword.IGNORE);} // OracleUpdateParser.java@Overrideprotected void skipBetweenUpdateAndTable() {  getSqlParser().skipIfEqual(OracleKeyword.ONLY);}

// 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后语句。

1234567891011121314151617181920212223242526272829303132333435363738394041424344

// 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();}

// 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解析》）会更加容易。

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