From 3c5d613a62234388802b45d07d0efd267c3cc0b4 Mon Sep 17 00:00:00 2001
From: Rohit Sinha <rosinha@google.com>
Date: Sun, 22 Dec 2024 13:17:37 -0800
Subject: [PATCH] Plumb BoundedTrie to StreamingStepMetricsContainer

---
 .../runners/dataflow/worker/BatchModeExecutionContext.java   | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/BatchModeExecutionContext.java b/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/BatchModeExecutionContext.java
index aeef7784c2c3..7cd0a18b83e5 100644
--- a/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/BatchModeExecutionContext.java
+++ b/runners/google-cloud-dataflow-java/worker/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())));
             });
   }