add handler for try connect event (#33)

youngmonkeys · Aug 3, 2023 · c6e81d4 · c6e81d4
1 parent c4a91f0
commit c6e81d4
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/support/EzySocketProxy.cs b/support/EzySocketProxy.cs
@@ -40,6 +40,7 @@ public class EzySocketProxy
         private readonly IDictionary<Object, EventHandler> disconnectedHandlers;
         private readonly IDictionary<Object, EventHandler> reconnectingHandlers;
         private readonly IDictionary<Object, EventHandler> pingLostHandlers;
+        private readonly IDictionary<Object, EventHandler> tryConnectHandlers;
 
         public EzySocketProxy(
             String zoneName,
@@ -60,6 +61,7 @@ EzyAppFactory appFactory
             this.disconnectedHandlers = new Dictionary<Object, EventHandler>();
             this.reconnectingHandlers = new Dictionary<Object, EventHandler>();
             this.pingLostHandlers = new Dictionary<Object, EventHandler>();
+            this.tryConnectHandlers = new Dictionary<Object, EventHandler>();
         }
 
         public EzySocketProxy setHost(String host)
@@ -195,6 +197,7 @@ private void init()
                 client.setup()
                     .addEventHandler(EzyEventType.DISCONNECTION, new DisconnectionHandler(this))
                     .addEventHandler(EzyEventType.LOST_PING, new PingLostHandler(this))
+                    .addEventHandler(EzyEventType.TRY_CONNECT, new TryConnectHandler(this))
                     .addDataHandler(EzyCommand.HANDSHAKE, new HandshakeEventHandler(this))
                     .addDataHandler(EzyCommand.LOGIN, new LoginSuccessHandler(this))
                     .addDataHandler(EzyCommand.LOGIN_ERROR, new LoginErrorHandler(this))
@@ -303,6 +306,18 @@ EzySocketProxyEventHandler<EzyLostPingEvent> handler
             return handler;
         }
 
+        public Object onTryConnect(
+            EzySocketProxyEventHandler<EzyTryConnectEvent> handler
+        )
+        {
+            EventHandler dataHandler = evt =>
+            {
+                handler.Invoke(this, (EzyTryConnectEvent)evt);
+            };
+            tryConnectHandlers[handler] = dataHandler;
+            return handler;
+        }
+
         public void unbind(Object handler)
         {
             loginSuccessHandlers.Remove(handler);
@@ -312,6 +327,7 @@ public void unbind(Object handler)
             disconnectedHandlers.Remove(handler);
             reconnectingHandlers.Remove(handler);
             pingLostHandlers.Remove(handler);
+            tryConnectHandlers.Remove(handler);
         }
 
         public class HandshakeEventHandler : EzyHandshakeHandler
@@ -508,6 +524,24 @@ protected override void process(EzyLostPingEvent evt)
                 }
             }
         }
+
+        public class TryConnectHandler : EzyAbstractEventHandler<EzyTryConnectEvent>
+        {
+            private readonly EzySocketProxy parent;
+
+            public TryConnectHandler(EzySocketProxy parent)
+            {
+                this.parent = parent;
+            }
+
+            protected override void process(EzyTryConnectEvent evt)
+            {
+                foreach (EventHandler handler in parent.tryConnectHandlers.Values)
+                {
+                    handler.Invoke(evt);
+                }
+            }
+        }
     }
 }