こちらはPostgreSQL Advent Calendar 2022 カレンダー2枚目・15日目の投稿となります。
前回は、@hmatsu47さんのSupabase で TCE（透過的列暗号化）を軽く試してみたでした。

はじめに

こんにちは、フューチャーでアルバイトをしている齋藤です。以前は同社のインターンでSQLフォーマッタを作成していました(記事)。現在はインターン中に作成していたSQLフォーマッタをPostgreSQLの構文に対応させる作業に取り組んでいます。

このフォーマッタではSQLパーサにtree-sitter-sqlを利用していますが、対応していない構文がいくつか存在します。本記事では、未対応の構文であるBETWEEN述語を例に、tree-sitterの構文拡張の手順を紹介します。開発中のSQLフォーマッタはOSS公開予定ですので、ぜひ仲間を増やしたいという思いから記事にしました。

また、現在作成中のフォーマッタのVSCode拡張機能化にも取り組んでいます。ぜひそちらも併せてご覧ください！

VSCode拡張機能化に関する記事:

アウトライン

本記事のアウトラインは以下の通りです。

tree-sitter、tree-sitter-sqlについて
tree-sitterの構文拡張用の環境構築
構文木を出力するプログラムの実装
構文についての説明
BETWEEN述語の規則を追加

tree-sitter

tree-sitterは文法からパーサ(構文解析器)を自動生成するパーサジェネレータツールであり、生成されたパーサで構文解析を行うライブラリでもあります。特徴として、一般的なパーサライブラリでは抽象構文木(AST)を構築するのに対し、tree-sitterで生成されたパーサは具象構文木(CST)を構築するという点があげられます。CSTについてはインターンの記事で取り上げています。

構築されるCSTにはコメントトークンも含まれてるため、シンタックスハイライトに用いられているようです。
参考:

tree-sitter-sql

tree-sitter-sqlはtree-sitter用に書かれたSQLの文法とその文法によって生成されたパーサライブラリです。SQLの中でも、PostgreSQLにフォーカスしていたようです。インターンで作成したフォーマッタは、このライブラリによる構文解析結果をもちいてSQLのフォーマットを行っています。

しかし、BETWEEN述語やUNION、INTERSECTなどの結合演算など、基本的な構文であるにもかかわらず、対応していない構文が存在します。本記事では、その中でもBETWEEN述語に対応させるための構文拡張を行います。

環境構築

まず、tree-sitterの構文拡張のために行った環境構築について説明します。

tree-sitter-cliのインストール

tree-sitterでパーサを生成するために、tree-sitter-cliをインストールします(参考Tree-sitter | Creating Parser)。また、tree-sitterによるパーサを開発するためには、Node.jsとCコンパイラが必要です。今回使用したバージョンは以下の通りです。

tools	バージョン
node	16.17.1
gcc	12.2.0
tree-sitter	0.20.7

tree-sitter-sqlのインストール

tree-sitter-sqlをcloneします。tree-sitter用のSQL構文はいくつかありますが、今回は最もスター数が多いものを選択しました。

git clone https://github.com/m-novikov/tree-sitter-sql.git

git cloneを行うと、以下のようなエラーが発生する場合があります。

error: unable to create file [filepath]: Filename too long

これはファイル名が長すぎることが問題であるようなので、以下の設定を行うことで解決します。

gitの設定

git config --global core.longpaths true

git clone したtree-sitter-sqlのルートディレクトリで、tree-sitter test コマンドでテストが動作したら環境構築終了です。

cd ./tree-sitter-sql
tree-sitter test

構文解析例

実際にパースしてみましょう。以下のファイルを用意します。

exapmles/simple.sql

SELECT
    ID
FROM
    STUDENT

tree-sitter parseコマンドで、ソースファイルをパースできます。

$ tree-sitter parse ./exapmles/simple.sql
(source_file [0, 0] - [3, 11]
  (select_statement [0, 0] - [3, 11]
    (select_clause [0, 0] - [1, 6]
      (select_clause_body [1, 4] - [1, 6]
        (identifier [1, 4] - [1, 6])))
    (from_clause [2, 0] - [3, 11]
      (identifier [3, 4] - [3, 11]))))

