Add metadata URL (#385)

Author: samikshya-db

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

@@ -49,8 +49,7 @@ public boolean allTablesAreSelectable() throws SQLException {
   @Override
   public String getURL() throws SQLException {
     LoggingUtil.log(LogLevel.DEBUG, "public String getURL()");
-    throw new UnsupportedOperationException(
-        "Not implemented in DatabricksDatabaseMetaData - getURL()");
+    return this.session.getConnectionContext().getConnectionURL();
   }
 
   @Override

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

@@ -24,6 +24,7 @@ public class DatabricksConnectionContext implements IDatabricksConnectionContext
   private final String host;
   @VisibleForTesting final int port;
   private final String schema;
+  private final String connectionURL;
   private final ComputeResource computeResource;
   private static DatabricksMetrics metricsExporter;
   @VisibleForTesting final ImmutableMap<String, String> parameters;
@@ -76,7 +77,8 @@ public static IDatabricksConnectionContext parse(String url, Properties properti
         parametersBuilder.put(entry.getKey().toString().toLowerCase(), entry.getValue().toString());
       }
       DatabricksConnectionContext context =
-          new DatabricksConnectionContext(hostValue, portValue, schema, parametersBuilder.build());
+          new DatabricksConnectionContext(
+              url, hostValue, portValue, schema, parametersBuilder.build());
       metricsExporter = new DatabricksMetrics(context);
       return context;
     } else {
@@ -106,8 +108,13 @@ private static void handleInvalidUrl(String url) throws DatabricksParsingExcepti
   }
 
   private DatabricksConnectionContext(
-      String host, int port, String schema, ImmutableMap<String, String> parameters)
+      String connectionURL,
+      String host,
+      int port,
+      String schema,
+      ImmutableMap<String, String> parameters)
       throws DatabricksSQLException {
+    this.connectionURL = connectionURL;
     this.host = host;
     this.port = port;
     this.schema = schema;
@@ -552,4 +559,9 @@ public boolean isFakeServiceTest() {
   public boolean enableTelemetry() {
     return Objects.equals(getParameter(ENABLE_TELEMETRY, "0"), "1");
   }
+
+  @Override
+  public String getConnectionURL() {
+    return connectionURL;
+  }
 }

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

@@ -150,4 +150,6 @@ public static AuthMech parseAuthMech(String authMech) {
   boolean isFakeServiceTest();
 
   boolean enableTelemetry();
+
+  String getConnectionURL();
 }

src/test/java/com/databricks/jdbc/core/DatabricksDatabaseMetaDataTest.java

@@ -1,13 +1,16 @@
 package com.databricks.jdbc.core;
 
+import static com.databricks.jdbc.TestConstants.WAREHOUSE_JDBC_URL;
 import static org.junit.jupiter.api.Assertions.*;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.Mockito.when;
 
 import com.databricks.jdbc.client.DatabricksMetadataClient;
+import com.databricks.jdbc.driver.DatabricksConnectionContext;
 import java.sql.*;
 import java.util.Arrays;
 import java.util.List;
+import java.util.Properties;
 import java.util.concurrent.Callable;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
@@ -38,6 +41,8 @@ public void setup() throws SQLException {
         .thenReturn(Mockito.mock(DatabricksResultSet.class));
     when(metadataClient.listSchemas(any(), any(), any()))
         .thenReturn(Mockito.mock(DatabricksResultSet.class));
+    when(session.getConnectionContext())
+        .thenReturn(DatabricksConnectionContext.parse(WAREHOUSE_JDBC_URL, new Properties()));
     when(metadataClient.listCatalogs(any())).thenReturn(Mockito.mock(DatabricksResultSet.class));
     when(metadataClient.listTableTypes(any())).thenReturn(Mockito.mock(DatabricksResultSet.class));
     when(metadataClient.listTypeInfo(any())).thenReturn(Mockito.mock(DatabricksResultSet.class));
@@ -879,6 +884,11 @@ public void testSupportsAlterTableWithAddColumn() throws SQLException {
     assertFalse(result);
   }
 
+  @Test
+  public void testGetUrl() throws SQLException {
+    assertEquals(metaData.getURL(), WAREHOUSE_JDBC_URL);
+  }
+
   @Test
   public void testSupportsAlterTableWithDropColumn() throws SQLException {
     boolean result = metaData.supportsAlterTableWithDropColumn();
@@ -973,7 +983,6 @@ public void testGetResultSetHoldability() throws SQLException {
   public void testUnsupportedOperations() {
     List<Callable<Object>> tasks =
         Arrays.asList(
-            () -> metaData.getURL(),
             () -> metaData.supportsTransactionIsolationLevel(0),
             () -> metaData.getDriverMajorVersion(),
             () -> metaData.getDriverMinorVersion(),

src/test/java/com/databricks/jdbc/driver/DatabricksConnectionContextTest.java

@@ -116,6 +116,7 @@ public void testParseValid() throws DatabricksSQLException {
         (DatabricksConnectionContext) DatabricksConnectionContext.parse(VALID_URL_1, properties);
     assertEquals(
         "https://adb-565757575.18.azuredatabricks.net:4423", connectionContext.getHostUrl());
+    assertEquals(VALID_URL_1, connectionContext.getConnectionURL());
     assertEquals("/sql/1.0/warehouses/erg6767gg", connectionContext.getHttpPath());
     assertEquals("passwd", connectionContext.getToken());
     assertEquals(