diff --git a/core/src/main/java/org/geysermc/floodgate/inject/CommonPlatformInjector.java b/core/src/main/java/org/geysermc/floodgate/inject/CommonPlatformInjector.java index 95e0a339..1547a2be 100644 --- a/core/src/main/java/org/geysermc/floodgate/inject/CommonPlatformInjector.java +++ b/core/src/main/java/org/geysermc/floodgate/inject/CommonPlatformInjector.java @@ -35,21 +35,21 @@ import org.geysermc.floodgate.api.inject.InjectorAddon; import org.geysermc.floodgate.api.inject.PlatformInjector; public abstract class CommonPlatformInjector implements PlatformInjector { - private final Map injectedClients = - Collections.synchronizedMap(new WeakHashMap<>()); + private final Set injectedClients = + Collections.synchronizedSet(Collections.newSetFromMap(new WeakHashMap<>())); private final Map, InjectorAddon> addons = new HashMap<>(); protected boolean addInjectedClient(Channel channel) { - return injectedClients.put(channel, null) != null; + return injectedClients.add(channel); } public boolean removeInjectedClient(Channel channel) { - return injectedClients.remove(channel) != null; + return injectedClients.remove(channel); } public Set injectedClients() { - return injectedClients.keySet(); + return injectedClients; } @Override