test commit apache#2

JayajP · Dec 14, 2023 · d50be23 · d50be23
1 parent 08f8840
commit d50be23
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/.../src/main/java/org/apache/beam/runners/dataflow/worker/StreamingStepMetricsContainer.java b/.../src/main/java/org/apache/beam/runners/dataflow/worker/StreamingStepMetricsContainer.java
@@ -56,8 +56,7 @@ public class StreamingStepMetricsContainer implements MetricsContainer {
   private MetricsMap<MetricName, DeltaCounterCell> counters =
       new MetricsMap<>(DeltaCounterCell::new);
 
-  private MetricsMap<MetricName, DeltaCounterCell> perWorkerCounters =
-      new MetricsMap<>(DeltaCounterCell::new);
+  private ConcurrentHashMap<MetricName, Long> perWorkerCounters = new ConcurrentHashMap<>();
 
   private MetricsMap<MetricName, GaugeCell> gauges = new MetricsMap<>(GaugeCell::new);
 
@@ -88,7 +87,7 @@ public Counter getCounter(MetricName metricName) {
   @Override
   public Counter getPerWorkerCounter(MetricName metricName) {
     if (enablePerWorkerMetrics) {
-      return perWorkerCounters.get(metricName);
+      return new RemoveSafeDeltaCounterCell(metricName, perWorkerCounters);
     } else {
       return MetricsContainer.super.getPerWorkerCounter(metricName);
     }