SQL解析在美團點評中的應用

美團技術團隊發表於2019-03-04

資料庫作為核心的基礎元件,是需要重點保護的物件。任何一個線上的不慎操作,都有可能給資料庫帶來嚴重的故障,從而給業務造成巨大的損失。為了避免這種損失,一般會在管理上下功夫。比如為研發人員制定資料庫開發規範;新上線的SQL,需要DBA進行稽核;維護操作需要經過領導審批等等。而且如果希望能夠有效地管理這些措施,需要有效的資料庫培訓,還需要DBA細心的進行SQL稽核。很多中小型創業公司,可以通過設定規範、進行培訓、完善稽核流程來管理資料庫。

隨著美團點評的業務不斷髮展和壯大,上述措施的實施成本越來越高。如何更多的依賴技術手段,來提高效率,越來越受到重視。業界已有不少基於MySQL原始碼開發的SQL稽核、優化建議等工具,極大的減輕了DBA的SQL稽核負擔。那麼我們能否繼續擴充套件MySQL的原始碼,來輔助DBA和研發人員來進一步提高效率呢?比如,更全面的SQL優化功能;多維度的慢查詢分析;輔助故障分析等。要實現上述功能,其中最核心的技術之一就是SQL解析。

現狀與場景

SQL解析是一項複雜的技術,一般都是由資料庫廠商來掌握,當然也有公司專門提供SQL解析的API。由於這幾年MySQL資料庫中介軟體的興起,需要支援讀寫分離、分庫分表等功能,就必須從SQL中抽出表名、庫名以及相關欄位的值。因此像Java語言編寫的Druid,C語言編寫的MaxScale,Go語言編寫的Kingshard等,都會對SQL進行部分解析。而真正把SQL解析技術用於資料庫維護的產品較少,主要有如下幾個:

  • 美團點評開源的SQLAdvisor。它基於MySQL原生態詞法解析,結合分析SQL中的where條件、聚合條件、多表Join關係給出索引優化建議。
  • 去哪兒開源的Inception。側重於根據內建的規則,對SQL進行稽核。
  • 阿里的Cloud DBA。根據官方文件介紹,其也是提供SQL優化建議和改寫。

上述產品都有非常合適的應用場景,在業界也被廣泛使用。但是SQL解析的應用場景遠遠沒有被充分發掘,比如:

  • 基於表粒度的慢查詢報表。比如,一個Schema中包含了屬於不同業務線的資料表,那麼從業務線的角度來說,其希望提供表粒度的慢查詢報表。
  • 生成SQL特徵。將SQL語句中的值替換成問號,方便SQL歸類。雖然可以使用正規表示式實現相同的功能,但是其Bug較多,可以參考pt-query-digest。比如pt-query-digest中,會把遇到的數字都替換成“?”,導致無法區別不同數字字尾的表。
  • 高危操作確認與規避。比如,DBA不小心Drop資料表,而此類操作,目前還無有效的工具進行回滾,尤其是大表,其後果將是災難性的。
  • SQL合法性判斷。為了安全、審計、控制等方面的原因,美團點評不會讓研發人員直接運算元據庫,而是提供RDS服務。尤其是對於資料變更,需要研發人員的上級主管進行業務上的審批。如果研發人員,寫了一條語法錯誤的SQL,而RDS無法判斷該SQL是否合法,就會造成不必要的溝通成本。

因此為了讓所有有需要的業務都能方便的使用SQL解析功能,我們認為應該具有如下特性。

  • 直接暴露SQL解析介面,使用盡量簡單。比如,輸入SQL,則輸出表名、特徵和優化建議。
  • 介面的使用不依賴於特定的語言,否則維護和使用的代價太高。比如,以HTTP等方式提供服務。

千里之行,始於足下,下面我先介紹下SQL的解析原理。

原理

SQL解析與優化是屬於編譯器範疇,和C等其他語言的解析沒有本質的區別。其中分為,詞法分析、語法和語義分析、優化、執行程式碼生成。對應到MySQL的部分,如下圖

SQL解析在美團點評中的應用

圖1 SQL解析原理

詞法分析

SQL解析由詞法分析和語法/語義分析兩個部分組成。詞法分析主要是把輸入轉化成一個個Token。其中Token中包含Keyword(也稱symbol)和非Keyword。例如,SQL語句 select username from userinfo,在分析之後,會得到4個Token,其中有2個Keyword,分別為select和from:

