add method recordUpdatedPositions to the OperatorContext class

trinodb · Jan 21, 2025 · 0f6ec97 · 0f6ec97
1 parent b45d6dc
commit 0f6ec97
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/core/trino-main/src/main/java/io/trino/operator/OperatorContext.java b/core/trino-main/src/main/java/io/trino/operator/OperatorContext.java
@@ -227,6 +227,12 @@ public void recordDynamicFilterSplitProcessed(long dynamicFilterSplits)
         dynamicFilterSplitsProcessed.getAndAdd(dynamicFilterSplits);
     }
 
+    public void recordUpdatedPositions(long updatedPositions)
+    {
+        checkArgument(updatedPositions >= 0, "updatedPositions is negative (%s)", updatedPositions);
+        this.updatedPositions.update(updatedPositions);
+    }
+
     /**
      * Overwrites the metrics with the latest one.
      *

diff --git a/core/trino-main/src/main/java/io/trino/operator/TableMutationOperator.java b/core/trino-main/src/main/java/io/trino/operator/TableMutationOperator.java
@@ -121,7 +121,7 @@ public Page getOutput()
 
         OptionalLong rowsUpdatedCount = operation.execute();
 
-        operatorContext.getUpdatedPositions().update(rowsUpdatedCount.orElse(0L));
+        operatorContext.recordUpdatedPositions(rowsUpdatedCount.orElse(0L));
 
         return buildUpdatedCountPage(rowsUpdatedCount);
     }