Snowflake中如何将JS UDF返回的SQL语句拼接至SQL UDF的WHERE子句

你原写法无法生效的核心原因是：静态SQL的语法校验阶段会提前验证WHERE后所有条件必须为布尔类型，返回字符串的UDF2会在解析阶段直接报类型不匹配错误，无法在运行时自动将字符串识别为SQL语法片段，必须通过动态SQL实现需求。

方案1：SQL匿名脚本动态拼接执行（适用于即席查询场景） #

前置准备：示例UDF2定义（JavaScript语言） #

CREATE OR REPLACE FUNCTION DB.SCHEMA.UDF2(PARM_1 VARCHAR)
RETURNS VARCHAR
LANGUAGE JAVASCRIPT
AS
$$
// 可根据业务需求修改返回的条件片段逻辑
if (PARM_1 === 'TEST') {
    return "COL_A = 'VALUE1' AND COL_B > 100";
} else {
    return "COL_C IS NOT NULL AND COL_D IN ('A','B')";
}
$$;

动态拼接执行脚本 #

DECLARE
    -- 存储UDF2返回的条件片段
    v_cond_segment VARCHAR;
    -- 存储最终拼接完成的SQL语句
    v_final_query VARCHAR;
BEGIN
    -- 调用UDF2获取条件字符串
    v_cond_segment := DB.SCHEMA.UDF2('TEST');
    -- 拼接完整SQL，将条件片段插入WHERE子句对应位置
    v_final_query := 'SELECT * FROM DB.SCHEMA.UDF1() WHERE condition1 AND condition2 AND (' || v_cond_segment || ')';
    -- 执行动态SQL并返回结果
    RETURN TABLE(EXECUTE IMMEDIATE :v_final_query);
END;
/

方案2：封装为存储过程（适用于复用/定时调度场景） #

如果需要对外提供统一调用入口，可将逻辑封装为存储过程：

CREATE OR REPLACE PROCEDURE DB.SCHEMA.CALL_UDF1_WITH_DYNAMIC_COND(PARM_1 VARCHAR)
-- 此处返回表结构需和UDF1的返回结构完全一致
RETURNS TABLE(
    COL1 VARCHAR,
    COL2 NUMBER,
    -- 补充其余字段定义
    COLN TIMESTAMP
)
LANGUAGE SQL
AS
$$
DECLARE
    v_cond_segment VARCHAR;
    v_final_query VARCHAR;
BEGIN
    v_cond_segment := DB.SCHEMA.UDF2(:PARM_1);
    v_final_query := 'SELECT * FROM DB.SCHEMA.UDF1() WHERE condition1 AND condition2 AND (' || v_cond_segment || ')';
    RETURN TABLE(EXECUTE IMMEDIATE :v_final_query);
END;
$$;

-- 调用示例
CALL DB.SCHEMA.CALL_UDF1_WITH_DYNAMIC_COND('TEST');

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