CSTの出力について

上述したtree-sitter parseにより出力される結果では、ノードのラベルのみ表示されており、識別子やキーワードなどが表示されません。そこで、パース結果からCSTを出力する処理を自作しました。

言語にはRustを使用します。

準備

tree-sitter-sqlの結果を利用してCSTを出力するためのプロジェクトを作成します。

cargo new print-cst

Cargo.tomlに次の依存関係を追加します。

print-cst/Cargo.toml

[dependencies]
tree-sitter = "~0.20.3"
tree-sitter-sql = {path = "{tree-sitter-sqlのパス}"}

また、Github上のtree-sitter-sqlが使用しているtree-sitterのバージョンが古い(2022年11月22日現在)ため、tree-sitter-cliとtree-sitterのバージョン不整合が生じる可能性があります。バージョン不整合が生じるとき、後述するプログラムを実行すると以下のような実行時エラーが発生します。

thread 'main' panicked at 'called `Result::unwrap()` on an `Err` value: LanguageError { version: 14 }', src\lib.rs:16:35
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

この場合、tree-sitter-sqlのCargo.tomlも修正する必要があります。

tree-sitter-sql/Cargo.toml

[dependencies]
tree-sitter = "~0.20.3"

実装

main.rsに次のように実装しました。

main.rs

use std::fs::read_to_string;

use tree_sitter::TreeCursor;

fn main() {
    let language = tree_sitter_sql::language();
    let mut parser = tree_sitter::Parser::new();
    parser.set_language(language).unwrap();
    let src_file = std::env::args().nth(1).expect("arguments error");
    let src = read_to_string(&src_file).unwrap();

    let tree = parser.parse(&src, None).unwrap();
    if tree.root_node().has_error() {
        println!("error");
    } else {
        let mut cursor = tree.walk();

        visit(&mut cursor, 0, &src);
    }
}

const UNIT: usize = 2;

fn visit(cursor: &mut TreeCursor, depth: usize, src: &str) {
    // インデント
    (0..(depth * UNIT)).for_each(|_| print!(" "));

    print!("{}", cursor.node().kind());

    // 子供がいないかつ、キーワードでない場合、対応する文字列を表示
    if cursor.node().child_count() == 0 && cursor.node().kind().chars().any(|c| c.is_lowercase()) {
        print!(" \"{}\"", cursor.node().utf8_text(src.as_bytes()).unwrap());
    }
    println!(
        " [{}-{}]",
        cursor.node().start_position(),
        cursor.node().end_position()
    );

    // 子供を走査
    if cursor.goto_first_child() {
        visit(cursor, depth + 1, src);
        while cursor.goto_next_sibling() {
            visit(cursor, depth + 1, src);
        }
        cursor.goto_parent();
    }
}

実行例

作成したプログラムを用いて、実際にCSTを表示してみましょう。

exapmles/simple.sql

SELECT
    ID
FROM
    STUDENT

$ cargo run ./examples/simple.sql
source_file [(0, 0)-(1, 12)]
  select_statement [(0, 0)-(0, 9)]
    select_clause [(0, 0)-(0, 6)]
      SELECT [(0, 0)-(0, 6)]
      select_clause_body [(0, 7)-(0, 9)]
        identifier "ID" [(0, 7)-(0, 9)]
    from_clause [(1, 0)-(1, 4)]
      FROM [(1, 0)-(1, 4)]
      identifier "STUDENT" [(1, 5)-(1, 12)]

ノードに対応する文字列とキーワードを出力できました。

構文例

次に、tree-sitter用の構文について簡単に紹介します。

tree-sitter では文法を grammar.js に記述します。clone した tree-sitter-sql のルートディレクトリにある grammar.jsを編集していきます。ここではDSL(ドメイン固有言語)について細かくは説明しないので、詳しく知りたい方はtree-sitterのドキュメントを参照してください。

規則

例えば、tree-sitter-sql で WHERE句は以下のように記述されています(where_clauseの定義)。

WHERE句の規則

where_clause: $ => seq(kw("WHERE"), $._expression)

