Snowflake存储过程中如何通过参数设置会话变量调用视图？

原代码存在的问题 #

• DECLARE块内不能直接执行SET语句，这类可执行操作必须放在BEGIN...END代码块中。
• 会话变量的赋值逻辑在视图查询之后执行，导致视图无法读取到传入的参数值。
• resultset的默认初始化会提前执行视图查询，此时会话变量还未被正确设置。

修正后的存储过程代码 #

CREATE PROCEDURE TEST_PROCEDURE(DATE_FROM_a DATE, DATE_TO_a DATE)
  RETURNS TABLE()
  LANGUAGE SQL
  EXECUTE AS CALLER
  AS 
  $$
  DECLARE
    res resultset;
  BEGIN
    -- 先设置会话变量
    SET DATE_FROM = :DATE_FROM_a;
    SET DATE_TO = :DATE_TO_a; -- 若视图v_TEST无需DATE_TO变量可删除此句
    
    -- 会话变量设置完成后，查询视图并赋值给结果集
    res := (SELECT TOP 100 * FROM v_TEST);
    
    RETURN TABLE(res);
  END;
  $$
  ;

关键调整说明 #

1. 将SET语句移至BEGIN块内，确保会话变量在视图查询前完成赋值，让视图能读取到正确的参数值。
2. 不在DECLARE中初始化resultset，而是在会话变量设置完成后通过赋值操作执行视图查询，避免提前执行导致的变量未生效问题。

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