Google Cloud Composer（Airflow）中Snowflake动态日期参数配置报错解决

你写的代码是Snowflake的存储过程匿名块（BEGIN/END）结构，但错误地使用了会话变量的引用语法:$VAR_QTR——块内通过DECLARE定义的是本地变量，而非会话级变量，因此无法通过会话变量的方式访问；同时存储过程内的变量赋值语法也有小问题，导致变量未被正确识别。

方案1：改用会话级变量（推荐，无需存储过程块） #

去掉DECLARE和BEGIN/END包裹，直接定义Snowflake会话级变量，整个SQL执行会话中都能复用：

-- 定义会话级变量
SET VAR_QTR = (
    SELECT 
    CASE 
        WHEN MONTH(CURRENT_DATE::DATE - INTERVAL '1 month') % 3 = 0 
        THEN DATE_TRUNC('quarter', CURRENT_DATE::DATE) - INTERVAL '1 quarter'
        ELSE DATE_TRUNC('quarter', CURRENT_DATE::DATE)
    END
);

-- 后续SQL直接引用会话变量
SOME SQL CODE HERE...
LEFT JOIN 
    TABLE dc
    ON dc.company_id = c.COMPANY_ID
    AND dc.deleted_at <= $VAR_QTR

这种方式无需依赖存储过程语法，Airflow的SnowflakeOperator可以直接执行，变量作用域覆盖整个SQL脚本，最适合你的复用需求。

方案2：修正存储过程块内的变量引用 #

如果必须保留BEGIN/END块（比如包含复杂逻辑），需调整变量的赋值和引用方式：

DECLARE
    VAR_QTR DATE;

BEGIN
    -- 存储过程内本地变量赋值用 := 或 SET
    VAR_QTR := (
        SELECT 
        CASE 
            WHEN MONTH(CURRENT_DATE::DATE - INTERVAL '1 month') % 3 = 0 
            THEN DATE_TRUNC('quarter', CURRENT_DATE::DATE) - INTERVAL '1 quarter'
            ELSE DATE_TRUNC('quarter', CURRENT_DATE::DATE)
        END
    );

    -- 块内直接用变量名引用，无需前缀
    SOME SQL CODE HERE...
    LEFT JOIN 
        TABLE dc
        ON dc.company_id = c.COMPANY_ID
        AND dc.deleted_at <= VAR_QTR;
END;

注意：Snowflake的匿名块可以直接被Airflow执行，无需额外的CALL语句，但要确保你的Snowflake账号有权限执行这类脚本。

方案3：用CTE预计算参数（无变量方案） #

如果不想用变量，可通过CTE预计算日期参数，后续SQL复用CTE的结果：

WITH qtr_param AS (
    SELECT 
    CASE 
        WHEN MONTH(CURRENT_DATE::DATE - INTERVAL '1 month') % 3 = 0 
        THEN DATE_TRUNC('quarter', CURRENT_DATE::DATE) - INTERVAL '1 quarter'
        ELSE DATE_TRUNC('quarter', CURRENT_DATE::DATE)
    END AS qtr_date
)

-- 后续SQL通过子查询引用CTE的参数
SOME SQL CODE HERE...
LEFT JOIN 
    TABLE dc
    ON dc.company_id = c.COMPANY_ID
    AND dc.deleted_at <= (SELECT qtr_date FROM qtr_param)

这种方式完全避免变量语法问题，适合逻辑简单、无需多次修改参数的场景。

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