fix using proto server as key in map

MrNavaStar · Aug 13, 2024 · e1901f7 · e1901f7
1 parent 306288a
commit e1901f7
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/proxy/src/main/java/me/mrnavastar/protoweaver/proxy/api/ProtoProxy.java b/proxy/src/main/java/me/mrnavastar/protoweaver/proxy/api/ProtoProxy.java
@@ -83,6 +83,10 @@ public void unregister(ProtoServer server) {
         Optional.ofNullable(servers.remove(server)).ifPresent(clients -> clients.forEach(ProtoClient::disconnect));
     }
 
+    /**
+     * Returns a list of servers connected on the supplied {@link Protocol}.
+     * @param protocol the protocol to check for.
+     */
     public static ArrayList<ProtoServer> getConnectedServers(@NonNull Protocol protocol) {
         ArrayList<ProtoServer> connected = new ArrayList<>();
         servers.forEach((server, clients) -> clients.stream()

diff --git a/proxy/src/main/java/me/mrnavastar/protoweaver/proxy/api/ProtoServer.java b/proxy/src/main/java/me/mrnavastar/protoweaver/proxy/api/ProtoServer.java
@@ -5,6 +5,7 @@
 import me.mrnavastar.protoweaver.api.netty.ProtoConnection;
 
 import java.net.SocketAddress;
+import java.util.Objects;
 
 @Setter
 @Getter
@@ -19,6 +20,16 @@ public ProtoServer(String name, SocketAddress address) {
         this.address = address;
     }
 
+    @Override
+    public boolean equals(Object obj) {
+        return obj instanceof ProtoServer server && Objects.equals(server.name, name);
+    }
+
+    @Override
+    public int hashCode() {
+        return name.hashCode();
+    }
+
     @Override
     public String toString() {
         return name + " : " + address;