Snowflake存储过程运行时切换默认数据库的可行方案问询

1. 修复包装存储过程的返回值问题 #

你的包装存储过程无法返回数据，核心原因是未捕获被调用存储过程的返回值并显式返回。以下是修改后的代码，同时加入动态数据库名逻辑避免硬编码：

CREATE OR REPLACE PROCEDURE DEV1.BI.AStoredProcedure_Wrapper()
    RETURNS VARCHAR(16777216)
    LANGUAGE SQL
    EXECUTE AS CALLER
AS '
DECLARE
    result VARCHAR(16777216);
    target_db VARCHAR(255) := CURRENT_FUNCTION_DATABASE(); -- 获取当前存储过程所属数据库
BEGIN
    -- 动态切换到存储过程所在数据库，适配环境克隆场景
    EXECUTE IMMEDIATE ''USE DATABASE '' || target_db;
    -- 捕获被调用存储过程的返回值
    CALL BI.AStoredProcedure() INTO :result;
    -- 返回最终结果
    RETURN :result;
END;
';

关键细节： #

• CURRENT_FUNCTION_DATABASE()：自动获取当前存储过程所在的数据库，无需硬编码，完美支持生产/测试环境克隆的架构需求。
• CALL ... INTO :result：将目标存储过程的返回值赋值给变量，再通过RETURN语句传递给调用方。
• EXECUTE IMMEDIATE：动态执行数据库切换语句，确保会话上下文切换到正确的数据库。

2. 会话初始化切换数据库（无需包装过程） #

如果应用程序支持在会话启动时执行初始化语句，可先执行：

USE DATABASE DEV1;

之后直接调用目标存储过程：

CALL BI.AStoredProcedure();

这种方式无需创建包装存储过程，直接修正会话的数据库上下文，从根源解决对象找不到的问题。

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