如何在Snowflake存储过程中遍历Schema内所有Stage并调用存储过程

你可以通过SHOW STAGES结合RESULT_SCAN()获取指定Schema下的所有外部Stage，再用游标循环遍历每个Stage名称，自动调用CHECK_LOAD()存储过程。以下是改进后的完整实现：

CREATE OR REPLACE PROCEDURE MANUAL_ADJUSTMENTS.TASK_FAILURE_ALERT()
RETURNS VARIANT
LANGUAGE SQL
EXECUTE AS CALLER
AS
$$
DECLARE
    -- 声明游标，用于遍历指定Schema下的所有外部Stage
    stage_cursor CURSOR FOR
        SELECT "name" AS stage_name
        FROM TABLE(RESULT_SCAN(LAST_QUERY_ID()))
        WHERE "schema_name" = 'MANUAL_ADJUSTMENTS' AND "type" = 'EXTERNAL';
    stage_name VARCHAR;
BEGIN
    -- 执行SHOW STAGES命令获取当前Schema下的所有Stage
    SHOW STAGES IN SCHEMA manual_adjustments;
    
    -- 循环遍历每个外部Stage
    FOR stage_name IN stage_cursor DO
        -- 动态调用CHECK_LOAD存储过程，传入当前Stage名称
        CALL manual_adjustments.CHECK_LOAD(stage_name);
    END FOR;
    
    RETURN {'status': 'success', 'message': '所有外部Stage检查已完成'};
END;
$$;

关键说明： #

• SHOW STAGES IN SCHEMA manual_adjustments;：获取指定Schema下的所有Stage（包含内部和外部）
• RESULT_SCAN(LAST_QUERY_ID())：将SHOW STAGES的结果转换为可查询的关系表，方便后续过滤和遍历
• WHERE "schema_name" = 'MANUAL_ADJUSTMENTS' AND "type" = 'EXTERNAL'：过滤出当前Schema下的外部Stage（匹配你需要的S3外部Stage）
• 游标stage_cursor：遍历过滤后的Stage列表，逐个取出Stage名称
• FOR stage_name IN stage_cursor DO ... END FOR：SQL存储过程中的循环语法，自动处理每个Stage的调用

额外优化点： #

如果需要确认只处理S3类型的外部Stage，可以在过滤条件中增加对storage_integration的判断（假设你的S3外部Stage关联了对应的存储集成）：

WHERE "schema_name" = 'MANUAL_ADJUSTMENTS' 
  AND "type" = 'EXTERNAL'
  AND "storage_integration" = '你的S3存储集成名称';

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