Fix ConcurrentModificationException in federation fetcher

smallrye · Sep 16, 2024 · b0eb3f2 · b0eb3f2
1 parent eb2b6db
commit b0eb3f2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/server/implementation/src/main/java/io/smallrye/graphql/bootstrap/FederationDataFetcher.java b/server/implementation/src/main/java/io/smallrye/graphql/bootstrap/FederationDataFetcher.java
@@ -11,6 +11,7 @@
 import java.util.Objects;
 import java.util.Set;
 import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.ConcurrentHashMap;
 import java.util.function.BiFunction;
 import java.util.function.Function;
 import java.util.stream.Collectors;
@@ -39,7 +40,7 @@ public class FederationDataFetcher implements DataFetcher<CompletableFuture<List
     public static final String TYPENAME = "__typename";
     private final GraphQLObjectType queryType;
     private final GraphQLCodeRegistry codeRegistry;
-    private final HashMap<TypeAndArgumentNames, TypeFieldWrapper> cache = new HashMap<>();
+    private final ConcurrentHashMap<TypeAndArgumentNames, TypeFieldWrapper> cache = new ConcurrentHashMap<>();
 
     public FederationDataFetcher(GraphQLObjectType queryType, GraphQLCodeRegistry codeRegistry) {
         this.queryType = queryType;