如何为 Presto 编写自定义窗口函数？

如何解决如何为 Presto 编写自定义窗口函数？

我想创建一个自定义 Window 函数，该函数处理已排序的双精度流并为每个分区生成单个输出双精度。必须对值进行排序。 Window 函数可以一次处理单行（无需向后/向前看），只要它可以保持每个分区的内部状态即可。

签名看起来像这样：

SELECT my_windows_func() OVER (PARTITION BY my_key ORDER BY my_val ASC) AS my_stuff

现在，我想出了如何创建 AggreagtionFunctions、ScalarFunctions，但使用 WindowFunctions 我不知道从哪里开始，也找不到任何在线文档。

我应该为我的用例实现哪个接口？我可以强制用户在调用时添加 ORDER BY 子句吗？我可以参考任何示例代码吗？

解决方法

函数实现需要从io.prestosql.spi.Plugin#getFunctions返回，并用@WindowFunctionSignature注解。 io.prestosql.spi.function.ValueWindowFunction 是实现窗口函数的有用基础。有关示例，请参阅 https://github.com/trinodb/trino/blob/4fba34f3d94291d5411a5c24bd47fa0213898e51/presto-main/src/main/java/io/prestosql/operator/window/LagFunction.java。