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

转载

芋道源码 2021-07-30 09:53:59

文章标签 SQL 文章分类 MySQL 数据库

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

本文主要基于 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解析主流程如下：

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

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

  
final
 
class
 
UpdateStatement
 
extends
 
AbstractSQLStatement
 
{

}

???? 对，没有其他属性。

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

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

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

}