telemetry latency models (#865)

telemetry latency models

Author: sreekanth-db

src/main/java/com/databricks/jdbc/model/telemetry/SqlExecutionEvent.java

@@ -2,6 +2,9 @@
 
 import com.databricks.jdbc.common.StatementType;
 import com.databricks.jdbc.model.telemetry.enums.ExecutionResultFormat;
+import com.databricks.jdbc.model.telemetry.latency.ChunkDetails;
+import com.databricks.jdbc.model.telemetry.latency.OperationDetail;
+import com.databricks.jdbc.model.telemetry.latency.ResultLatency;
 import com.databricks.sdk.support.ToStringer;
 import com.fasterxml.jackson.annotation.JsonProperty;
 
@@ -21,6 +24,15 @@ public class SqlExecutionEvent {
   @JsonProperty("retry_count")
   Integer retryCount;
 
+  @JsonProperty("chunk_details")
+  ChunkDetails chunkDetails;
+
+  @JsonProperty("result_latency")
+  ResultLatency resultLatency;
+
+  @JsonProperty("operation_detail")
+  OperationDetail operationDetail;
+
   public SqlExecutionEvent setDriverStatementType(StatementType driverStatementType) {
     this.driverStatementType = driverStatementType;
     return this;
@@ -46,6 +58,21 @@ public SqlExecutionEvent setRetryCount(Integer retryCount) {
     return this;
   }
 
+  public SqlExecutionEvent setChunkDetails(ChunkDetails chunkDetails) {
+    this.chunkDetails = chunkDetails;
+    return this;
+  }
+
+  public SqlExecutionEvent setResultLatency(ResultLatency resultLatency) {
+    this.resultLatency = resultLatency;
+    return this;
+  }
+
+  public SqlExecutionEvent setOperationDetails(OperationDetail operationDetail) {
+    this.operationDetail = operationDetail;
+    return this;
+  }
+
   @Override
   public String toString() {
     return new ToStringer(SqlExecutionEvent.class)
@@ -54,6 +81,9 @@ public String toString() {
         .add("execution_result", executionResultFormat)
         .add("chunk_id", chunkId)
         .add("retry_count", retryCount)
+        .add("chunk_details", chunkDetails)
+        .add("result_latency", resultLatency)
+        .add("operation_details", operationDetail)
         .toString();
   }
 }

src/main/java/com/databricks/jdbc/model/telemetry/latency/ChunkDetails.java

@@ -0,0 +1,58 @@
+package com.databricks.jdbc.model.telemetry.latency;
+
+import com.databricks.sdk.support.ToStringer;
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+public class ChunkDetails {
+
+  @JsonProperty("initial_chunk_latency_millis")
+  private Long initialChunkLatencyMillis;
+
+  @JsonProperty("slowest_chunk_latency_millis")
+  private Long slowestChunkLatencyMillis;
+
+  @JsonProperty("total_chunks_present")
+  private Integer totalChunksPresent;
+
+  @JsonProperty("total_chunks_iterated")
+  private Integer totalChunksIterated;
+
+  @JsonProperty("sum_chunks_download_time_millis")
+  private Long sumChunksDownloadTimeMillis;
+
+  public ChunkDetails setInitialChunkLatencyMillis(Long initialChunkLatencyMillis) {
+    this.initialChunkLatencyMillis = initialChunkLatencyMillis;
+    return this;
+  }
+
+  public ChunkDetails setSlowestChunkLatencyMillis(Long slowestChunkLatencyMillis) {
+    this.slowestChunkLatencyMillis = slowestChunkLatencyMillis;
+    return this;
+  }
+
+  public ChunkDetails setTotalChunksPresent(Integer totalChunksPresent) {
+    this.totalChunksPresent = totalChunksPresent;
+    return this;
+  }
+
+  public ChunkDetails setTotalChunksIterated(Integer totalChunksIterated) {
+    this.totalChunksIterated = totalChunksIterated;
+    return this;
+  }
+
+  public ChunkDetails setSumChunksDownloadTimeMillis(Long sumChunksDownloadTimeMillis) {
+    this.sumChunksDownloadTimeMillis = sumChunksDownloadTimeMillis;
+    return this;
+  }
+
+  @Override
+  public String toString() {
+    return new ToStringer(ChunkDetails.class)
+        .add("initialChunkLatencyMillis", initialChunkLatencyMillis)
+        .add("slowestChunkLatencyMillis", slowestChunkLatencyMillis)
+        .add("totalChunksPresent", totalChunksPresent)
+        .add("totalChunksIterated", totalChunksIterated)
+        .add("sumChunksDownloadTimeMillis", sumChunksDownloadTimeMillis)
+        .toString();
+  }
+}

src/main/java/com/databricks/jdbc/model/telemetry/latency/OperationDetail.java

@@ -0,0 +1,49 @@
+package com.databricks.jdbc.model.telemetry.latency;
+
+import com.databricks.sdk.support.ToStringer;
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+public class OperationDetail {
+
+  @JsonProperty("n_operation_status_calls")
+  private Integer nOperationStatusCalls;
+
+  @JsonProperty("operation_status_latency_millis")
+  private Long operationStatusLatencyMillis;
+
+  @JsonProperty("operation_type")
+  private OperationType operationType;
+
+  @JsonProperty("is_internal_call")
+  private Boolean isInternalCall;
+
+  public OperationDetail setNOperationStatusCalls(Integer nOperationStatusCalls) {
+    this.nOperationStatusCalls = nOperationStatusCalls;
+    return this;
+  }
+
+  public OperationDetail setOperationStatusLatencyMillis(Long operationStatusLatencyMillis) {
+    this.operationStatusLatencyMillis = operationStatusLatencyMillis;
+    return this;
+  }
+
+  public OperationDetail setOperationType(OperationType operationType) {
+    this.operationType = operationType;
+    return this;
+  }
+
+  public OperationDetail setInternalCall(Boolean isInternalCall) {
+    this.isInternalCall = isInternalCall;
+    return this;
+  }
+
+  @Override
+  public String toString() {
+    return new ToStringer(OperationDetail.class)
+        .add("nOperationStatusCalls", nOperationStatusCalls)
+        .add("operationLatencyMillis", operationStatusLatencyMillis)
+        .add("operationName", operationType)
+        .add("isInternalCall", isInternalCall)
+        .toString();
+  }
+}

src/main/java/com/databricks/jdbc/model/telemetry/latency/OperationType.java

@@ -0,0 +1,22 @@
+package com.databricks.jdbc.model.telemetry.latency;
+
+public enum OperationType {
+  TYPE_UNSPECIFIED,
+  CREATE_SESSION,
+  DELETE_SESSION,
+  EXECUTE_STATEMENT,
+  EXECUTE_STATEMENT_ASYNC,
+  CLOSE_STATEMENT,
+  CANCEL_STATEMENT,
+  LIST_TYPE_INFO,
+  LIST_CATALOGS,
+  LIST_SCHEMAS,
+  LIST_TABLES,
+  LIST_TABLE_TYPES,
+  LIST_COLUMNS,
+  LIST_FUNCTIONS,
+  LIST_PRIMARY_KEYS,
+  LIST_IMPORTED_KEYS,
+  LIST_EXPORTED_KEYS,
+  LIST_CROSS_REFERENCES
+}

src/main/java/com/databricks/jdbc/model/telemetry/latency/ResultLatency.java

@@ -0,0 +1,32 @@
+package com.databricks.jdbc.model.telemetry.latency;
+
+import com.databricks.sdk.support.ToStringer;
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+public class ResultLatency {
+
+  @JsonProperty("result_set_ready_latency_millis")
+  private Long resultSetReadyLatencyMillis;
+
+  @JsonProperty("result_set_consumption_latency_millis")
+  private Long resultSetConsumptionLatencyMillis;
+
+  public ResultLatency setResultSetReadyLatencyMillis(Long resultSetReadyLatencyMillis) {
+    this.resultSetReadyLatencyMillis = resultSetReadyLatencyMillis;
+    return this;
+  }
+
+  public ResultLatency setResultSetConsumptionLatencyMillis(
+      Long resultSetConsumptionLatencyMillis) {
+    this.resultSetConsumptionLatencyMillis = resultSetConsumptionLatencyMillis;
+    return this;
+  }
+
+  @Override
+  public String toString() {
+    return new ToStringer(ResultLatency.class)
+        .add("resultSetReadyLatencyMillis", resultSetReadyLatencyMillis)
+        .add("resultSetConsumptionLatencyMillis", resultSetConsumptionLatencyMillis)
+        .toString();
+  }
+}