perf: handle some concurrency in ModelSet (#3736)

INRIA · Dec 22, 2020 · 3435fac · 3435fac
1 parent b6ec8a3
commit 3435fac
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/spoon/support/util/ModelSet.java b/src/main/java/spoon/support/util/ModelSet.java
@@ -10,6 +10,7 @@
 import java.io.Serializable;
 import java.util.AbstractSet;
 import java.util.Collection;
+import java.util.Collections;
 import java.util.Comparator;
 import java.util.Iterator;
 import java.util.LinkedHashSet;
@@ -35,7 +36,7 @@ public abstract class ModelSet<T extends CtElement> extends AbstractSet<T> imple
 	private final Set<T> set;
 
 	protected ModelSet(Comparator<? super CtElement> comparator) {
-		set = new TreeSet<>(comparator);
+		set = Collections.synchronizedSet(new TreeSet<>(comparator));
 	}
 
 	protected abstract CtElement getOwner();