Plumb BoundedTrie to StreamingStepMetricsContainer

apache · Dec 22, 2024 · 3c5d613 · 3c5d613
1 parent 4b0b070
commit 3c5d613
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/...rker/src/main/java/org/apache/beam/runners/dataflow/worker/BatchModeExecutionContext.java b/...rker/src/main/java/org/apache/beam/runners/dataflow/worker/BatchModeExecutionContext.java
@@ -519,6 +519,11 @@ public Iterable<CounterUpdate> extractMetricUpdates(boolean isFinalUpdate) {
                       .transform(
                           update ->
                               MetricsToCounterUpdateConverter.fromStringSet(
+                                  update.getKey(), update.getUpdate())),
+                  FluentIterable.from(updates.boundedTrieUpdates())
+                      .transform(
+                          update ->
+                              MetricsToCounterUpdateConverter.fromBoundedTrie(
                                   update.getKey(), update.getUpdate())));
             });
   }