在 BigQuery Java UDF 中对数组进行累积求和时出现问题

Question

我需要在 BigQuery Java UDF 中对一个数组进行累积求和。对于我在下面给出的示例，我期望结果为：[1,3,6,54,54]。但相反，我得到：

谁能告诉我我做错了什么？我是 Java 和 UDF 新手。谢谢！

#Standard SQL    
CREATE TEMP FUNCTION
      test(a ARRAY<int64>)
      RETURNS ARRAY<int64>
      LANGUAGE js AS """
    n=a.length;
    for (var i=1;i<= n-1;i++)
    {
    a[i]+=a[i-1];
    }
    return a;
    """;
    WITH
      test AS (
      SELECT
        [
        1,
        2,
        3,
        48,
        0] AS a)
    SELECT
      test(a)
    FROM
      test

Answer 1

INT64 is not officially supported with JavaScript UDFs此时因为没有 64 位整数类型。相反，您应该将 FLOAT64 用于数值，例如：

#StandardSQL    
CREATE TEMP FUNCTION
  test(a ARRAY<float64>)
  RETURNS ARRAY<float64>
  LANGUAGE js AS """
n=a.length;
for (var i=1;i<= n-1;i++)
{
a[i]+=a[i-1];
}
return a;
""";
WITH
  test AS (
  SELECT
    ARRAY<FLOAT64>[
    1,
    2,
    3,
    48,
    0] AS a)
SELECT
  test(a)
FROM
  test