查看原文
其他

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

2017-08-19 王文斌(芋艿) 芋道源码

🙂🙂🙂关注微信公众号:【芋道源码】有福利:

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

道友,帮我分享一波怎么样!


    您可能也对以下帖子感兴趣

    文章有问题?点此查看未经处理的缓存