資料庫[分庫分表]中介軟體 Sharding-JDBC 原始碼分析 —— SQL 解析(五)之更新SQL解析
阿新 • • 發佈:2022-04-29
本文主要基於 Sharding-JDBC 1.5.0 正式版
- 1. 概述
- 2. UpdateStatement
- 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
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 }
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()小節。