Use a WeakHashMap in WorldAttached to prevent memory leaks

This commit is contained in:
KJP12 2022-11-27 08:05:06 -06:00
parent 450359b212
commit d9354263a7
No known key found for this signature in database
GPG key ID: 87562A9A10CAB22F

View file

@ -2,9 +2,9 @@ package com.simibubi.create.foundation.utility;
import java.lang.ref.WeakReference; import java.lang.ref.WeakReference;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.HashMap;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.WeakHashMap;
import java.util.function.BiConsumer; import java.util.function.BiConsumer;
import java.util.function.Consumer; import java.util.function.Consumer;
import java.util.function.Function; import java.util.function.Function;
@ -22,7 +22,9 @@ public class WorldAttached<T> {
public WorldAttached(Function<LevelAccessor, T> factory) { public WorldAttached(Function<LevelAccessor, T> factory) {
this.factory = factory; this.factory = factory;
attached = new HashMap<>(); // Weak key hashmaps prevent worlds not existing anywhere else from leaking memory.
// This is only a fallback in the event that unload events fail to fire for any reason.
attached = new WeakHashMap<>();
allMaps.add(new WeakReference<>(attached)); allMaps.add(new WeakReference<>(attached));
} }