Merge pull request #187 from databricks/madhav-db/params

Adds Parameters in Connection String to switch between Metadata Clients

Author: madhav-db

src/main/java/com/databricks/jdbc/core/DatabricksSession.java

@@ -4,6 +4,7 @@
 import com.databricks.jdbc.client.DatabricksClientType;
 import com.databricks.jdbc.client.DatabricksMetadataClient;
 import com.databricks.jdbc.client.impl.sdk.DatabricksMetadataSdkClient;
+import com.databricks.jdbc.client.impl.sdk.DatabricksNewMetadataSdkClient;
 import com.databricks.jdbc.client.impl.sdk.DatabricksSdkClient;
 import com.databricks.jdbc.client.impl.thrift.DatabricksThriftServiceClient;
 import com.databricks.jdbc.core.types.CompressionType;
@@ -52,8 +53,13 @@ public DatabricksSession(IDatabricksConnectionContext connectionContext)
       this.databricksMetadataClient = null;
     } else {
       this.databricksClient = new DatabricksSdkClient(connectionContext);
-      this.databricksMetadataClient =
-          new DatabricksMetadataSdkClient((DatabricksSdkClient) databricksClient);
+      if (connectionContext.getUseLegacyMetadata()) {
+        this.databricksMetadataClient =
+            new DatabricksMetadataSdkClient((DatabricksSdkClient) databricksClient);
+      } else {
+        this.databricksMetadataClient =
+            new DatabricksNewMetadataSdkClient((DatabricksSdkClient) databricksClient);
+      }
     }
     this.isSessionOpen = false;
     this.sessionInfo = null;

src/main/java/com/databricks/jdbc/driver/DatabricksConnectionContext.java

@@ -292,6 +292,13 @@ public DatabricksClientType getClientType() {
     return DatabricksClientType.SQL_EXEC;
   }
 
+  @Override
+  public Boolean getUseLegacyMetadata() {
+    // Defaults to use legacy metadata client
+    return getParameter(DatabricksJdbcConstants.USE_LEGACY_METADATA) == null
+        || getParameter(DatabricksJdbcConstants.USE_LEGACY_METADATA).equals("1");
+  }
+
   private static boolean nullOrEmptyString(String s) {
     return s == null || s.isEmpty();
   }

src/main/java/com/databricks/jdbc/driver/DatabricksJdbcConstants.java

@@ -130,4 +130,6 @@ public enum FakeServiceType {
   }
 
   public static final String USE_THRIFT_CLIENT = "usethriftclient";
+
+  public static final String USE_LEGACY_METADATA = "uselegacymetadata";
 }

src/main/java/com/databricks/jdbc/driver/IDatabricksConnectionContext.java

@@ -117,4 +117,6 @@ public static AuthMech parseAuthMech(String authMech) {
   int getAsyncExecPollInterval();
 
   DatabricksClientType getClientType();
+
+  Boolean getUseLegacyMetadata();
 }