fix registry to persist script order (#810)

KubeJS-Mods · May 20, 2024 · e293687 · e293687
1 parent d1a1111
commit e293687
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/common/src/main/java/dev/latvian/mods/kubejs/event/EventHandler.java b/common/src/main/java/dev/latvian/mods/kubejs/event/EventHandler.java
@@ -13,8 +13,8 @@
 import org.jetbrains.annotations.Nullable;
 
 import java.util.HashMap;
-import java.util.HashSet;
-import java.util.IdentityHashMap;
+import java.util.LinkedHashMap;
+import java.util.LinkedHashSet;
 import java.util.Map;
 import java.util.Set;
 import java.util.function.Consumer;
@@ -141,7 +141,7 @@ public void listen(ScriptType type, @Nullable Object extraId, IEventHandler hand
 			map = eventContainers;
 		} else {
 			if (extraEventContainers == null) {
-				extraEventContainers = extra.identity ? new IdentityHashMap<>() : new HashMap<>();
+				extraEventContainers = extra.identity ? new LinkedHashMap<>() : new HashMap<>();
 			}
 
 			map = extraEventContainers.get(extraId);
@@ -359,7 +359,7 @@ public Set<Object> findUniqueExtraIds(ScriptType type) {
 			return Set.of();
 		}
 
-		var set = new HashSet<>();
+		var set = new LinkedHashSet<>();
 
 		forEachListener(type, c -> {
 			if (c.extraId != null) {