test connection fixed

blake2002 · Jun 14, 2022 · 2cc2234 · 2cc2234
1 parent f87f0d6
commit 2cc2234
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/plugins/packages/athena/lib/index.ts b/plugins/packages/athena/lib/index.ts
@@ -41,6 +41,28 @@ export default class Athena implements QueryService {
   }
 
   async testConnection(sourceOptions: SourceOptions): Promise<ConnectionTestResult> {
+    const AthenaExpress = require('athena-express'),
+      AWS = require('aws-sdk'),
+      awsCredentials = {
+        region: sourceOptions.region,
+        accessKeyId: sourceOptions.access_key,
+        secretAccessKey: sourceOptions.secret_key,
+      };
+
+    AWS.config.update(awsCredentials);
+
+    const athenaExpressConfig = {
+      aws: AWS,
+      getStats: true,
+      db: sourceOptions.database,
+    };
+
+    const athenaExpress = new AthenaExpress(athenaExpressConfig);
+    try {
+      await athenaExpress.query('SHOW TABLES');
+    } catch (error) {
+      throw new Error(error);
+    }
     return {
       status: 'ok',
     };