From c84dd0cae507b1fb0205a7f4d19dab403aaa7f8a Mon Sep 17 00:00:00 2001
From: Tim203 <mctim203@gmail.com>
Date: Tue, 27 Dec 2022 12:05:47 +0100
Subject: [PATCH] Use newSetFromMap

---
 .../floodgate/inject/CommonPlatformInjector.java       | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

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<Channel, ?> injectedClients =
-            Collections.synchronizedMap(new WeakHashMap<>());
+    private final Set<Channel> injectedClients =
+            Collections.synchronizedSet(Collections.newSetFromMap(new WeakHashMap<>()));
 
     private final Map<Class<?>, 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<Channel> injectedClients() {
-        return injectedClients.keySet();
+        return injectedClients;
     }
 
     @Override