如何将用户定义的变量添加到 Apache Calcite 架构？

如何解决如何将用户定义的变量添加到 Apache Calcite 架构？

我在使用 sqlValidator 来验证用户输入的 sql 包含一些表引用（这还不错）和某种类型的变量的预定义列表时遇到了一些麻烦。

例如，用户可能会输入一些依赖于变量 an_id_cutoff 的 sql，该变量通过其他来源填充​​。我不想将其重写为他们使用 sqlVisitor 提供的值，而是想检查它是否正确，就像在 MysqL 或类似的东西中定义了该变量一样。

目前，我从我编写的扩展 AbstractSchema 的 SimpleSchema 创建了一个 CalciteSchema。它只有一个 Map。在这个 Map 中，我存储了我允许引用的表，这很好用。但我不确定如何表示变量。

例如;

• 假设变量 an_id_cutoff 和 some_string 是允许的，并且将填充 1 和 "abc" 等值
• 假设存在一个名为 a_table 的表，其中包含一个整数列 id 和一个 varchar 列 string_field

SELECT * FROM a_table T WHERE T.id = an_id_cutoff AND T.string_field = some_string

有效。

SELECT * FROM a_table T WHERE T.id = some_unexpected_variable

无效。

我猜这些就像 MysqL 变量一样，所以我认为必须有某种方式在 Schema 中表示它们。但我真的不明白怎么做。

SET @an_id_cutoff=1;
SELECT * FROM a_table T WHERE T.id = @an_id_cutoff