關鍵字 非關鍵字 關鍵字 非關鍵字
select username from userinfo

通常情況下,詞法分析可以使用Flex來生成,但是MySQL並未使用該工具,而是手寫了詞法分析部分(據說是為了效率和靈活性,參考此文)。具體程式碼在sql/lex.h和sql/sql_lex.cc檔案中。

MySQL中的Keyword定義在sql/lex.h中,如下為部分Keyword

{ "&&",               SYM(AND_AND_SYM)},
{ "<",                SYM(LT)},
{ "<=",               SYM(LE)},
{ "<>",               SYM(NE)},
{ "!=",               SYM(NE)},
{ "=",                SYM(EQ)},
{ ">",                SYM(GT_SYM)},
{ ">=",               SYM(GE)},
{ "<<",               SYM(SHIFT_LEFT)},
{ ">>",               SYM(SHIFT_RIGHT)},
{ "<=>",              SYM(EQUAL_SYM)},
{ "ACCESSIBLE",       SYM(ACCESSIBLE_SYM)},
{ "ACTION",           SYM(ACTION)},
{ "ADD",              SYM(ADD)},
{ "AFTER",            SYM(AFTER_SYM)},
{ "AGAINST",          SYM(AGAINST)},
{ "AGGREGATE",        SYM(AGGREGATE_SYM)},
{ "ALL",              SYM(ALL)},
複製程式碼

詞法分析的核心程式碼在sql/sql_lex.c檔案中的,MySQLLex→lex_one_Token,有興趣的同學可以下載原始碼研究。

語法分析

語法分析就是生成語法樹的過程。這是整個解析過程中最精華,最複雜的部分,不過這部分MySQL使用了Bison來完成。即使如此,如何設計合適的資料結構以及相關演算法,去儲存和遍歷所有的資訊,也是值得在這裡研究的。

a)語法分析樹

SQL語句:

select username, ismale from userinfo where age > 20 and level > 5 and 1 = 1
複製程式碼

會生成如下語法樹。

SQL解析在美團點評中的應用

圖2 語法樹

對於未接觸過編譯器實現的同學,肯定會好奇如何才能生成這樣的語法樹。其背後的原理都是編譯器的範疇,可以參考維基百科的一篇文章,以及該連結中的參考書籍。本人也是在學習MySQL原始碼過程中,閱讀了部分內容。由於編譯器涉及的內容過多,本人精力和時間有限,不做過多探究。從工程的角度來說,學會如何使用Bison去構建語法樹,來解決實際問題,對我們的工作也許有更大幫助。下面我就以Bison為基礎,探討該過程。

b)MySQL語法分析樹生成過程

全部的原始碼在sql/sql_yacc.yy中,在MySQL5.6中有17K行左右程式碼。這裡列出涉及到SQL:

select username, ismale from userinfo where age > 20 and level > 5 and 1 = 1 
複製程式碼

解析過程的部分程式碼摘錄出來。其實有了Bison之後,SQL解析的難度也沒有想象的那麼大。特別是這裡給出瞭解析的脈絡之後。

select /*select語句入口*/:
 
          select_init
 
          {
 
            LEX *lex= Lex;
 
            lex->sql_command= SQLCOM_SELECT;
 
          }
 
        ;
 
select_init:
          SELECT_SYM /*select 關鍵字*/ select_init2
 
        | '(' select_paren ')' union_opt
 
        ;
 
select_init2:
          select_part2
          {
            LEX *lex= Lex;
            SELECT_LEX * sel= lex->current_select;
            if (lex->current_select->set_braces(0))
            {
              my_parse_error(ER(ER_SYNTAX_ERROR));
              MYSQL_YYABORT;
            }
            if (sel->linkage == UNION_TYPE &&
                sel->master_unit()->first_select()->braces)
            {
              my_parse_error(ER(ER_SYNTAX_ERROR));
              MYSQL_YYABORT;
            }
          }
          union_clause
        ;
select_part2:
          {
            LEX *lex= Lex;
            SELECT_LEX *sel= lex->current_select;
            if (sel->linkage != UNION_TYPE)
              mysql_init_select(lex);
            lex->current_select->parsing_place= SELECT_LIST;
          }
  
          select_options select_item_list /*解析列名*/
          {
            Select->parsing_place= NO_MATTER;
          }
          select_into select_lock_type
        ;
 
