将含变量重赋值的SQL代码转换为Snowflake存储过程遇阻求助

原始需求SQL（SQL Server风格） #

DECLARE @NumofDays int
SET @NumofDays = 15

Declare @loaddt smalldatetime

If @NumofDays = 1
BEGIN set @loaddt = getdate() end
Else
Begin set @loaddt = '1900-01-01' end

Insert into ##temptable
Select @loaddt as Loaddt

你的代码问题分析 #

1. 第一个版本报错原因：loaddt变量在内部BEGIN...END块中用LET声明，属于局部变量，外部INSERT语句无法访问，导致INVALID IDENTIFIER 'LOADDT'。
2. 第二个版本赋值不生效原因：内部块里再次用LET loaddt声明变量，相当于创建了一个新的局部变量，外部的loaddt根本没被修改，所以不管NumofDays怎么变，插入的都是初始值2023-02-10。

正确实现方案 #

核心要点 #

• 变量要在外部作用域声明，确保整个代码块都能访问
• 修改变量值时，直接用:=赋值，不要再次用LET声明（否则会创建局部变量覆盖外部变量）
• Snowflake中统一日期类型，推荐用DATE或TIMESTAMP_NTZ（DATETIME是TIMESTAMP_NTZ的别名）
• 临时表无需指定schema，LOCAL TEMPORARY TABLE是会话级的，自动隔离

写法1：用IF-ELSE分支赋值 #

-- 先创建临时表（如果需要重复执行，加OR REPLACE）
CREATE OR REPLACE LOCAL TEMPORARY TABLE TEMP (loaddate DATE);

BEGIN
    LET NUMofDays INT := 1;
    -- 在外部声明loaddt，确保全局可用
    LET loaddt DATE;

    -- 直接赋值，不重新声明变量
    IF NUMofDays = 1 THEN
        loaddt := CURRENT_DATE();
    ELSE
        loaddt := DATE('1900-01-01');
    END IF;

    INSERT INTO TEMP
    SELECT :loaddt AS loaddate;
END;

写法2：用CASE表达式直接赋值（更简洁） #

CREATE OR REPLACE LOCAL TEMPORARY TABLE TEMP (loaddate DATE);

BEGIN
    LET NUMofDays INT := 15;
    -- 用CASE表达式直接完成条件赋值，无需分支块
    LET loaddt DATE := CASE 
        WHEN NUMofDays = 1 THEN CURRENT_DATE()
        ELSE DATE('1900-01-01')
    END;

    INSERT INTO TEMP
    SELECT :loaddt AS loaddate;
END;

封装成存储过程（如果需要复用） #

CREATE OR REPLACE PROCEDURE LOAD_DATE_TO_TEMP(NUMofDays INT)
RETURNS VARCHAR
LANGUAGE SQL
AS
$$
BEGIN
    -- 创建临时表（每次调用重建）
    CREATE OR REPLACE LOCAL TEMPORARY TABLE TEMP (loaddate DATE);

    LET loaddt DATE := CASE 
        WHEN NUMofDays = 1 THEN CURRENT_DATE()
        ELSE DATE('1900-01-01')
    END;

    INSERT INTO TEMP
    SELECT :loaddt AS loaddate;

    RETURN '数据插入完成，当前loaddt值: ' || :loaddt;
END;
$$;

-- 调用存储过程
CALL LOAD_DATE_TO_TEMP(1);
CALL LOAD_DATE_TO_TEMP(15);

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