Declare equals and hashCode in Socket classes for use in Java maps

src/main/java/io/gomint/jraknet/ClientSocket.java

@@ -18,6 +18,7 @@
 import java.net.InetSocketAddress;
 import java.net.SocketException;
 import java.nio.charset.StandardCharsets;
+import java.util.Objects;
 
 import static io.gomint.jraknet.RakNetConstraints.*;
 
@@ -373,4 +374,19 @@ private void handleUnconnectedPong( InetSocketAddress sender, PacketBuffer datag
         this.propagateEvent( new SocketEvent( SocketEvent.Type.UNCONNECTED_PONG, info ) );
     }
 
-}
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        if (!super.equals(o)) return false;
+        ClientSocket that = (ClientSocket) o;
+        return mojangModificationEnabled == that.mojangModificationEnabled &&
+                Objects.equals(logger, that.logger) &&
+                Objects.equals(connection, that.connection);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(super.hashCode(), logger, connection, mojangModificationEnabled);
+    }
+}

src/main/java/io/gomint/jraknet/ServerSocket.java

@@ -18,10 +18,7 @@
 import java.net.SocketAddress;
 import java.net.SocketException;
 import java.nio.charset.StandardCharsets;
-import java.util.HashSet;
-import java.util.Iterator;
-import java.util.Map;
-import java.util.Set;
+import java.util.*;
 import java.util.concurrent.ConcurrentHashMap;
 
 import static io.gomint.jraknet.RakNetConstraints.*;
@@ -448,4 +445,19 @@ private ServerConnection getConnection( InetSocketAddress sender, PacketBuffer d
 		return connection;
 	}
 
+	@Override
+	public boolean equals(Object o) {
+		if (this == o) return true;
+		if (o == null || getClass() != o.getClass()) return false;
+		if (!super.equals(o)) return false;
+		ServerSocket that = (ServerSocket) o;
+		return maxConnections == that.maxConnections &&
+				mojangModificationEnabled == that.mojangModificationEnabled &&
+				Objects.equals(bindAddress, that.bindAddress);
+	}
+
+	@Override
+	public int hashCode() {
+		return Objects.hash(super.hashCode(), bindAddress, maxConnections, mojangModificationEnabled);
+	}
 }

src/main/java/io/gomint/jraknet/Socket.java

@@ -13,6 +13,7 @@
 import java.nio.channels.SelectionKey;
 import java.nio.channels.Selector;
 import java.util.Arrays;
+import java.util.Objects;
 import java.util.Random;
 import java.util.Set;
 import java.util.concurrent.ThreadFactory;
@@ -260,4 +261,16 @@ private void update() {
         this.cleanupUpdateThread();
     }
 
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        Socket socket = (Socket) o;
+        return guid == socket.guid;
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(guid);
+    }
 }