如何让AWS boto3 redshift-data客户端等待SQL语句执行完成？

boto3 目前没有为 redshift-data 客户端提供官方内置的SQL执行状态等待器，现有Redshift服务下的waiter均针对集群创建、扩容、快照等资源管理类操作，无法用于SQL语句执行的状态监听，需要自行实现轮询逻辑。

以下是可直接使用的轮询实现示例：

import boto3
import time

client = boto3.client('redshift-data')

def wait_for_statement_completion(statement_id, poll_interval=5, timeout=3600):
    start_time = time.time()
    while True:
        # 超时判断，避免无限等待
        if time.time() - start_time > timeout:
            raise TimeoutError(f"SQL执行超时，已等待{timeout}秒")
        
        describe_resp = client.describe_statement(Id=statement_id)
        current_status = describe_resp['Status']
        
        # 命中终止状态直接返回结果
        if current_status in ['SUCCESS', 'FAILED', 'ABORTED']:
            return describe_resp
        
        # 中间状态等待后重试
        time.sleep(poll_interval)

# 业务调用示例
exec_resp = client.execute_statement(
    ClusterIdentifier='xyz',
    Database='dev',
    DbUser='user1',
    Sql='vacuum table1;',
    StatementName='vacuum_table1',
    WithEvent=False
)

# 阻塞等待执行完成
final_result = wait_for_statement_completion(exec_resp['Id'])
print(f"执行最终状态：{final_result['Status']}")
if final_result['Status'] == 'FAILED':
    print(f"执行失败原因：{final_result['Error']}")

补充说明 #

• 执行查询类SQL拿到SUCCESS状态后，可以调用 client.get_statement_result(Id=statement_id) 接口获取具体的查询返回数据
• 针对Vacuum、大表DDL等长耗时操作，可以适当调大poll_interval（轮询间隔）和timeout（超时时间）参数，减少不必要的API调用
• 如果配置了WithEvent=True，也可以配合EventBridge监听SQL执行完成的事件触发后续逻辑，无需自行维护轮询进程

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