如何通过T-SQL获取rds_restore_database返回的task_id并查询恢复状态

首先明确核心前提：rds_restore_database和rds_task_status两个RDS内置存储过程，没有提供OUTPUT类型的输出参数，所有业务返回值（包括task_id、lifecycle等字段）都通过存储过程执行后的结果集返回，你之前用exec @task_id = 存储过程的写法是错误的——该语法仅能获取存储过程的RETURN返回码，无法拿到结果集中的业务字段。

不需要反复创建/删除临时表，用表变量承接结果是最轻量高效的方案，表变量在作用域内可复用，出作用域自动释放，小数据量下内存驻留，开销远低于临时表。

步骤1：执行恢复操作并获取task_id #

先定义和存储过程返回结构兼容的表变量，执行存储过程后直接从表变量提取task_id即可：

DECLARE @RestoreResult TABLE (
    task_id INT,
    task_type NVARCHAR(100),
    database_name NVARCHAR(100),
    [status] NVARCHAR(50),
    lifecycle NVARCHAR(50),
    create_time DATETIME,
    -- 剩余不需要的字段用占位列，保证列数、类型和返回结果兼容即可
    col_placeholder1 NVARCHAR(MAX),
    col_placeholder2 NVARCHAR(MAX),
    col_placeholder3 NVARCHAR(MAX)
)
DECLARE @MyTaskID INT

-- 执行恢复，结果写入表变量
INSERT INTO @RestoreResult
EXEC msdb.dbo.rds_restore_database 
    @restore_db_name = 'whatever',
    @s3_arn_to_restore_from = 'S3_bucket_bak'

-- 提取task_id到变量
SELECT TOP 1 @MyTaskID = task_id FROM @RestoreResult

步骤2：轮询任务状态，提取lifecycle字段判断进度 #

定义一次状态承接用的表变量，轮询时仅需清空表数据即可重复使用，不需要重复建删表：

DECLARE @TaskStatusResult TABLE (
    task_id INT,
    task_type NVARCHAR(100),
    database_name NVARCHAR(100),
    [status] NVARCHAR(50),
    lifecycle NVARCHAR(50),
    create_time DATETIME,
    end_time DATETIME,
    task_info NVARCHAR(MAX),
    col_placeholder NVARCHAR(MAX)
)
DECLARE @CurrentLifecycle NVARCHAR(50)
-- 轮询间隔设为5秒即可，过于频繁会给实例造成不必要的压力
DECLARE @PollInterval INT = 5

WHILE 1=1
BEGIN
    -- 清空表变量历史数据，开销远低于删重建临时表
    DELETE FROM @TaskStatusResult

    -- 查询最新任务状态写入表变量
    INSERT INTO @TaskStatusResult
    EXEC msdb.dbo.rds_task_status @task_id = @MyTaskID

    -- 提取当前生命周期状态
    SELECT TOP 1 @CurrentLifecycle = lifecycle FROM @TaskStatusResult

    -- 状态分支判断
    IF @CurrentLifecycle = 'SUCCESS'
    BEGIN
        PRINT '数据库恢复任务执行成功'
        BREAK
    END
    IF @CurrentLifecycle IN ('ERROR', 'FAILED', 'CANCELLED')
    BEGIN
        DECLARE @ErrorInfo NVARCHAR(MAX)
        SELECT TOP 1 @ErrorInfo = task_info FROM @TaskStatusResult
        RAISERROR('数据库恢复任务失败，详情：%s', 16, 1, @ErrorInfo)
        BREAK
    END

    -- 等待间隔后进入下一轮查询
    WAITFOR DELAY CONVERT(VARCHAR, DATEADD(SECOND, @PollInterval, 0), 108)
END

注意事项 #

• 定义承接结果的表变量时，不需要完全匹配所有返回列的名称，但必须保证列的总数量、对应位置列的数据类型兼容，否则会触发插入列不匹配报错。如果不确定返回结构，可以先单独执行一次对应存储过程，查看结果集字段后调整表变量定义。
• 表变量相比临时表没有额外的日志开销、不需要手动释放、不存在多会话命名冲突问题，非常适合这类单次存储过程内的小结果集承接场景。
• 轮询间隔不要设置小于3秒，避免高频调用rds_task_status占用实例资源。

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