param to use new metadata client

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,12 @@ 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,12 @@ public DatabricksClientType getClientType() {
     return DatabricksClientType.SQL_EXEC;
   }
 
+  @Override
+  public Boolean getUseLegacyMetadata() {
+    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();
 }