From f3621a915c6548b13aeeaed0ec3fdf7cd2f52e71 Mon Sep 17 00:00:00 2001
From: Yosi Attias <yosy101@gmail.com>
Date: Mon, 16 Sep 2019 12:44:54 +0300
Subject: [PATCH] Add support for event interceptors which catch all events

---
 .../com/corundumstudio/socketio/SocketIOServer.java    |  7 +++++++
 .../socketio/listener/ClientListeners.java             |  3 +++
 .../socketio/listener/EventInterceptor.java            |  9 +++++++++
 .../corundumstudio/socketio/namespace/Namespace.java   | 10 ++++++++++
 4 files changed, 29 insertions(+)
 create mode 100644 src/main/java/com/corundumstudio/socketio/listener/EventInterceptor.java

diff --git a/src/main/java/com/corundumstudio/socketio/SocketIOServer.java b/src/main/java/com/corundumstudio/socketio/SocketIOServer.java
index 5419d2a..ff06769 100644
--- a/src/main/java/com/corundumstudio/socketio/SocketIOServer.java
+++ b/src/main/java/com/corundumstudio/socketio/SocketIOServer.java
@@ -229,6 +229,13 @@ public class SocketIOServer implements ClientListeners {
         mainNamespace.addEventListener(eventName, eventClass, listener);
     }
 
+    @Override
+    public void addEventInterceptor(EventInterceptor eventInterceptor) {
+        mainNamespace.addEventInterceptor(eventInterceptor);
+
+    }
+
+
     @Override
     public void removeAllListeners(String eventName) {
         mainNamespace.removeAllListeners(eventName);
diff --git a/src/main/java/com/corundumstudio/socketio/listener/ClientListeners.java b/src/main/java/com/corundumstudio/socketio/listener/ClientListeners.java
index 124e67b..af6b82a 100644
--- a/src/main/java/com/corundumstudio/socketio/listener/ClientListeners.java
+++ b/src/main/java/com/corundumstudio/socketio/listener/ClientListeners.java
@@ -15,12 +15,15 @@
  */
 package com.corundumstudio.socketio.listener;
 
+
 public interface ClientListeners {
 
     void addMultiTypeEventListener(String eventName, MultiTypeEventListener listener, Class<?> ... eventClass);
 
     <T> void addEventListener(String eventName, Class<T> eventClass, DataListener<T> listener);
 
+    void addEventInterceptor(EventInterceptor eventInterceptor);
+
     void addDisconnectListener(DisconnectListener listener);
 
     void addConnectListener(ConnectListener listener);
diff --git a/src/main/java/com/corundumstudio/socketio/listener/EventInterceptor.java b/src/main/java/com/corundumstudio/socketio/listener/EventInterceptor.java
new file mode 100644
index 0000000..7594ca2
--- /dev/null
+++ b/src/main/java/com/corundumstudio/socketio/listener/EventInterceptor.java
@@ -0,0 +1,9 @@
+package com.corundumstudio.socketio.listener;
+
+import com.corundumstudio.socketio.AckRequest;
+import com.corundumstudio.socketio.transport.NamespaceClient;
+import java.util.List;
+
+public interface  EventInterceptor {
+    void onEvent(NamespaceClient client, String eventName, List<Object> args, AckRequest ackRequest);
+}
diff --git a/src/main/java/com/corundumstudio/socketio/namespace/Namespace.java b/src/main/java/com/corundumstudio/socketio/namespace/Namespace.java
index cbbc227..4eb1df3 100644
--- a/src/main/java/com/corundumstudio/socketio/namespace/Namespace.java
+++ b/src/main/java/com/corundumstudio/socketio/namespace/Namespace.java
@@ -59,6 +59,7 @@ public class Namespace implements SocketIONamespace {
     private final Queue<ConnectListener> connectListeners = new ConcurrentLinkedQueue<ConnectListener>();
     private final Queue<DisconnectListener> disconnectListeners = new ConcurrentLinkedQueue<DisconnectListener>();
     private final Queue<PingListener> pingListeners = new ConcurrentLinkedQueue<PingListener>();
+    private final Queue<EventInterceptor> eventInterceptors = new ConcurrentLinkedQueue<EventInterceptor>();
 
     private final Map<UUID, SocketIOClient> allClients = PlatformDependent.newConcurrentHashMap();
     private final ConcurrentMap<String, Set<UUID>> roomClients = PlatformDependent.newConcurrentHashMap();
@@ -126,6 +127,11 @@ public class Namespace implements SocketIONamespace {
         jsonSupport.addEventMapping(name, eventName, eventClass);
     }
 
+    @Override
+    public void addEventInterceptor(EventInterceptor eventInterceptor) {
+        eventInterceptors.add(eventInterceptor);
+    }
+
     @SuppressWarnings({"rawtypes", "unchecked"})
     public void onEvent(NamespaceClient client, String eventName, List<Object> args, AckRequest ackRequest) {
         EventEntry entry = eventListeners.get(eventName);
@@ -139,6 +145,10 @@ public class Namespace implements SocketIONamespace {
                 Object data = getEventData(args, dataListener);
                 dataListener.onData(client, data, ackRequest);
             }
+
+            for (EventInterceptor eventInterceptor : eventInterceptors) {
+                eventInterceptor.onEvent(client, eventName, args, ackRequest);
+            }
         } catch (Exception e) {
             exceptionListener.onEventException(e, args, client);
             if (ackMode == AckMode.AUTO_SUCCESS_ONLY) {