Fix perf slowdown caused by AnyHolderSet#contains (#1775)

Co-authored-by: Brennan Ward <[email protected]>
neoforged · Dec 15, 2024 · f7a5bc8 · f7a5bc8
1 parent 32a1087
commit f7a5bc8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/net/neoforged/neoforge/registries/holdersets/AnyHolderSet.java b/src/main/java/net/neoforged/neoforge/registries/holdersets/AnyHolderSet.java
@@ -80,7 +80,8 @@ public Holder<T> get(int i) {
 
     @Override
     public boolean contains(Holder<T> holder) {
-        return holder.unwrapKey().map(key -> this.registryLookup.listElementIds().anyMatch(key::equals)).orElse(false);
+        ResourceKey<T> key = holder.getKey();
+        return key != null && this.registryLookup().get(key).isPresent();
     }
 
     @Override