From 0fc0d4350049657e7068852a2217a431bfe63957 Mon Sep 17 00:00:00 2001
From: "google-labs-jules[bot]"
 <161369871+google-labs-jules[bot]@users.noreply.github.com>
Date: Sat, 27 Dec 2025 04:07:23 +0000
Subject: [PATCH] feat: Parallelize source JAR lookup

Parallelize the source JAR lookup process to improve performance.
Switched to `parallelStream` for processing dependencies and updated `GraphMemoizator` and `SourcesJarLocator` to use `ConcurrentHashMap` for thread safety.

[LLM] Bolt
---
 .../evendanan/bazel/mvn/merger/GraphMemoizator.java  | 12 +++---------
 .../bazel/mvn/merger/SourcesJarLocator.java          |  6 +++---
 2 files changed, 6 insertions(+), 12 deletions(-)

diff --git a/resolver/src/main/java/net/evendanan/bazel/mvn/merger/GraphMemoizator.java b/resolver/src/main/java/net/evendanan/bazel/mvn/merger/GraphMemoizator.java
index eaa0a21..b3f5e4f 100644
--- a/resolver/src/main/java/net/evendanan/bazel/mvn/merger/GraphMemoizator.java
+++ b/resolver/src/main/java/net/evendanan/bazel/mvn/merger/GraphMemoizator.java
@@ -1,11 +1,11 @@
 package net.evendanan.bazel.mvn.merger;
 
-import java.util.HashMap;
 import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
 import javax.annotation.Nonnull;
 
 public abstract class GraphMemoizator<T> {
-  private final Map<String, T> cache = new HashMap<>();
+  private final Map<String, T> cache = new ConcurrentHashMap<>();
 
   @Nonnull
   protected abstract T calculate(@Nonnull T original);
@@ -13,13 +13,7 @@ public abstract class GraphMemoizator<T> {
   @Nonnull
   public T map(@Nonnull T original) {
     final String key = getKeyForObject(original);
-    if (cache.containsKey(key)) {
-      return cache.get(key);
-    } else {
-      final T revised = calculate(original);
-      cache.put(key, revised);
-      return revised;
-    }
+    return cache.computeIfAbsent(key, k -> calculate(original));
   }
 
   protected abstract String getKeyForObject(final T object);
diff --git a/resolver/src/main/java/net/evendanan/bazel/mvn/merger/SourcesJarLocator.java b/resolver/src/main/java/net/evendanan/bazel/mvn/merger/SourcesJarLocator.java
index 035ec1b..f86a27a 100644
--- a/resolver/src/main/java/net/evendanan/bazel/mvn/merger/SourcesJarLocator.java
+++ b/resolver/src/main/java/net/evendanan/bazel/mvn/merger/SourcesJarLocator.java
@@ -5,9 +5,9 @@
 import java.net.HttpURLConnection;
 import java.net.URL;
 import java.util.Collection;
-import java.util.HashMap;
 import java.util.Locale;
 import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
 import java.util.stream.Collectors;
 import javax.annotation.Nonnull;
 import net.evendanan.bazel.mvn.api.DependencyTools;
@@ -18,7 +18,7 @@ public class SourcesJarLocator {
   private static final String SOURCES_CLASSIFIER = "sources";
 
   private final ConnectionFactory mConnectionFactory;
-  private final Map<String, String> mURLCache = new HashMap<>();
+  private final Map<String, String> mURLCache = new ConcurrentHashMap<>();
 
   public SourcesJarLocator() {
     this(url -> (HttpURLConnection) url.openConnection());
@@ -31,7 +31,7 @@ public SourcesJarLocator() {
 
   private static Collection<Dependency> fillSourcesAttribute(
       Collection<Dependency> dependencies, DependencyMemoizator memoizator) {
-    return dependencies.stream().map(memoizator::map).collect(Collectors.toList());
+    return dependencies.parallelStream().map(memoizator::map).collect(Collectors.toList());
   }
 
   public Collection<Dependency> fillSourcesAttribute(Collection<Dependency> dependencies) {