seqはtree-sitterの文法のDSLの1つで、複数の規則を連結できます。上の例では、kw("WHERE")のあとに$._expressionが現れることを示しています。

kw関数はtree-sitter-sqlのgrammar.jsで定義されている関数で、キーワード(keyword)が大文字か小文字であるかを考慮しなくするなどの処理を行います。パース時には、whereやWHEREというキーワードとマッチします(kw関数の定義)。

アンダースコアから始まる規則

規則名の先頭の文字をアンダースコアから始めることで、生成されるCSTにノードとして出現させないように設定できます(ドキュメント)。例えば、算術演算や識別子、リテラルなどの式は_expressionという名前で以下のように定義されています。

式に対応する規則

_expression: $ =>
     choice(
       $.string,
       $.TRUE,
       $.FALSE,
       $.NULL,
       $._identifier,
       $.number,
       $.unary_expression,
       // 一部省略
     ),

choiceはtree-sitterのDSLで、引数のうちいずれか1つとマッチすることを意味しています。つまり、この規則は、文字列やTRUE、FALSEなど各式に対応した規則を呼び出し、いずれか1つとマッチすることになります。つまり、ソースファイル中に式が現れるたびに_expressionが呼び出されています。これがCST上に現れると、例えば1+2-3という式のパース結果が以下のようになってしまいます。

