From c6c7e9601da96c8b98bb6ab592d1296d425e3e8e Mon Sep 17 00:00:00 2001
From: Artyom Keydunov <artyom.keydunov@gmail.com>
Date: Wed, 13 Feb 2019 11:24:59 +0300
Subject: [PATCH] Add support for Athena in JDBC driver

---
 .../cubejs-jdbc-driver/driver/JDBCDriver.js   | 24 +++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/packages/cubejs-jdbc-driver/driver/JDBCDriver.js b/packages/cubejs-jdbc-driver/driver/JDBCDriver.js
index af4ac9311237d..47fe3a77b580a 100644
--- a/packages/cubejs-jdbc-driver/driver/JDBCDriver.js
+++ b/packages/cubejs-jdbc-driver/driver/JDBCDriver.js
@@ -36,7 +36,26 @@ const DbTypes = {
       "artifactId": "mysql-connector-java",
       "version": "8.0.13"
     },
+    properties: {
+      user: process.env.CUBEJS_DB_USER,
+      password: process.env.CUBEJS_DB_PASS,
+    },
     jdbcUrl: () => `jdbc:mysql://${process.env.CUBEJS_DB_HOST}:3306/${process.env.CUBEJS_DB_NAME}`
+  },
+  athena: {
+    driverClass: "com.qubole.jdbc.jdbc41.core.QDriver",
+    prepareConnectionQueries: [],
+    mavenDependency: {
+      "groupId": "com.syncron.amazonaws",
+      "artifactId": "simba-athena-jdbc-driver",
+      "version": "2.0.2"
+    },
+    jdbcUrl: () => `jdbc:awsathena://AwsRegion=${process.env.CUBEJS_AWS_REGION}`,
+    properties: {
+      UID: process.env.CUBEJS_AWS_KEY,
+      PWD: process.env.CUBEJS_AWS_SECRET,
+      S3OutputLocation: process.env.CUBEJS_AWS_S3_OUTPUT_LOCATION
+    }
   }
 };
 
@@ -50,10 +69,7 @@ class JDBCDriver extends BaseDriver {
       dbType: process.env.CUBEJS_DB_TYPE,
       url: process.env.CUBEJS_JDBC_URL || dbTypeDescription && dbTypeDescription.jdbcUrl(),
       drivername: process.env.CUBEJS_JDBC_DRIVER || dbTypeDescription && dbTypeDescription.driverClass,
-      properties: {
-        user: process.env.CUBEJS_DB_USER,
-        password: process.env.CUBEJS_DB_PASS
-      },
+      properties: dbTypeDescription && dbTypeDescription.properties,
       ...config
     };