Snowflake存储过程：如何根据布尔变量动态添加SELECT列

你遇到的核心问题是静态SQL无法根据变量动态调整返回列数，Snowflake的SQL存储过程要实现这种逻辑，必须通过动态SQL构造查询语句后执行。

以下是修正后的完整代码：

CREATE OR REPLACE PROCEDURE XXX (CUSTOMER_ACCOUNT_ID VARCHAR, GROUP_BY ARRAY) 
RETURNS TABLE (
  SUM_XXX NUMBER,
  SUM_YYY NUMBER,
  SUM_ZZZ NUMBER -- 固定返回结构，FALSE时该列返回NULL
)
LANGUAGE SQL 
COMMENT = 'Test'
AS 
$$
DECLARE 
    groupByConnector BOOLEAN;
    dynamic_sql STRING;
    res RESULTSET;
BEGIN
    groupByConnector := ARRAY_CONTAINS('connector'::VARIANT, GROUP_BY);

    -- 动态拼接SQL语句
    dynamic_sql := 'SELECT 
        SUM(xxx) AS SUM_XXX,
        SUM(yyy) AS SUM_YYY' || 
        IFF(groupByConnector, ', SUM(zzz) AS SUM_ZZZ', '') || '
    FROM RANDOM_TABLE';

    -- 执行动态SQL并获取结果集
    res := EXECUTE IMMEDIATE :dynamic_sql;

    RETURN TABLE(res);
END
$$;

关键说明 #

1. 动态SQL拼接：用Snowflake的IFF三元运算符，根据groupByConnector的值决定是否追加SUM(zzz)列的SQL片段。TRUE时追加, SUM(zzz) AS SUM_ZZZ，FALSE时追加空字符串。
2. 执行动态SQL：通过EXECUTE IMMEDIATE执行构造好的SQL字符串，结果存入RESULTSET类型变量。
3. 固定返回结构：存储过程的RETURNS TABLE必须定义固定的列结构，因此当groupByConnector为FALSE时，SUM_ZZZ列会返回NULL，避免因列数不一致报错。

进阶方案（若需完全移除列） #

如果必须在FALSE时完全去掉SUM_ZZZ列，由于Snowflake SQL存储过程要求返回结构固定，可选择两种折中方式：

• 改用返回VARIANT类型，将查询结果转为JSON格式后返回，外部再解析结构。
• 分支执行不同查询：根据groupByConnector的值分别执行包含/不包含SUM_ZZZ的查询，但需确保两个查询的返回结构与RETURNS TABLE定义匹配（或放弃强类型返回）。

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