select_into:
          opt_order_clause opt_limit_clause {}
        | into
        | select_from /*from 字句*/
        | into select_from
        | select_from into
        ;
select_from:
          FROM join_table_list /*解析表名*/ where_clause /*where字句*/ group_clause having_clause
          opt_order_clause opt_limit_clause procedure_analyse_clause
          {
            Select->context.table_list=
              Select->context.first_name_resolution_table=
                Select->table_list.first;
          }
        | FROM DUAL_SYM where_clause opt_limit_clause
          /* oracle compatibility: oracle always requires FROM clause,
             and DUAL is system table without fields.
             Is "SELECT 1 FROM DUAL" any better than "SELECT 1" ?
          Hmmm :) */
        ;
 
where_clause:
          /* empty */  { Select->where= 0; }
        | WHERE
          {
            Select->parsing_place= IN_WHERE;
          }
          expr /*各種表示式*/
          {
            SELECT_LEX *select= Select;
            select->where= $3;
            select->parsing_place= NO_MATTER;
            if ($3)
              $3->top_level_item();
          }
        ;
  
/* all possible expressions */
expr:
           | expr and expr %prec AND_SYM
          {
            /* See comments in rule expr: expr or expr */
            Item_cond_and *item1;
            Item_cond_and *item3;
            if (is_cond_and($1))
            {
              item1= (Item_cond_and*) $1;
              if (is_cond_and($3))
              {
                item3= (Item_cond_and*) $3;
                /*
                  (X1 AND X2) AND (Y1 AND Y2) ==> AND (X1, X2, Y1, Y2)
                */
                item3->add_at_head(item1->argument_list());
                $$ = $3;
              }
              else
              {
                /*
                  (X1 AND X2) AND Y ==> AND (X1, X2, Y)
                */
                item1->add($3);
                $$ = $1;
              }
            }
            else if (is_cond_and($3))
            {
              item3= (Item_cond_and*) $3;
              /*
                X AND (Y1 AND Y2) ==> AND (X, Y1, Y2)
              */
              item3->add_at_head($1);
              $$ = $3;
            }
            else
            {
              /* X AND Y */
              $$ = new (YYTHD->mem_root) Item_cond_and($1, $3);
              if ($$ == NULL)
                MYSQL_YYABORT;
            }
          }
複製程式碼

在大家瀏覽上述程式碼的過程,會發現Bison中嵌入了C++的程式碼。通過C++程式碼,把解析到的資訊儲存到相關物件中。例如表資訊會儲存到TABLE_LIST中,order_list儲存order by子句裡的資訊,where字句儲存在Item中。有了這些資訊,再輔助以相應的演算法就可以對SQL進行更進一步的處理了。

c)核心資料結構及其關係

在SQL解析中,最核心的結構是SELECT_LEX,其定義在sql/sql_lex.h中。下面僅列出與上述例子相關的部分。

SQL解析在美團點評中的應用

圖3 SQL解析樹結構

上面圖示中,列名username、ismale儲存在item_list中,表名儲存在table_list中,條件儲存在where中。其中以where條件中的Item層次結構最深,表達也較為複雜,如下圖所示。

SQL解析在美團點評中的應用

圖4 where條件

SQL解析的應用

為了更深入的瞭解SQL解析器,這裡給出2個應用SQL解析的例子。

無用條件去除

無用條件去除屬於優化器的邏輯優化範疇,可以僅僅根據SQL本身以及表結構即可完成,其優化的情況也是較多的,程式碼在sql/sql_optimizer.cc檔案中的remove_eq_conds函式。為了避免過於繁瑣的描述,以及大段程式碼的貼上,這裡通過圖來分析以下四種情況。

a)1=1 and (m > 3 and n > 4)

b)1=2 and (m > 3 and n > 4)

c)1=1 or (m > 3 and n > 4)

d)1=2 or (m > 3 and n > 4)

SQL解析在美團點評中的應用

圖5 無用條件去除a

SQL解析在美團點評中的應用

圖6 無用條件去除b

SQL解析在美團點評中的應用

圖7 無用條件去除c

SQL解析在美團點評中的應用

圖8 無用條件去除d

