(beam) branch master updated: Use HashMap.computeIfAbsent instead of get/put (#36990)

[scwhittle]

This is an automated email from the ASF dual-hosted git repository.

scwhittle pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/master by this push:
     new 9bcae0d5e03 Use HashMap.computeIfAbsent instead of get/put (#36990)
9bcae0d5e03 is described below

commit 9bcae0d5e034e4f2fc09809d5f30f2b579c17ef6
Author: Sam Whittle <scwhit...@users.noreply.github.com>
AuthorDate: Thu Dec 4 20:58:07 2025 +0100

    Use HashMap.computeIfAbsent instead of get/put (#36990)
---
 .../apache/beam/runners/dataflow/worker/WindmillSink.java  | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git 
a/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/WindmillSink.java
 
b/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/WindmillSink.java
index d54d94f47d7..5cb3cb56d9e 100644
--- 
a/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/WindmillSink.java
+++ 
b/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/WindmillSink.java
@@ -264,11 +264,15 @@ class WindmillSink<T> extends Sink<WindowedValue<T>> {
         }
       }
 
-      Windmill.KeyedMessageBundle.Builder keyedOutput = productionMap.get(key);
-      if (keyedOutput == null) {
-        keyedOutput = Windmill.KeyedMessageBundle.newBuilder().setKey(key);
-        productionMap.put(key, keyedOutput);
-      }
+      Windmill.KeyedMessageBundle.Builder keyedOutput =
+          productionMap.computeIfAbsent(
+              key,
+              (k) -> {
+                Windmill.KeyedMessageBundle.Builder builder =
+                    Windmill.KeyedMessageBundle.newBuilder();
+                builder.setKey(k);
+                return builder;
+              });
 
       try {
         messageBuilder