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,为用户带来了更高效、愉悦的数据处理体验。