如果對其程式碼實現有興趣的同學,需要對MySQL中的一個重要資料結構Item類有所瞭解。因為其比較複雜,所以MySQL官方文件,專門介紹了Item類。阿里的MySQL小組,也有類似的文章。如需更詳細的瞭解,就需要去檢視原始碼中sql/item_*等檔案。

SQL特徵生成

為了確保資料庫,這一系統基礎元件穩定、高效執行,業界有很多輔助系統。比如慢查詢系統、中介軟體系統。這些系統採集、收到SQL之後,需要對SQL進行歸類,以便統計資訊或者應用相關策略。歸類時,通常需要獲取SQL特徵。比如SQL:

select username, ismale from userinfo where age > 20 and level > 5```
SQL特徵為: 
```sql
select username, ismale from userinfo where age > ? and level > ? 
複製程式碼

業界著名的慢查詢分析工具pt-query-digest,通過正規表示式實現這個功能但是這類處理辦法Bug較多。接下來就介紹如何使用SQL解析,完成SQL特徵的生成。

SQL特徵生成分兩部分組成。

a) 生成Token陣列

b) 根據Token陣列,生成SQL特徵

首先回顧在詞法解析章節,我們介紹了SQL中的關鍵字,並且每個關鍵字都有一個16位的整數對應,而非關鍵字統一用ident表示,其也對應了一個16位整數。如下表:

標識 select from where > ? and ident
整數 728 448 878 463 893 272 476

將一個SQL轉換成特徵的過程:

原SQL select username from userinfo where age > 20
SQL特徵 select ident:length:value from ident:length:value where ident:length:value > ?

在SQL解析過程中,可以很方便的完成Token陣列的生成。而一旦完成Token陣列的生成,就可以很簡單的完成SQL特徵的生成。SQL特徵被廣泛用於各個系統中,比如pt-query-digest需要根據特徵對SQL歸類,然而其基於正規表示式的實現有諸多bug。下面列舉幾個已知bug。

原始SQL pt-query-digest生成的特徵 SQL解析器生成的特徵
select * from email_template2 where id = 1 select * from mail_template? where id = ? select * from email_template2 where id = ?
REPLACE INTO a VALUES('INSERT INTO foo VALUES (1),(2)') replace into a values('insert into foo values(?+) replace into a values (?)

因此可以看出SQL解析的優勢是很明顯的。

學習建議

最近,在對SQL解析器和優化器探索的過程中,從一開始的茫然無措到有章可循,也總結了一些心得體會,在這裡跟大家分享一下。

首先,閱讀相關圖書書籍。圖書能給我們系統認識解析器和優化器的角度。但是針對MySQL的此類圖書市面上很少,目前中文作品可以看一看《資料庫查詢優化器的藝術:原理解析與SQL效能優化》。

其次,要閱讀原始碼,但是最好以某個版本為基礎,比如MySQL5.6.23,因為SQL解析、優化部分的程式碼在不斷變化。尤其是在跨越大的版本時,改動力度大。

再次,多使用GDB除錯,驗證自己的猜測,檢驗閱讀質量。

最後,需要寫相關程式碼驗證,只有寫出來了才能算真正的掌握。

作者簡介

廣友,美團點評到店綜合事業群MySQL DBA專家,2012年畢業於中國科學技術大學,2017年加入美團點評,長期致力於MySQL及周邊工具的研究。

金龍,2014年加入美團點評,主要從事相關的資料庫運維、高可用和相關的運維平臺建設。對運維高可用與架構相關感興趣的同學可以關注個人微信公眾號“自己的設計師”,定期推送運維相關原創內容。

邢帆,美團點評到店綜合事業群MySQL DBA,2017年研究生畢業後加入美團點評,目前已經對MySQL運維有一定經驗,並編寫了一些自動化指令碼。

美團點評DBA團隊招聘各類DBA人才,base北京上海均可。我們致力於為公司提供穩定、可靠、高效的線上儲存服務,打造業界領先的資料庫團隊。這裡有基於Redis Cluster構建的大規模分散式快取系統Squirrel,也有基於Tair進行大刀闊斧改進的分散式KV儲存系統Cellar,還有數千各類架構的MySQL例項,每天提供萬億級的OLTP訪問請求。真正的海量、分散式、高併發環境。歡迎各位朋友推薦或自薦至jinlong.cai#dianping.com。

如果對我們團隊感興趣,可以關注我們的專欄

SQL解析在美團點評中的應用

相關文章