apache-calcite - 如何添加一个参数是表名的udf

Question

我想在方解石中添加一个 udf。udf的参数是一个表名，它会返回一个varchar值。有这方面的样本吗？谢谢。

我的测试sql是：

SELECT MYTEST22(a), MYTEST(a), MYTEST1(a), COUNT(*)
FROM alisis.table1 
WHERE a=max_dt(alisis.table1) 
GROUP BY MYTEST22(a),MYTEST(a),MYTEST1(a) 

max_dt是我要添加的功能。

Answer 1

我不知道你想做什么，你能解释清楚吗？

任何函数都不能接受table作为参数，参数必须具有一定的类型，String、Int 或其他。所以constants和columns可以用作参数。如果你的意思是表名作为参数，sql应该如下：

SELECT MYTEST22(a), MYTEST(a), MYTEST1(a), COUNT(*)
FROM alisis.table1 
WHERE a=max_dt('alisis.table1') 
GROUP BY MYTEST22(a),MYTEST(a),MYTEST1(a)

据我所知，支持的所有功能 Calcite（版本为 1.11）都定义在SqlStdOperatorTable. 如果你想支持其他功能，你可以实现一个扩展的类，SqlStdOperatorTable例如：

public class TestOperatorTable extends SqlStdOperatorTable {
  private static volatile TestOperatorTable instance;
  public static TestOperatorTable instance() {
    if (instance == null) {
      synchronized(TestOperatorTable.class) {
        if (instance == null) {
          instance = new TestOperatorTable();
          // Use reflection to register the expressions stored in public fields.
          instance.init();
        }
      }
    }
    return instance;
  }

  /** Follow add your functions */
  public static SqlFunction LTRIM = new SqlTrimFunction("LTRIM");
}

SqlTrimFunction如下：

public class SqlTrimFunction extends SqlFunction {
  public SqlTrimFunction(String funcName) {
    super(
      funcName,
      SqlKind.OTHER_FUNCTION,
      ReturnTypes.VARCHAR_2000,  // return type is String
      InferTypes.FIRST_KNOWN,
      OperandTypes.STRING,       // parameter type is String
      SqlFunctionCategory.STRING
    ); 
  }
}

例如，当您创建一个 SqlValidator 时SqlValidatorImpl，您应该使用TestOperatorTable它作为它的第一个参数。