Update Trino to v416 (#31)

regadas · May 7, 2023 · a2ede20 · a2ede20
1 parent b74b290
commit a2ede20
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 2 deletions.
diff --git a/plugin/build.gradle b/plugin/build.gradle
@@ -14,7 +14,7 @@ dependencies {
     compileOnly         "com.google.auto.value:auto-value-annotations:${autoValueVersion}"
     annotationProcessor "com.google.auto.value:auto-value:${autoValueVersion}"
 
-    def trinoVersion = '414'
+    def trinoVersion = '416'
     api "io.trino:trino-spi:${trinoVersion}"
 
     implementation platform('com.google.cloud:libraries-bom:26.14.0')

diff --git a/plugin/src/main/java/dev/regadas/trino/pubsub/listener/SchemaHelpers.java b/plugin/src/main/java/dev/regadas/trino/pubsub/listener/SchemaHelpers.java
@@ -229,6 +229,9 @@ static final Schema.QueryCompletedEvent from(QueryCompletedEvent event) {
                 .ifPresent(statsBuilder::setResourceWaitingTime);
         stats.getAnalysisTime().map(SchemaHelpers::from).ifPresent(statsBuilder::setAnalysisTime);
         stats.getPlanningTime().map(SchemaHelpers::from).ifPresent(statsBuilder::setPlanningTime);
+        stats.getPlanningCpuTime()
+                .map(SchemaHelpers::from)
+                .ifPresent(statsBuilder::setPlanningCpuTime);
         stats.getExecutionTime().map(SchemaHelpers::from).ifPresent(statsBuilder::setExecutionTime);
         stats.getInputBlockedTime()
                 .map(SchemaHelpers::from)

diff --git a/plugin/src/main/proto/schema.proto b/plugin/src/main/proto/schema.proto
@@ -175,6 +175,7 @@ message QueryStatistics {
   repeated CpuTimeDistribution cpu_time_distribution = 36;
   repeated OutputBufferUtilization output_buffer_utilization = 37;
   repeated string operator_summaries = 38;
+  Duration planning_cpu_time = 39;
 }
 
 message StageGcStatistics {
@@ -274,4 +275,4 @@ message SplitStatistics {
 message SplitFailureInfo {
   string failure_type = 1;
   string failure_message = 2;
-}
+}
diff --git a/plugin/src/test/java/dev/regadas/trino/pubsub/listener/TestData.java b/plugin/src/test/java/dev/regadas/trino/pubsub/listener/TestData.java
@@ -60,6 +60,7 @@ class TestData {
                     Optional.of(ofMillis(107)),
                     Optional.of(ofMillis(108)),
                     Optional.of(ofMillis(109)),
+                    Optional.of(ofMillis(1091)),
                     Optional.of(ofMillis(110)),
                     Optional.of(ofMillis(111)),
                     Optional.of(ofMillis(112)),
@@ -213,6 +214,7 @@ class TestData {
                     Optional.empty(),
                     Optional.empty(),
                     Optional.empty(),
+                    Optional.empty(),
                     115L,
                     116L,
                     117L,