Skip to content

feat(athena-driver): add session-token and database env vars #6018

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 7 commits into from
Closed

feat(athena-driver): add session-token and database env vars #6018

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
c3359c1
feat(athena-driver): add session-token and database env vars
paulorsbrito Jan 16, 2023
39d478e
update Athena driver docs
paulorsbrito Jan 16, 2023
c35c82d
fix comment
paulorsbrito Jan 17, 2023
93d406f
chore: use CUBEJS_DB_NAME for Athena default database
paulorsbrito Jan 26, 2023
8078de7
chore: revert doc changes
paulorsbrito Jan 26, 2023
b3dccf6
chore: revert add env to aws access token
paulorsbrito Feb 8, 2023
afa66cf
Merge branch 'master' into analytics-customizations
KSDaemon Jul 1, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 8 additions & 1 deletion packages/cubejs-athena-driver/src/AthenaDriver.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,7 @@ interface AthenaDriverOptions extends AthenaClientConfig {
workGroup?: string
catalog?: string
schema?: string
database?: string
S3OutputLocation?: string
exportBucket?: string
pollTimeout?: number
Expand Down Expand Up @@ -147,6 +148,9 @@ export class AthenaDriver extends BaseDriver implements DriverInterface {
catalog:
config.catalog ||
getEnv('athenaAwsCatalog', { dataSource }),
database:
config.database ||
getEnv('dbName', { dataSource }),
exportBucket:
config.exportBucket ||
getEnv('dbExportBucket', { dataSource }),
Expand Down Expand Up @@ -477,7 +481,10 @@ export class AthenaDriver extends BaseDriver implements DriverInterface {
ResultConfiguration: {
OutputLocation: this.config.S3OutputLocation
},
...(this.config.catalog != null ? { QueryExecutionContext: { Catalog: this.config.catalog } } : {})
QueryExecutionContext: {
Catalog: this.config.catalog,
Database: this.config.database
}
};
const { QueryExecutionId } = await this.athena.startQueryExecution(request);
return { QueryExecutionId: checkNonNullable('StartQueryExecution', QueryExecutionId) };
Expand Down
Loading