From 42f248e074d35425fed0b5df4cf96d08bcbad53b Mon Sep 17 00:00:00 2001
From: shedaniel <daniel@shedaniel.me>
Date: Sun, 19 Jun 2022 02:49:23 +0800
Subject: Synchronize DisplayRegistry displays

---
 .../rei/impl/client/registry/display/DisplayRegistryImpl.java     | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'runtime')

diff --git a/runtime/src/main/java/me/shedaniel/rei/impl/client/registry/display/DisplayRegistryImpl.java b/runtime/src/main/java/me/shedaniel/rei/impl/client/registry/display/DisplayRegistryImpl.java
index 73bf3122c..eaa53a23d 100644
--- a/runtime/src/main/java/me/shedaniel/rei/impl/client/registry/display/DisplayRegistryImpl.java
+++ b/runtime/src/main/java/me/shedaniel/rei/impl/client/registry/display/DisplayRegistryImpl.java
@@ -56,7 +56,7 @@ import java.util.function.UnaryOperator;
 
 public class DisplayRegistryImpl extends RecipeManagerContextImpl<REIClientPlugin> implements DisplayRegistry {
     private final WeakHashMap<Display, Object> displaysBase = new WeakHashMap<>();
-    private final Map<CategoryIdentifier<?>, DisplaysList> displays = new ConcurrentHashMap<>();
+    private final Map<CategoryIdentifier<?>, List<Display>> displays = new ConcurrentHashMap<>();
     private final Map<CategoryIdentifier<?>, List<Display>> unmodifiableDisplays;
     private final Map<CategoryIdentifier<?>, List<DynamicDisplayGenerator<?>>> displayGenerators = new ConcurrentHashMap<>();
     private final List<DynamicDisplayGenerator<?>> globalDisplayGenerators = new ArrayList<>();
@@ -130,9 +130,11 @@ public class DisplayRegistryImpl extends RecipeManagerContextImpl<REIClientPlugi
     
     private static class DisplaysList extends ArrayList<Display> {
         private final List<Display> unmodifiableList;
+        private final List<Display> synchronizedList;
         
         public DisplaysList() {
-            this.unmodifiableList = Collections.unmodifiableList(this);
+            this.synchronizedList = Collections.synchronizedList(this);
+            this.unmodifiableList = Collections.unmodifiableList(synchronizedList);
         }
     }
     
@@ -159,7 +161,7 @@ public class DisplayRegistryImpl extends RecipeManagerContextImpl<REIClientPlugi
             }
         }
         
-        displays.computeIfAbsent(display.getCategoryIdentifier(), location -> new DisplaysList())
+        displays.computeIfAbsent(display.getCategoryIdentifier(), location -> new DisplaysList().synchronizedList)
                 .add(display);
         displayCount.increment();
         if (origin != null) {
-- 
cgit