BigQuery UDF是否支持可变参数？如何实现多值空值检测？

一、创建支持任意数量参数的any_are_null UDF #

BigQuery完全支持实现这类可变参数UDF，通过ANY TYPE...语法定义可变参数列表，内部将传入的多个参数自动转为数组处理即可。

以下是具体的SQL UDF定义：

CREATE OR REPLACE FUNCTION mydataset.variant_test.any_are_null(args ANY TYPE...) AS (
  EXISTS(
    SELECT 1 
    FROM UNNEST(args) AS val 
    WHERE val IS NULL
  )
);

调用方式：直接传入任意数量的列或值，无需预先指定参数个数：

SELECT 
  mydataset.variant_test.any_are_null(col1, col2, col3, col4),
  mydataset.variant_test.any_are_null(col1, col2);

二、参数解包的实现 #

BigQuery没有类似其他语言的...或*显式解包语法，但可以通过两种方式实现等效效果：

1. 兼容数组与多参数的UDF逻辑
   如果你的参数已经是一个数组（比如预先定义的数组变量或查询生成的数组），可以调整UDF逻辑，同时支持多参数自动打包和显式数组输入：

CREATE OR REPLACE FUNCTION mydataset.variant_test.any_are_null(args ANY TYPE) AS (
  EXISTS(
    SELECT 1 
    FROM UNNEST(
      -- 兼容单个值、多参数打包数组、显式传入数组三种场景
      IF(ARRAY_LENGTH(args) IS NOT NULL, args, [args])
    ) AS val 
    WHERE val IS NULL
  )
);

调整后即可同时支持两种调用方式：

-- 方式1：传入多个独立参数
SELECT mydataset.variant_test.any_are_null(col1, col2, col3);

-- 方式2：传入已有的数组
DECLARE target_array ARRAY<INT64> DEFAULT [1, NULL, 3];
SELECT mydataset.variant_test.any_are_null(target_array);

2. 通过UNNEST解包数组元素
   如果需要将数组元素作为独立参数传入原始的ANY TYPE...版本UDF，可以结合UNNEST和子查询实现：

DECLARE target_array ARRAY<INT64> DEFAULT [1, NULL, 3];
SELECT ANY_VALUE(mydataset.variant_test.any_are_null(UNNEST(target_array)))
FROM UNNEST([1]); -- 用单元素数组触发一行查询

内容的提问来源于stack exchange，提问作者David542