(_expression
  (binary_expression
    (_expression
      (binary_expression
        (_expression
          (number "1"))
        ("+")
        (_expression
          (number "2")))
    ("-")
    (_expression
      (number "3"))))

アンダースコアから始めることで、CST上に現れないように設定でき、以下のようにシンプルな木にできます。

(binary_expression
  (binary_expression
    (number "1")
    ("+")
    (number "2"))
  ("-")
  (number "3))

優先度、結合性

ここで詳細は述べませんが、tree-sitterは明示しない場合、曖昧な文法を扱うことができません(参考)。

例えば、以下のような論理式を考えてみます。

NOT X AND Y OR Z

この式はどのように解釈されるでしょうか？NOT (X AND (Y OR Z))や(NOT X) AND (Y OR Z)、((NOT X) AND Y) OR Zなど、複数通りに解釈できてしまうと思います。このように、複数通りの解釈ができてしまうような文法を曖昧な文法といい、そのままではパースできません。

これは、優先度・結合性を文法に記述することで対処できます。tree-sitter-sqlでは優先度をJavaScriptの定数として以下のように定義しています。

const PREC = {
  primary: 8,
  unary: 7,             // 単項演算子
  exp: 6,               // 累乗
  multiplicative: 5,    // 乗除算
  additive: 4,          // 加減算
  comparative: 3,       // 比較演算子
  and: 2,               // AND
  or: 1,                // OR
};

これを用いて、論理式に優先度・結合性を加えて記述した規則は次のようになります。

boolean_expression: $ =>
  choice(
    prec.left(PREC.unary, seq(kw("NOT"), $._expression)), // 優先度7
    prec.left(PREC.and, seq($._expression, kw("AND"), $._expression)),  // 優先度2
    prec.left(PREC.or, seq($._expression, kw("OR"), $._expression)), // 優先度1
  ),

優先度は、NOT > AND > ORになっています。優先度が高いものほど優先して結合されるため、上述の論理式をtree-sitter-sqlでパースすると、((NOT X) AND Y) OR Zと解釈されます。なお、prec.leftは左結合であることを意味しています。

extras

ファイルのどこに現れてもよい規則をextrasで記述できます。
これを使って、コメントや空白、改行を簡単に記述できます(コメント、空白の定義)が、CST上では直感的でない場所位置に現れる場合もあります(インターンの記事後編参照)。

BETWEEN述語への対応

現状のtree-sitter-sqlを使用して、BETWEENを含むSQLをパースできるか確認してみましょう。以下のようなファイルを用意します。

examples/between.sql

SELECT
    ID
FROM
    STUDENT
WHERE
    GRADE   BETWEEN 80  AND 100
AND ID      BETWEEN 0   AND 100

$ tree-sitter parse .\examples\between.sql
(source_file [0, 0] - [7, 0]
  (select_statement [0, 0] - [6, 31]
    (select_clause [0, 0] - [1, 6]
      (select_clause_body [1, 4] - [1, 6]
        (identifier [1, 4] - [1, 6])))
    (from_clause [2, 0] - [3, 11]
      (identifier [3, 4] - [3, 11]))
    (where_clause [4, 0] - [6, 31]
      (boolean_expression [5, 4] - [6, 31]
        (boolean_expression [5, 4] - [6, 6]
          (boolean_expression [5, 4] - [5, 31]
            (identifier [5, 4] - [5, 9])
            (ERROR [5, 12] - [5, 22])
            (number [5, 28] - [5, 31]))
          (identifier [6, 4] - [6, 6]))
        (ERROR [6, 12] - [6, 21])
        (number [6, 28] - [6, 31])))))
.\examples\between.sql  0 ms    (ERROR [5, 12] - [5, 22])

構文エラーが発生し、WHERE句内のBETWEEN述語には対応していないことがわかります。grammar.jsを見てみるとBETWEENというキーワードはWINDOW関数のFRAME句にしか想定していないため、BETWEENがERRORノードと扱われているようです。

規則の追加

BETWEEN述語に対応する規則がそもそも存在していないことがわかったため、文法を拡張することで対応していきます。

BETWEEN述語は次のような構文になっています。PostgreSQLのドキュメントでは構文について詳しく書かれていなかったので、Oracle SQLのドキュメントを参考にしました。

BETWEEN述語の構文

(expression) (NOT)? BETWEEN (expression) AND (expression)

なお、(NOT)? は正規表現で使われる ? と同じ意味で、 NOT が0回または1回現れることを表現しています。tree-sitterの構文では、optionalというDSLで表現されます。

率直にDSLに直すと、次のような規則が書けます。

率直に書いたBETWEENの規則

between_and_expression: $ =>
  seq($._expression, optional(kw("NOT")), kw("BETWEEN"),
      $._expression, kw("AND"), $._expression)

この規則をSQLの式に対応する規則_expressionに追加します。

_expressionへの追加

    _expression: $ =>
      choice(
        $.string,
        $.TRUE,
        $.FALSE,
        // 省略
+       $.between_and_expression,
      ),

これでBETWEEN述語の規則を追加できました。拡張した文法をもとにパーサを生成してみましょう。以下のコマンドを実行します。

$ tree-sitter generate
Unresolved conflict for symbol sequence:

  'grant_statement_token4'  _expression  'create_trigger_statement_token1'  _expression  •  'cte_token2'  …

Possible interpretations:

  1:  'grant_statement_token4'  (boolean_expression  _expression  'create_trigger_statement_token1'  _expression)  •  'cte_token2'  …
                                                            (precedence: 1, associativity: Left)
  2:  'grant_statement_token4'  _expression  'create_trigger_statement_token1'  (between_and_expression  _expression  •  'cte_token2'  'frame_clause_token1'  _expression  'frame_clause_token2'  _expression)
  3:  'grant_statement_token4'  _expression  'create_trigger_statement_token1'  (in_expression  _expression  •  'cte_token2'  'create_function_parameter_token1'  tuple)                                        (precedence: 3, associativity: Left)

Possible resolutions:

  1:  Specify a higher precedence in `in_expression` and `between_and_expression` than in the other rules.
  2:  Specify a higher precedence in `boolean_expression` than in the other rules.
  3:  Add a conflict for these rules: `in_expression`, `between_and_expression`, `boolean_expression`

エラーが発生してしまい、パーサが生成できませんでした。これは、上述した規則では優先度を記述していないため、文法が曖昧になってしまっていることが原因です。例えば、X BETWEEN Y AND Z AND WのANDがBETWEEN述語のものなのか、論理式のものなのかをパーサが自動で判別できません。つまり、X BETWEEN (Y AND Z) AND Wや(X BETWEEN Y AND Z) AND Wなど、複数の解釈ができてしまいます。

そこで、優先度と結合性を追加します。

優先度と結合性を追加したBETWEENの規則

between_and_expression: $ =>
  prec.left(
    PREC.comparative,
    seq($._expression, optional(kw("NOT")), kw("BETWEEN"),
        $._expression, kw("AND"), $._expression)
  ),

prec.leftは左結合であることを示し、PREC.comparativeで比較演算子と同じ優先度であることを指定しています。比較演算子はANDよりも高い優先度であるため、X BETWEEN Y AND Z AND Wは(X BETWEEN Y AND Z) AND Wと解釈されます。

動作確認

次のファイルをパースしてみましょう。

examples/between.sql

SELECT
    ID
FROM
    STUDENT
WHERE
    GRADE   BETWEEN 80  AND 100
AND ID      BETWEEN 0   AND 100

以下のコマンドでパーサを生成します。

tree-sitter generate

先ほど作成した print-cstを用いて、パース結果を出力します。

$ cd [print-cstのパス]
$ cargo run ./examples/between.sql
source_file [(0, 0)-(6, 31)]
  select_statement [(0, 0)-(1, 6)]
    select_clause [(0, 0)-(0, 6)]
      SELECT [(0, 0)-(0, 6)]
      select_clause_body [(1, 4)-(1, 6)]
        identifier "ID" [(1, 4)-(1, 6)]
    from_clause [(2, 0)-(2, 4)]
      FROM [(2, 0)-(2, 4)]
      identifier "STUDENT" [(3, 4)-(3, 11)]
    where_clause [(4, 0)-(4, 5)]
      WHERE [(4, 0)-(4, 5)]
      boolean_expression [(5, 4)-(5, 31)]
        between_and_expression [(5, 4)-(5, 9)]
          identifier "GRADE" [(5, 4)-(5, 9)]
          BETWEEN [(5, 12)-(5, 19)]
          number "80" [(5, 20)-(5, 22)]
          AND [(5, 24)-(5, 27)]
          number "100" [(5, 28)-(5, 31)]
        AND [(6, 0)-(6, 3)]
        between_and_expression [(6, 4)-(6, 6)]
          identifier "ID" [(6, 4)-(6, 6)]
          BETWEEN [(6, 12)-(6, 19)]
          number "0" [(6, 20)-(6, 21)]
          AND [(6, 24)-(6, 27)]
          number "100" [(6, 28)-(6, 31)]

これでBETWEENを含むSQLがパースできるようになりました！

テストの追加

最後に、今回追加したBETWEEN述語の拡張をtree-sitter test(Tree-sitter｜Creating Parsers)でテストできるようにしましょう。

test/corpus/between.txtを作成して、以下のように記述します。

test/corpus/between.txt

=======================================
BETWEEN predicates
=======================================

SELECT
    ID
FROM
    STUDENT
WHERE
    GRADE   BETWEEN 80  AND 100
AND ID      BETWEEN 0   AND 100

---------------------------------------

(source_file
  (select_statement
    (select_clause
      (select_clause_body
        (identifier)))
    (from_clause
      (identifier))
    (where_clause
      (boolean_expression
        (between_and_expression
          (identifier)
          (number)
          (number))
        (between_and_expression
          (identifier)
          (number)
          (number))))))

=で囲まれた行にテスト名を書きます
次に、入力として与えるソースコードを記述し、下に---を記述します
最後に期待する結果をS式で記述します

tree-sitter testでテストを行います。-fフラグを加えることで、特定のテストのみを実行できます。

$ tree-sitter test -f 'BETWEEN predicates'
  between:
    ✓ BETWEEN predicates
  create:
  delete:
  insert:
  select:
  statements:
  update:
syntax highlighting:
  ✓ builtin.sql (49 assertions)
  ✓ function.sql (16 assertions)
  ✓ insert.sql (6 assertions)
  ✓ keywords.sql (9 assertions)
  ✓ punctuation.sql (2 assertions)
  ✓ select.sql (43 assertions)
  ✓ statements.sql (25 assertions)
  ✓ table.sql (33 assertions)
  ✓ type.sql (5 assertions)
  ✓ update.sql (10 assertions)

まとめ

本記事では、tree-sitter-sqlでBETWEEN述語を扱えるように構文拡張を行いました。tree-sitter用のSQL構文はまだまだ未完成なので、皆さんも一緒によりよいパーサを作ってみませんか？

tree-sitter文法入門