如何在Snowflake SnowSQL中结合变量使用IF语句？

在PROD环境执行克隆语句时，即使加了IF NOT EXISTS依然报错，核心原因是Snowflake不允许克隆表到自身——当你在PROD_RAW.S1库下执行CREATE TABLE IF NOT EXISTS PM CLONE PROD_RAW.S1.PM时，源表和目标表完全是同一个对象，这种情况下IF NOT EXISTS的检查不会生效，因为克隆操作的前置验证会直接报错。

1. 强化环境判断的健壮性 #

你已经通过IF $database_name <> 'PROD_RAW'跳过了PROD环境的克隆，但可以优化判断逻辑，避免大小写或空格导致的判断失效：

BEGIN
    -- 统一转为大写并去除空格，避免环境参数大小写不一致的问题
    IF UPPER(TRIM($database_name)) <> 'PROD_RAW' THEN
        CREATE TABLE IF NOT EXISTS PM
        CLONE PROD_RAW.S1.PM;

        CREATE TABLE IF NOT EXISTS AP
        CLONE PROD_RAW.S1.AP;
    END IF;
END;

2. 确保后续ALTER语句的兼容性 #

对于ALTER TABLE PM ADD COLUMN IF NOT EXISTS status BOOLEAN;，在PROD环境中PM表是原始表，直接执行没问题；在非PROD环境中是克隆出来的表，IF NOT EXISTS也能保证不会重复添加。如果担心PROD环境中PM表还未创建（虽然作为源表应该已经存在），可以套一层条件判断：

-- 先检查表是否存在，再执行ALTER，增强健壮性
IF EXISTS (SELECT 1 FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = 'S1' AND TABLE_NAME = 'PM') THEN
    ALTER TABLE PM
    ADD COLUMN IF NOT EXISTS status BOOLEAN;
END IF;

3. 变量初始化的优化 #

为了避免环境参数&env带空格或特殊字符，初始化database_name时增加清理逻辑：

SET database_name = CONCAT(UPPER(TRIM('&env')), '_RAW');

ALTER SESSION SET TRANSACTION_ABORT_ON_ERROR = true;

SELECT '&env'; -- 返回PROD、STG或DEV

-- 清理环境参数并拼接数据库名
SET database_name = CONCAT(UPPER(TRIM('&env')), '_RAW');
SHOW VARIABLES;
USE IDENTIFIER($database_name);

CREATE SCHEMA IF NOT EXISTS S1;
USE SCHEMA S1;

BEGIN
    IF UPPER(TRIM($database_name)) <> 'PROD_RAW' THEN
        CREATE TABLE IF NOT EXISTS PM
        CLONE PROD_RAW.S1.PM;

        CREATE TABLE IF NOT EXISTS AP
        CLONE PROD_RAW.S1.AP;
    END IF;
END;

------------------------------------------------------------------

-- 先检查表存在再执行ALTER，增强健壮性
IF EXISTS (SELECT 1 FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = 'S1' AND TABLE_NAME = 'PM') THEN
    ALTER TABLE PM
    ADD COLUMN IF NOT EXISTS status BOOLEAN;
END IF;

------------------------------------------------------------------

CREATE SCHEMA IF NOT EXISTS D1;
USE SCHEMA D1;

------------------------------------------------------------------
CREATE TABLE IF NOT EXISTS DATA_MAPPING (
    key_column VARCHAR,
    scope VARCHAR,
    value_from VARCHAR,
    value_to VARCHAR
);

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