[hotfix] Fix collectSizeStats for SegmentFileStateHandle (apache#24480)

Zakelly · Apr 7, 2024 · 0b7f0fd · 0b7f0fd
1 parent ea4e498
commit 0b7f0fd
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/...time/src/main/java/org/apache/flink/runtime/state/filemerging/SegmentFileStateHandle.java b/...time/src/main/java/org/apache/flink/runtime/state/filemerging/SegmentFileStateHandle.java
@@ -109,6 +109,11 @@ public long getStateSize() {
         return stateSize;
     }
 
+    @Override
+    public void collectSizeStats(StateObjectSizeStatsCollector collector) {
+        collector.add(StateObjectLocation.REMOTE, getStateSize());
+    }
+
     public CheckpointedStateScope getScope() {
         return scope;
     }