java - Presto 定制 UDF

Question

我创建了一个已注册的自定义 udf，但是当我尝试选择 custom_udf(10) 时，出现以下错误:BasicPlatform 的精确实现与预期的 java 类型不匹配这是我的 udf，我似乎无法弄清楚它出了什么问题:公共(public)类 ScalarUdfs {

private ScalarUdfs() {};

@ScalarFunction("basic_platform")
@SqlType(StandardTypes.VARCHAR)
public static Slice BasicPlatform(@SqlNullable @SqlType(StandardTypes.INTEGER) Integer id) {
    final Slice IOS = Slices.utf8Slice("iOS");
    final Slice ANDROID = Slices.utf8Slice("Android");
    final Slice WEB = Slices.utf8Slice("Web");
    final Slice OTHER = Slices.utf8Slice("Other");
    final Map<Integer, Slice> PLATFORM_MAP = new HashMap<Integer, Slice>() {{
        put(20, IOS);
        put(42, ANDROID);
        put(100, WEB);
    }};

    if (id == null || !PLATFORM_MAP.containsKey(id)) {
        return OTHER;
    }
    return PLATFORM_MAP.get(id);
}

}

有什么明显错误的地方吗？我希望它返回一个给定 int 作为参数的字符串，并且我认为 java 和 sql 类型匹配 (Integer -> Integer)、(Slice -> varchar)。

谢谢

Answer 1

这个问题也在 presto-users 上被提出和回答。 :

您必须使用@SqlNullable @SqlType(StandardTypes.INTEGER) Long id(因为SQL integer由Java中的Long支持) .