Statement Latency

Author: samikshya-db

src/main/java/com/databricks/jdbc/dbclient/IDatabricksClient.java

@@ -28,6 +28,7 @@ public interface IDatabricksClient {
    * @param sessionConf session configuration
    * @return created session
    */
+  @DatabricksMetricsTimed
   ImmutableSessionInfo createSession(
       IDatabricksComputeResource computeResource,
       String catalog,
@@ -73,6 +74,7 @@ DatabricksResultSet executeStatement(
    * @param parentStatement statement instance if called from a statement
    * @return response for statement execution
    */
+  @DatabricksMetricsTimed
   DatabricksResultSet executeStatementAsync(
       String sql,
       IDatabricksComputeResource computeResource,
@@ -86,13 +88,15 @@ DatabricksResultSet executeStatementAsync(
    *
    * @param statementId statement which should be closed
    */
+  @DatabricksMetricsTimed
   void closeStatement(StatementId statementId) throws DatabricksSQLException;
 
   /**
    * Cancels a statement in Databricks server
    *
    * @param statementId statement which should be aborted
    */
+  @DatabricksMetricsTimed
   void cancelStatement(StatementId statementId) throws DatabricksSQLException;
 
   /**

src/main/java/com/databricks/jdbc/dbclient/IDatabricksMetadataClient.java

@@ -2,14 +2,17 @@
 
 import com.databricks.jdbc.api.impl.DatabricksResultSet;
 import com.databricks.jdbc.api.internal.IDatabricksSession;
+import com.databricks.jdbc.telemetry.latency.DatabricksMetricsTimed;
 import java.sql.SQLException;
 
 public interface IDatabricksMetadataClient {
 
   /** Returns information about types supported by Databricks server */
+  @DatabricksMetricsTimed
   DatabricksResultSet listTypeInfo(IDatabricksSession session) throws SQLException;
 
   /** Returns the list of catalogs */
+  @DatabricksMetricsTimed
   DatabricksResultSet listCatalogs(IDatabricksSession session) throws SQLException;
 
   /**
@@ -21,6 +24,7 @@ public interface IDatabricksMetadataClient {
    *     absolute name)
    * @return a DatabricksResultSet representing list of schemas
    */
+  @DatabricksMetricsTimed
   DatabricksResultSet listSchemas(
       IDatabricksSession session, String catalog, String schemaNamePattern) throws SQLException;
 
@@ -35,6 +39,7 @@ DatabricksResultSet listSchemas(
    *     absolute name)
    * @return a DatabricksResultSet representing list of tables
    */
+  @DatabricksMetricsTimed
   DatabricksResultSet listTables(
       IDatabricksSession session,
       String catalog,
@@ -44,6 +49,7 @@ DatabricksResultSet listTables(
       throws SQLException;
 
   /** Returns list of table types */
+  @DatabricksMetricsTimed
   DatabricksResultSet listTableTypes(IDatabricksSession session) throws SQLException;
 
   /**
@@ -59,6 +65,7 @@ DatabricksResultSet listTables(
    *     absolute name)
    * @return a DatabricksResultSet representing list of columns
    */
+  @DatabricksMetricsTimed
   DatabricksResultSet listColumns(
       IDatabricksSession session,
       String catalog,
@@ -78,6 +85,7 @@ DatabricksResultSet listColumns(
    *     absolute name)
    * @return a DatabricksResultSet representing list of functions
    */
+  @DatabricksMetricsTimed
   DatabricksResultSet listFunctions(
       IDatabricksSession session,
       String catalog,
@@ -94,6 +102,7 @@ DatabricksResultSet listFunctions(
    * @param table must match to a table in database
    * @return a DatabricksResultSet representing list of functions
    */
+  @DatabricksMetricsTimed
   DatabricksResultSet listPrimaryKeys(
       IDatabricksSession session, String catalog, String schema, String table) throws SQLException;
 
@@ -106,6 +115,7 @@ DatabricksResultSet listPrimaryKeys(
    * @param table must match to a table in database
    * @return a DatabricksResultSet representing list of imported keys
    */
+  @DatabricksMetricsTimed
   DatabricksResultSet listImportedKeys(
       IDatabricksSession session, String catalog, String schema, String table) throws SQLException;
 
@@ -118,6 +128,7 @@ DatabricksResultSet listImportedKeys(
    * @param table must match to a table in database
    * @return a DatabricksResultSet representing list of imported keys
    */
+  @DatabricksMetricsTimed
   DatabricksResultSet listExportedKeys(
       IDatabricksSession session, String catalog, String schema, String table) throws SQLException;
 
@@ -133,6 +144,7 @@ DatabricksResultSet listExportedKeys(
    * @param foreignTable must match to a foreign table in database
    * @return a DatabricksResultSet representing list of cross references
    */
+  @DatabricksMetricsTimed
   DatabricksResultSet listCrossReferences(
       IDatabricksSession session,
       String parentCatalog,