Skip to content

fix: Support Athena workgroups with managed query results (#3174)#3275

Open
Will282 wants to merge 4 commits intoaws:mainfrom
Will282:fix/issue_#3174
Open

fix: Support Athena workgroups with managed query results (#3174)#3275
Will282 wants to merge 4 commits intoaws:mainfrom
Will282:fix/issue_#3174

Conversation

@Will282
Copy link

@Will282 Will282 commented Mar 4, 2026

Feature or Bugfix

  • Bugfix

Detail

  • Support Athena workgroups with managed query results by detecting ManagedQueryResultsConfiguration and conditionally omitting ResultConfiguration from StartQueryExecution API calls
  • Add GetQueryResults API-based result reader (_fetch_api_result) to read results from Athena-owned storage when no S3 OutputLocation is available, with support for both full and chunked reads
  • Skip unnecessary _get_s3_output() call (and its bucket-creation side effect) in the regular query path when managed results is enabled
  • Validate that result_reuse_configuration is not used with managed results workgroups, as AWS does not support this combination
  • Route get_query_results() and cached-result paths through the API result reader when output_location is None
  • Update s3_output docstrings on start_query_execution and read_sql_query to document managed results behaviour

Relates

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

This PR was co-written with AI assistance.

@kukushking
Copy link
Contributor

AWS CodeBuild CI Report

  • CodeBuild project: GitHubCodeBuild8756EF16-4rfo0GHQ0u9a
  • Commit ID: 279f48f
  • Result: SUCCEEDED
  • Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

@kukushking
Copy link
Contributor

AWS CodeBuild CI Report

  • CodeBuild project: GitHubCodeBuild8756EF16-4rfo0GHQ0u9a
  • Commit ID: 9b3b975
  • Result: SUCCEEDED
  • Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

@kukushking
Copy link
Contributor

AWS CodeBuild CI Report

  • CodeBuild project: GitHubCodeBuild8756EF16-4rfo0GHQ0u9a
  • Commit ID: 9229da1
  • Result: SUCCEEDED
  • Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants