google-bigquery - 大查询 : does it support UDAF?

Question

我猜 BigQuery 不支持 UDAF，因为我只能找到关于 UDF 的信息。 BigQuery 是否支持 UDAF？如果没有，有没有办法以某种方式在聚合结果上运行 UDF？也许通过使用 ARRAY_AGG？

Answer 1

是的，您可以使用 ARRAY_AGG 为 UDAF 建模。例如，您可以这样做:

CREATE TEMP FUNCTION AggX(arr ARRAY<INT64>) AS (
  (SELECT SUM(x * x) FROM UNNEST(arr) AS x)
);

SELECT id, AggX(ARRAY_AGG(x)) AS result
FROM dataset.table
GROUP BY id

根据您要实现的逻辑类型，您甚至可以让函数采用 ARRAY<STRUCT<...>>。使用您希望函数对其进行操作的任何列，并使用 ARRAY_AGG(STRUCT(...))构建输入。