uber
Support AWS Managed Output Buckets in AthenaDriver Configuration
Key Changes
-
Config API Enhancements:
Added a new optional flag to the configuration allowing users to enable AWS managed output buckets during setup instead of specifying a custom S3 output bucket.
Updated the NewDefaultConfig function and added SetManagedBucket/GetManagedBucket logic.
-
Query Execution Logic:
The QueryContext was refactored to conditionally set the ResultConfiguration in Athena queries.
If the AWS managed bucket feature is enabled, the ResultConfiguration.OutputLocation is omitted, aligning with best practices for Athena workgroups that use managed buckets.
-
Documentation and Example Updates:
The README.md and code examples now include documentation on configuring AthenaDriver to use AWS managed output buckets.
Provided a new example function: useAthenaDriverConfigForAuthWithManagedBucket, demonstrating authentication and connection with managed buckets in examples/auth.go.
Why This enhancement simplifies configuration for users leveraging AWS managed buckets and avoids conflicts with workgroups that already have managed output buckets enabled in AWS.
