Linux环境Snowflake SnowSQL中If-Then语句语法报错求助

你遇到的错误是因为Snowflake原生SQL不支持在顶层脚本中直接使用IF-THEN-ELSE控制流语句，这类逻辑语法只能在存储过程、用户定义函数（UDF）中使用，或者通过SnowSQL变量结合Shell分支实现，不能作为独立的SQL语句执行。

方案1：用CASE表达式简化为单条UPDATE语句 #

如果只是简单的条件更新，可将逻辑合并到UPDATE的SET子句中，无需IF分支：

#!/bin/ksh

export SNOWSQL_USER="my_usr"
export SNOWSQL_PWD="my_pwd"
export SNOWSQL_WAREHOUSE="my_wh"
export SNOWSQL_DATABASE="my_db"
export SNOWSQL_SCHEMA="my_schm"
export SNOWSQL_ACCOUNT="my_acct"

snowsql -o echo=true -o exit_on_error=True<<EOSQL

-- 创建临时表
CREATE TEMPORARY TABLE sample_table (id INT, value INT);

-- 插入测试数据
INSERT INTO sample_table VALUES (1, 10), (2, 20), (3, 30);

-- 用CASE表达式实现条件更新（用EXISTS替代COUNT(*)性能更优）
UPDATE sample_table 
SET value = CASE 
    WHEN EXISTS (SELECT 1 FROM sample_table WHERE value > 25) THEN 
        CASE WHEN value > 25 THEN value * 2 ELSE value * 3 END
    ELSE value * 3
END;

-- 查询更新结果
SELECT * FROM sample_table;

EOSQL

方案2：用Snowflake存储过程封装IF-THEN逻辑 #

如果需要复杂的多步骤分支逻辑，适合将代码封装到存储过程中执行：

#!/bin/ksh

export SNOWSQL_USER="my_usr"
export SNOWSQL_PWD="my_pwd"
export SNOWSQL_WAREHOUSE="my_wh"
export SNOWSQL_DATABASE="my_db"
export SNOWSQL_SCHEMA="my_schm"
export SNOWSQL_ACCOUNT="my_acct"

snowsql -o echo=true -o exit_on_error=True<<EOSQL

-- 创建临时表
CREATE TEMPORARY TABLE sample_table (id INT, value INT);

-- 插入测试数据
INSERT INTO sample_table VALUES (1, 10), (2, 20), (3, 30);

-- 创建存储过程
CREATE OR REPLACE PROCEDURE update_sample_table()
RETURNS VARCHAR
LANGUAGE SQL
AS
$$
BEGIN
    IF (SELECT COUNT(*) FROM sample_table WHERE value > 25) > 0 THEN
        UPDATE sample_table SET value = value * 2 WHERE value > 25;
    ELSE
        UPDATE sample_table SET value = value * 3;
    END IF;
    RETURN '更新完成';
END;
$$;

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

-- 查询更新结果
SELECT * FROM sample_table;

-- 清理存储过程（可选）
DROP PROCEDURE update_sample_table();

EOSQL

方案3：用SnowSQL变量+Shell分支 #

先通过SnowSQL查询条件结果，赋值给Shell变量，再根据变量值执行不同的SQL块：

#!/bin/ksh

export SNOWSQL_USER="my_usr"
export SNOWSQL_PWD="my_pwd"
export SNOWSQL_WAREHOUSE="my_wh"
export SNOWSQL_DATABASE="my_db"
export SNOWSQL_SCHEMA="my_schm"
export SNOWSQL_ACCOUNT="my_acct"

-- 先创建并填充临时表
snowsql -o echo=true -o exit_on_error=True<<EOSQL
CREATE TEMPORARY TABLE sample_table (id INT, value INT);
INSERT INTO sample_table VALUES (1, 10), (2, 20), (3, 30);
EOSQL

-- 查询是否存在value>25的记录，结果存入Shell变量
has_high_value=$(snowsql -o quiet=true -o header=false -q "SELECT IFF(EXISTS(SELECT 1 FROM sample_table WHERE value>25), 'YES', 'NO');")

-- 根据变量值执行不同的UPDATE
if [ "$has_high_value" = "YES" ]; then
    snowsql -o echo=true -o exit_on_error=True -q "UPDATE sample_table SET value = value * 2 WHERE value > 25;"
else
    snowsql -o echo=true -o exit_on_error=True -q "UPDATE sample_table SET value = value * 3;"
fi

-- 查询更新结果
snowsql -o echo=true -o exit_on_error=True -q "SELECT * FROM sample_table;"

• Snowflake原生SQL是声明式语言，不支持顶层控制流语句，必须借助存储过程、UDF或外部脚本分支实现条件逻辑。
• 优先使用CASE表达式、EXISTS这类SQL原生语法，性能和可读性更优；复杂逻辑再考虑存储过程。

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