Snowflake SQL动态查询：将condition列条件转为result布尔列

方案1：使用存储过程批量生成执行语句 #

如果逐行调用EXECUTE IMMEDIATE失败，可通过存储过程遍历表数据，批量生成并执行更新语句，避免逐行调用的性能或语法问题。

存储过程代码 #

CREATE OR REPLACE PROCEDURE evaluate_conditions()
RETURNS VARCHAR
LANGUAGE SQL
AS
$$
DECLARE
    rec RECORD;
    stmt VARCHAR;
BEGIN
    -- 先给temp_table添加result列（如果还没加）
    ALTER TABLE temp_table ADD COLUMN IF NOT EXISTS result BOOLEAN;
    
    -- 遍历所有行，生成更新语句
    FOR rec IN SELECT col1, col2, condition, CURRENT_ROW() AS row_id FROM temp_table DO
        stmt := 'UPDATE temp_table SET result = ' || rec.condition || ' WHERE CURRENT_ROW() = ' || rec.row_id || ';';
        EXECUTE IMMEDIATE stmt;
    END FOR;
    
    RETURN '条件计算完成';
END;
$$;

-- 调用存储过程
CALL evaluate_conditions();

说明 #

• 用CURRENT_ROW()作为行唯一标识，确保更新对应行；如果表有主键，用主键更稳妥。
• 批量执行比逐行手动调用更可靠，避免单次执行的语法错误积累。

方案2：使用JavaScript UDF（更高效） #

利用Snowflake支持的JavaScript UDF，直接在UDF中计算条件表达式，无需循环遍历，性能更优。

UDF及调用代码 #

-- 添加result列（如果还没加）
ALTER TABLE temp_table ADD COLUMN IF NOT EXISTS result BOOLEAN;

-- 创建JavaScript UDF，计算条件表达式
CREATE OR REPLACE FUNCTION evaluate_condition(col1 INT, col2 INT, cond VARCHAR)
RETURNS BOOLEAN
LANGUAGE JAVASCRIPT
AS
$$
    // 将表达式中的col1、col2替换为传入的实际值
    const evalExpr = cond.replace(/col1/g, COL1).replace(/col2/g, COL2);
    // 执行表达式并返回布尔结果
    return Boolean(eval(evalExpr));
$$;

-- 更新result列
UPDATE temp_table SET result = evaluate_condition(col1, col2, condition);

注意事项 #

• 安全风险：如果condition列的数据不可信（比如来自用户输入），eval()可能导致恶意代码执行，仅在可控环境下使用。
• 确保condition中的表达式语法正确，比如col1 > col2、col1 + col2 = 10这类合法的布尔表达式。

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