AWS Lambda调用Athena查询无返回且报InvalidRequestException求助

你当前的报错本质是Athena查询实际执行失败，但你直接固定等待10秒就调用get_query_results接口，既没有提前校验查询状态，也没有捕获具体的失败原因。你在Athena控制台可以正常运行SQL，说明问题出在运行时权限、配置层面，和SQL本身无关。

第一步：获取查询失败的具体原因 #

你可以先调用get_query_execution接口拿到查询的状态和错误详情，在获取结果前先打印错误信息，90%的问题都可以通过这个操作直接定位：

# 替换原有的sleep + get_query_results逻辑片段
query_execution = client.get_query_execution(QueryExecutionId=queryId)
query_status = query_execution['QueryExecution']['Status']['State']
print(f"查询状态: {query_status}")
if query_status == 'FAILED':
    print(f"失败原因: {query_execution['QueryExecution']['Status']['StateChangeReason']}")

常见的失败原因如下：

• Lambda执行角色的权限不足：需要给Lambda绑定的IAM角色添加三类权限：
  1. Athena操作权限：athena:StartQueryExecution、athena:GetQueryExecution、athena:GetQueryResults
  2. S3输出桶权限：对应你配置的s3://mybucket/路径的s3:PutObject、s3:GetObject权限，如果桶开启了KMS加密，还要额外加KMS解密权限
  3. Glue元数据权限：如果你的raw数据库托管在Glue中，还要给角色添加Glue数据库、表的读取权限
• 输出路径配置错误：检查OutputLocation填写的桶路径是否正确，桶是否和Athena、Lambda在同一区域
• 资源并发限制：如果同一时间有大量Athena查询在运行，也可能导致查询排队超时失败

第二步：替换固定等待逻辑，使用轮询判断查询状态 #

固定sleep10秒的逻辑可靠性很差，如果查询耗时超过10秒还没跑完，调用get_query_results也会触发报错，建议改成轮询校验状态的逻辑：

query_execution_id = QueryResponse['QueryExecutionId']
state = 'RUNNING'
# 可根据业务需求调整最大超时时间，单位为秒
max_wait_time = 30
while max_wait_time > 0 and state in ['RUNNING', 'QUEUED']:
    max_wait_time -= 1
    response = client.get_query_execution(QueryExecutionId = query_execution_id)
    state = response['QueryExecution']['Status']['State']
    if state == 'SUCCEEDED':
        results = client.get_query_results(QueryExecutionId=query_execution_id)
        for row in results['ResultSet']['Rows']:
            print(row)
        break
    if state == 'FAILED':
        print(f"查询失败: {response['QueryExecution']['Status']['StateChangeReason']}")
        break
    time.sleep(1)

你可以先在本地用相同的身份凭证运行代码，打印具体的失败原因，比在Lambda环境中调试效率高很多，确认代码和权限配置没问题之后再部署到Lambda即可。

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