Presto 自定义 UDF

Question

我创建了一个已注册的自定义 udf，但是当我尝试 select custom_udf(10) 时，出现以下错误： Exact implementation of BasicPlatform do not match expected java types 这是我的 udf，我似乎无法弄清楚它有什么问题： public class 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);
}

}

有没有明显的错误？我希望它 return 一个给定 int 作为参数的字符串，我认为 java 和 sql 类型匹配 (Integer -> Integer), (Slice -> varchar).

谢谢

Answer 1

此问题也在 presto-users 上被问及回答：

您必须使用 @SqlNullable @SqlType(StandardTypes.INTEGER) Long id（因为 SQL integer 在 Java 中得到 Long 的支持）。

Presto 自定义 UDF

Presto Custom UDF

java

presto