Databend 集成 PRQL：现代数据处理的一小步

PRQL，读作 “Prequel”，是一种与 SQL 并肩的查询语言，它的独到之处在于采用了管道式的语法，在查询关系数据库时显得更加直观和高效。

Databend 拥抱 PRQL

在 v1.2.380-nightly 版本中，得益于社区贡献者 @ncuwaln 提交的重要 PR，Databend 成功地引入了对 PRQL 语言的支持。这一新特性进一步增强了 Databend 查询的灵活性和用户友好性。

用户仅需进行简单的配置更改，即可启用 PRQL 方言：

set sql_dialect = 'prql'; 

集成细节

Databend 和 PRQL 都采用了 Rust 语言进行开发，这使得集成 PRQL 编译器 prqlc 成为可能。

prqlc = "0.11.3" 

该 PR 以不到一百行代码实现了对 PRQL 的支持，核心是添加了一个新的 SQL 方言选项 prql。

启用此选项后，Databend 将调用 prqlc 解析 PRQL 语句，并将其转换为 SQL 语句，随后由 Databend 的 SQL 解析器进一步处理。

 let final_sql: String = match sql_dialect == Dialect::PRQL { true => { let options = prqlc::Options::default(); match prqlc::compile(sql, &options) { Ok(res) => { prql_converted = true; res } Err(e) => { warn!( "Try convert prql to sql failed, still use raw sql to parse. error: {}", e.to_string() ); sql.to_string() } } } false => sql.to_string(), }; .... 

值得一提的是，prqlc 使用了 chumsky 库作为其解析器，而 Databend 则采用了 nom-rule 。尽管两者在技术实现上有所不同，但都提供了用户友好的 errror recovery 信息。

错误提示的对比展示

以下是 PRQL 解析器和 Databend 解析器的错误提示示例：

PRQL 解析器 错误提示：

from tracks select {album_id, name, unit_price} sort {-unit_price, name} grouq album_id ( aggregate { track_count = count name, album_price = sum unit_price } ) Error: ╭─[:4:1] │ 4 │ grouq album_id ( │ ──┬── │ ╰──── Unknown name `grouq` ───╯ 

Databend 解析器 错误提示：

🐳 :) SELECT album_id, COUNT(*) AS track_count, COALESCE(SUM(unit_price), 0) AS album_price FROM tracks GROUP BX album_id; error: APIError: ResponseError with 1005: error: --> SQL:7:7 | 1 | SELECT | ------ while parsing `SELECT ...` . 4 | COALESCE(SUM(unit_price), 0) AS album_price 5 | FROM 6 | tracks 7 | GROUP BX | ^^ unexpected `BX`, expecting `BY` 

如何使用 PRQL

用户只需进行简单的配置，即可切换到 PRQL 方言，并开始享受 PRQL 直观、管道式的查询体验。以下是一个简单的查询示例：

🐳 :) set sql_dialect = 'prql'; 🐳 :) from lineitem ## run tpch q1 filter `l_shipdate` <= '1998-09-02' group {l_returnflag, l_linestatus} ( aggregate { sum_qty = sum l_quantity, sum_base_price = sum l_extendedprice, sum_disc_price = sum l_extendedprice * (1 - l_discount), sum_charge = sum l_extendedprice * (1 - l_discount) * (1 + l_tax), avg_qty = average l_quantity, avg_price = average l_extendedprice, avg_disc = average l_discount, count_order = count 0 , } ); 

通过这个例子，我们可以看到 PRQL 强大的功能和优雅的语法设计，它让复杂的数据查询变得简单而直观。当然，用户也可以随时根据需要切换回标准的 SQL 方言或其他支持的方言：

🐳 :) set sql_dialect = 'postgresql'; 

通过这次集成，Databend 拥抱了前沿的查询语言PRQL，为用户带来了更高效、愉悦的数据处理体验。