Boto3调用Athena查询结果与AWS控制台不一致的解决求助

我想用Athena执行查询语句：select * from information_schema.tables，但遇到了结果不一致的问题：

• 通过Boto3客户端执行（代码如下）时，返回9行结果，耗时0.6秒
• 在AWS控制台执行相同查询，返回正确的500行结果，耗时6秒

已排查的情况：

• 对比查询历史，确认查询语句完全一致
• 移除Boto3调用中的QueryExecutionContext参数，结果无变化
• 尝试了单双引号的所有组合，问题依旧

Boto3执行代码：

def run_query(query_string):
    query_execution_context = {"Catalog": "awsdatacatalog", "Database": "information_schema"}
    response = athena_client.start_query_execution(
        QueryString=query_string, QueryExecutionContext=query_execution_context, WorkGroup="primary"
    )
    return response

query_string_get_tables = "select * from information_schema.tables"
response = run_query(query_string_get_tables)

• 核心问题：未完整获取查询结果
  你当前的Boto3代码仅调用了start_query_execution，这个方法只返回查询的执行元数据（比如QueryExecutionId），并没有拉取实际的查询结果。AWS控制台会自动完成结果拉取和分页处理，而Boto3需要手动实现结果获取逻辑。

• 添加完整的结果获取逻辑
  修改代码，在启动查询后轮询查询状态，待查询完成后分页获取全部结果：

import time

def run_query_and_get_results(query_string):
    query_execution_context = {"Catalog": "awsdatacatalog", "Database": "information_schema"}
    # 启动查询
    start_response = athena_client.start_query_execution(
        QueryString=query_string, 
        QueryExecutionContext=query_execution_context, 
        WorkGroup="primary"
    )
    query_execution_id = start_response["QueryExecutionId"]
    
    # 轮询等待查询完成
    while True:
        exec_status = athena_client.get_query_execution(QueryExecutionId=query_execution_id)
        state = exec_status["QueryExecution"]["Status"]["State"]
        if state in ["SUCCEEDED", "FAILED", "CANCELLED"]:
            break
        time.sleep(1)
    
    if state != "SUCCEEDED":
        raise Exception(f"Query failed: {exec_status['QueryExecution']['Status'].get('StateChangeReason', 'Unknown error')}")
    
    # 分页获取全部结果
    full_results = []
    next_token = None
    while True:
        result_args = {"QueryExecutionId": query_execution_id}
        if next_token:
            result_args["NextToken"] = next_token
        
        result_response = athena_client.get_query_results(**result_args)
        full_results.extend(result_response["ResultSet"]["Rows"])
        
        next_token = result_response.get("NextToken")
        if not next_token:
            break
    
    return full_results

# 调用并查看结果数量
query_string_get_tables = "select * from information_schema.tables"
results = run_query_and_get_results(query_string_get_tables)
# 减去1是因为第一行是表头
print(f"Total rows returned: {len(results) - 1}")

• 验证WorkGroup配置
  检查primary工作组的设置，确认是否存在结果集大小限制或超时配置。如果工作组设置了最大返回行数，会导致Boto3调用只返回部分结果。可以在AWS控制台Athena的「工作组」页面查看primary组的查询限制参数，确保和控制台使用的配置一致。

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