diff --git a/src/main/java/net/raphimc/viaproxy/proxy/util/TransferDataHolder.java b/src/main/java/net/raphimc/viaproxy/proxy/util/TransferDataHolder.java index 7b75b71..5638490 100644 --- a/src/main/java/net/raphimc/viaproxy/proxy/util/TransferDataHolder.java +++ b/src/main/java/net/raphimc/viaproxy/proxy/util/TransferDataHolder.java @@ -17,18 +17,19 @@ */ package net.raphimc.viaproxy.proxy.util; +import com.google.common.cache.CacheBuilder; import com.viaversion.viabackwards.protocol.v1_20_5to1_20_3.storage.CookieStorage; import io.netty.channel.Channel; import java.net.InetAddress; import java.net.InetSocketAddress; import java.util.Map; -import java.util.concurrent.ConcurrentHashMap; +import java.util.concurrent.TimeUnit; public class TransferDataHolder { - private static final Map TEMP_REDIRECTS = new ConcurrentHashMap<>(); - private static final Map COOKIE_STORAGES = new ConcurrentHashMap<>(); + private static final Map TEMP_REDIRECTS = CacheBuilder.newBuilder().expireAfterWrite(1, TimeUnit.MINUTES).build().asMap(); + private static final Map COOKIE_STORAGES = CacheBuilder.newBuilder().expireAfterWrite(1, TimeUnit.MINUTES).build().asMap(); public static void addTempRedirect(final Channel channel, final InetSocketAddress redirect) { TEMP_REDIRECTS.put(getChannelAddress(channel), redirect);