萬盛學電腦網

 萬盛學電腦網 >> 數據庫 >> mssql數據庫 >> mixer:sql詞法分析器設計

mixer:sql詞法分析器設計

   介紹

  mixer希望在proxy這層就提供自定義路由,sql黑名單,防止sql注入攻擊等功能,而這些的基石就在於將用戶發上來的sql語句進行解析。也就是我最頭大的詞法分析和語法分析。

  到現在為止,我只是實現了一個比較簡單的詞法分析器,用以將sql語句分解成多個token。而對於從token在進行語法分析,構建sql的AST,我現在還真沒啥經驗(編譯原理太差了),急需牛人幫忙。

  所以,這裡只是簡單介紹一下mixer的詞法分析。

  tokenize

  在很多地方,我們都需要進行詞法分析,通常會有幾種方式:

  使用一個強大的工具,譬如lex,mysql-proxy就用的這種方式

  使用正則表達式

  state machine

  對於使用工具,我覺得有一個不怎麼好的地方在於學習成本,譬如我用lex的時候就需要學習它的語法,同時通過工具生成的代碼可讀性都不怎麼好,代碼量大,更嚴重的是可能會比較慢。所以mysql自身也是自己實現一個詞法分析模塊。

  而對於正則表達式,性能問題可能是一個很需要考慮的,而且復雜度並不比使用類似lex這樣的工具低。

  狀態機可能是我覺得自己動手實現詞法解析一個很好的方式,對於sql的詞法解析,我覺得使用state machine的方式來自己寫一個難度並不大,所以mixer自己實現了一個。

  state machine

  通常,一個狀態機的實現采用的是state + action + switch的做法,可能如下:

?

1 2 3 4 5 6 7 8 switch state {     case state1:         state = action1()     case state2:         state = action2()     case state3:         state = action3() }

  對於一個state,我們通過switch知道它將會由哪一個action進行處理,而對於每一個action,我們則知道執行完成之後下一個state是什麼。

  對於上面的實現,如果state過多,可能會導致太多的case語句,我們可以通過state function進行簡化。

  一個state function就是執行當前的state action,並且直接返回下一個state function。

  我們可以這樣做:

?

1 2 3 4 5 type stateFn func(*Lexer) stateFn   for state := startState; state != nil {     state = state(lexer) }

  所以我們需要實現的就是每一個state function以及對應的它的下一個需要執行的state function。

  mixer lexer

  mixer的詞法分析實現主要參考這個。主要實現在parser模塊。

  對於一個lexer,需要提供的是NextToken的功能,供外部獲取下一個token,從而進行後續的操作(譬如語法分析)。

  lexer的next token如下:

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 func (l *Lexer) NextToken() (Token, error) {     for {         select {             case t := <-l.tokens:                 return t, nil             default:                 if l.state == nil {                     return Token{TK_EOF, ""}, l.err                 }                 l.state = l.state(l)                 if l.err != nil {                     return Token{TK_UNKNOWN, ""}, l.err                 }         }     } }

  tokens是一個channel,每次state解析的token都會emit到這個channel上面,供NextToken獲取,如果channel為空了,則再次調用state function。

  可以看到,用go實現一個詞法解析是很容易的事情,剩下的就是寫相應的state function用來解析sql。

  todo

  mixer的詞法分析還有很多不完善的地方,譬如對於科學計數法數值的解析就不完善,後續准備參考mysql官方的詞法分析模塊在好好完善一下。

copyright © 萬盛學電腦網 all rights reserved