Merge pull request #341 from eagletmt/aws-profile

Support AWS profile for Athena/Timestream

Author: morishin

src/lib/AthenaClient.ts

@@ -3,6 +3,7 @@ import retry from "./Util/retry";
 
 interface AthenaClientConfig {
   region: string;
+  profile?: string;
   accessKeyId?: string;
   secretAccessKey?: string;
   database: string;
@@ -17,11 +18,15 @@ export default class AthenaClient {
   executionId: string;
 
   constructor(config: AthenaClientConfig) {
-    const { region, accessKeyId, secretAccessKey } = config;
+    const { region, profile, accessKeyId, secretAccessKey } = config;
     this.config = config;
 
     const clientConfig: any = { region };
 
+    if (profile) {
+      clientConfig.profile = profile;
+    }
+
     // Only add credentials if BOTH accessKeyId and secretAccessKey are provided.
     if (accessKeyId && secretAccessKey) {
       clientConfig.credentials = { accessKeyId, secretAccessKey };

src/lib/DataSourceDefinition/Athena.ts

@@ -20,6 +20,12 @@ export default class Athena extends Base {
         placeholder: "us-east-1",
         required: true,
       },
+      {
+        name: "profile",
+        label: "Profile",
+        type: "string",
+        placeholder: "Optional - uses AWS default profile if empty",
+      },
       {
         name: "accessKeyId",
         label: "Access key ID",

src/lib/DataSourceDefinition/Timestream.ts

@@ -24,6 +24,12 @@ export default class Timestream extends Base {
         placeholder: "us-east-1",
         required: true,
       },
+      {
+        name: "profile",
+        label: "Profile",
+        type: "string",
+        placeholder: "Optional - uses AWS default profile if empty",
+      },
       {
         name: "accessKeyId",
         label: "Access key ID",
@@ -50,6 +56,10 @@ export default class Timestream extends Base {
 
     const clientConfig: any = { region: config.region };
 
+    if (config.profile) {
+      clientConfig.profile = config.profile;
+    }
+
     // Only add credentials if BOTH accessKeyId and secretAccessKey are provided.
     if (config.accessKeyId && config.secretAccessKey) {
       clientConfig.credentials = {