Snowflake SQL存储过程中SET变量编译通过却运行报错求助

问题原因 #

你在存储过程里用SET retVal = ''是错误的，核心原因是混淆了两种变量的赋值规则：

• 存储过程中DECLARE块定义的本地变量，不能用SET语句赋值。SET仅适用于设置会话级变量（比如SET MY_GLOBAL_VAR = 'value';），这也是官方文档提到的适用场景，但完全不适用于存储过程内部的本地变量。
• 编译阶段通过是因为SET本身是合法SQL语句，但运行时Snowflake会校验存储过程内的语句类型，发现SET用于本地变量赋值不被支持，因此抛出异常。

解决方法 #

给存储过程本地变量赋值有两种正确方式：

方式一：在DECLARE块中直接初始化 #

DROP PROCEDURE IF EXISTS MY_DB.MY_SCHEMA.MY_SP_SQL();
CREATE PROCEDURE MY_DB.MY_SCHEMA.MY_SP_SQL()
RETURNS STRING NOT NULL
  LANGUAGE SQL
AS
  DECLARE
    retValue VARCHAR DEFAULT ''; -- 直接在声明时初始化
  BEGIN
    -- 后续业务逻辑
    RETURN retValue;
  END;

方式二：在BEGIN块中使用赋值运算符:= #

DROP PROCEDURE IF EXISTS MY_DB.MY_SCHEMA.MY_SP_SQL();
CREATE PROCEDURE MY_DB.MY_SCHEMA.MY_SP_SQL()
RETURNS STRING NOT NULL
  LANGUAGE SQL
AS
  DECLARE
    retValue VARCHAR;
  BEGIN
    retValue := ''; -- 使用:=为本地变量赋值
    -- 后续业务逻辑
    RETURN retValue;
  END;

补充说明 #

如果确实需要在存储过程中操作会话变量，需确保变量是会话级别的，示例如下：

BEGIN
  SET MY_SESSION_VAR = 'test'; -- 操作会话变量是合法的
  RETURN CURRENT_SESSION_VARIABLE('MY_SESSION_VAR');
END;

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