Make PipelineStatus a record class

Author: pettyjamesm

core/trino-main/src/main/java/io/trino/execution/SqlTask.java

@@ -363,10 +363,10 @@ else if (taskHolder.getTaskExecution() != null) {
             TaskContext taskContext = taskHolder.getTaskExecution().getTaskContext();
             for (PipelineContext pipelineContext : taskContext.getPipelineContexts()) {
                 PipelineStatus pipelineStatus = pipelineContext.getPipelineStatus();
-                queuedPartitionedDrivers += pipelineStatus.getQueuedPartitionedDrivers();
-                queuedPartitionedSplitsWeight += pipelineStatus.getQueuedPartitionedSplitsWeight();
-                runningPartitionedDrivers += pipelineStatus.getRunningPartitionedDrivers();
-                runningPartitionedSplitsWeight += pipelineStatus.getRunningPartitionedSplitsWeight();
+                queuedPartitionedDrivers += pipelineStatus.queuedPartitionedDrivers();
+                queuedPartitionedSplitsWeight += pipelineStatus.queuedPartitionedSplitsWeight();
+                runningPartitionedDrivers += pipelineStatus.runningPartitionedDrivers();
+                runningPartitionedSplitsWeight += pipelineStatus.runningPartitionedSplitsWeight();
                 physicalWrittenBytes += pipelineContext.getPhysicalWrittenDataSize();
             }
             writerInputDataSize = succinctBytes(taskContext.getWriterInputDataSize());

core/trino-main/src/main/java/io/trino/operator/PipelineContext.java

@@ -509,13 +509,13 @@ else if (pipelineLevelMetrics != Metrics.EMPTY) {
                 outputPipeline,
 
                 totalDrivers,
-                pipelineStatus.getQueuedDrivers(),
-                pipelineStatus.getQueuedPartitionedDrivers(),
-                pipelineStatus.getQueuedPartitionedSplitsWeight(),
-                pipelineStatus.getRunningDrivers(),
-                pipelineStatus.getRunningPartitionedDrivers(),
-                pipelineStatus.getRunningPartitionedSplitsWeight(),
-                pipelineStatus.getBlockedDrivers(),
+                pipelineStatus.queuedDrivers(),
+                pipelineStatus.queuedPartitionedDrivers(),
+                pipelineStatus.queuedPartitionedSplitsWeight(),
+                pipelineStatus.runningDrivers(),
+                pipelineStatus.runningPartitionedDrivers(),
+                pipelineStatus.runningPartitionedSplitsWeight(),
+                pipelineStatus.blockedDrivers(),
                 completedDrivers,
 
                 succinctBytes(pipelineMemoryContext.getUserMemory()),

core/trino-main/src/main/java/io/trino/operator/PipelineStatus.java

@@ -13,62 +13,13 @@
  */
 package io.trino.operator;
 
-import com.google.errorprone.annotations.Immutable;
-
-@Immutable
-public final class PipelineStatus
+public record PipelineStatus(
+        int queuedDrivers,
+        int runningDrivers,
+        int blockedDrivers,
+        int queuedPartitionedDrivers,
+        long queuedPartitionedSplitsWeight,
+        int runningPartitionedDrivers,
+        long runningPartitionedSplitsWeight)
 {
-    private final int queuedDrivers;
-    private final int runningDrivers;
-    private final int blockedDrivers;
-    private final int queuedPartitionedDrivers;
-    private final long queuedPartitionedSplitsWeight;
-    private final int runningPartitionedDrivers;
-    private final long runningPartitionedSplitsWeight;
-
-    public PipelineStatus(int queuedDrivers, int runningDrivers, int blockedDrivers, int queuedPartitionedDrivers, long queuedPartitionedSplitsWeight, int runningPartitionedDrivers, long runningPartitionedSplitsWeight)
-    {
-        this.queuedDrivers = queuedDrivers;
-        this.runningDrivers = runningDrivers;
-        this.blockedDrivers = blockedDrivers;
-        this.queuedPartitionedDrivers = queuedPartitionedDrivers;
-        this.queuedPartitionedSplitsWeight = queuedPartitionedSplitsWeight;
-        this.runningPartitionedDrivers = runningPartitionedDrivers;
-        this.runningPartitionedSplitsWeight = runningPartitionedSplitsWeight;
-    }
-
-    public int getQueuedDrivers()
-    {
-        return queuedDrivers;
-    }
-
-    public int getRunningDrivers()
-    {
-        return runningDrivers;
-    }
-
-    public int getBlockedDrivers()
-    {
-        return blockedDrivers;
-    }
-
-    public int getQueuedPartitionedDrivers()
-    {
-        return queuedPartitionedDrivers;
-    }
-
-    public long getQueuedPartitionedSplitsWeight()
-    {
-        return queuedPartitionedSplitsWeight;
-    }
-
-    public int getRunningPartitionedDrivers()
-    {
-        return runningPartitionedDrivers;
-    }
-
-    public long getRunningPartitionedSplitsWeight()
-    {
-        return runningPartitionedSplitsWeight;
-    }
 }