Merge 58fd191 into 895b929

KoteiIto · Mar 7, 2018 · 5166cd0 · 5166cd0
2 parents 895b929 + 58fd191
commit 5166cd0
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 1 deletion.
diff --git a/build/lib/request.d.ts b/build/lib/request.d.ts
@@ -5,6 +5,7 @@ export interface AthenaRequestConfig {
     baseRetryWait?: number;
     retryWaitMax?: number;
     retryCountMax?: number;
+    database?: string;
 }
 export declare class AthenaRequest {
     private athena;

diff --git a/build/lib/request.js b/build/lib/request.js
diff --git a/build/lib/request.js.map b/build/lib/request.js.map
diff --git a/src/lib/request.ts b/src/lib/request.ts
@@ -6,6 +6,7 @@ export interface AthenaRequestConfig {
   baseRetryWait?: number
   retryWaitMax?: number
   retryCountMax?: number
+  database?: string
 }
 
 const defaultBaseRetryWait = 200
@@ -27,6 +28,9 @@ export class AthenaRequest {
         ResultConfiguration: {
           OutputLocation: config.bucketUri,
         },
+        QueryExecutionContext: {
+          Database: config.database || 'default',
+        },
       }
       const loopFunc = () => {
         this.athena.startQueryExecution(params, (err: AWSError, data: any) => {