Connection: add proxy support for write operation + fix hashCode/equals.

Author: Robin Duda

core/main/java/com/codingchili/core/listener/transport/Connection.java

@@ -1,6 +1,7 @@
 package com.codingchili.core.listener.transport;
 
 import java.util.*;
+import java.util.function.BiConsumer;
 import java.util.function.Consumer;
 
 import com.codingchili.core.listener.Messageable;
@@ -15,8 +16,9 @@
 public class Connection implements Messageable {
     private final Map<String, String> properties = new HashMap<>();
     private final Map<String, Runnable> closeHandlers = new HashMap<>();
-    private final Consumer<Object> writer;
     private static final String ID = "id";
+    private final Consumer<Object> writer;
+    private BiConsumer<Consumer<Object>, Object> proxy;
 
     /**
      * Creates a new stateful connection that properly implements the ID method.
@@ -29,9 +31,24 @@ public Connection(Consumer<Object> writer, String id) {
         properties.put(ID, id);
     }
 
+    /**
+     * Allows all writing to this connection to be handled by a proxy.
+     *
+     * @param proxy the method that will invoke the writer.
+     * @return fluent.
+     */
+    public Connection proxy(BiConsumer<Consumer<Object>, Object> proxy) {
+        this.proxy = proxy;
+        return this;
+    }
+
     @Override
     public void write(Object object) {
-        writer.accept(object);
+        if (proxy == null) {
+            writer.accept(object);
+        } else {
+            proxy.accept(writer, object);
+        }
     }
 
     /**
@@ -113,4 +130,18 @@ public Connection setProperty(String key, String value) {
         properties.put(key, value);
         return this;
     }
+
+    @Override
+    public int hashCode() {
+        return id().hashCode();
+    }
+
+    @Override
+    public boolean equals(Object other) {
+        if (other instanceof Connection) {
+            return ((Connection) other).id().equals(id());
+        } else {
+            return false;
+        }
+    }
 }

core/main/java/com/codingchili/core/listener/transport/WebsocketListener.java

@@ -100,12 +100,8 @@ private Connection connected(ServerWebSocket socket) {
             } else {
                 socket.writeTextMessage(buffer.toString());
             }
-        }, socket.textHandlerID()).setProperty(PROTOCOL_CONNECTION, socket.remoteAddress().host());
-    }
-
-    @Override
-    public void stop(Promise<Void> stop) {
-        handler.stop(stop);
+        }, socket.textHandlerID())
+                .setProperty(PROTOCOL_CONNECTION, socket.remoteAddress().host());
     }
 
     private void handle(Connection connection, Buffer buffer) {
@@ -118,6 +114,11 @@ private void handle(Connection connection, Buffer buffer) {
         }
     }
 
+    @Override
+    public void stop(Promise<Void> stop) {
+        handler.stop(stop);
+    }
+
     @Override
     public String toString() {
         return handler.getClass().getSimpleName() + LOG_AT + handler.address() + " port :" +