feat: add client connection parameter to messager handler

Author: insthync

SimpleNetworkManager.NET.Tests/Messages/TestMessageHandler.cs

@@ -1,5 +1,6 @@
 using Cysharp.Threading.Tasks;
 using Insthync.SimpleNetworkManager.NET.Messages;
+using Insthync.SimpleNetworkManager.NET.Network;
 
 namespace Insthync.SimpleNetworkManager.NET.Tests.Messages
 {
@@ -9,7 +10,7 @@ public class TestMessageHandler : BaseMessageHandler<TestMessage>
         public bool boolVal;
         public string? stringVal;
 
-        protected override UniTask HandleAsync(TestMessage data)
+        protected override UniTask HandleAsync(BaseClientConnection clientConnection, TestMessage data)
         {
             intVal = data.intVal;
             boolVal = data.boolVal;

SimpleNetworkManager.NET/Messages/BaseMessageHandler.cs

@@ -1,4 +1,5 @@
 using Cysharp.Threading.Tasks;
+using Insthync.SimpleNetworkManager.NET.Network;
 using System;
 
 namespace Insthync.SimpleNetworkManager.NET.Messages
@@ -15,13 +16,13 @@ public virtual BaseMessage GetMessageInstance()
             return s_messageInstance;
         }
 
-        public UniTask HandleDataAsync(object? data)
+        public UniTask HandleDataAsync(BaseClientConnection clientConnection, object? data)
         {
             if (data == null)
                 throw new ArgumentNullException(nameof(data));
-            return HandleAsync((T)data);
+            return HandleAsync(clientConnection, (T)data);
         }
 
-        protected abstract UniTask HandleAsync(T data);
+        protected abstract UniTask HandleAsync(BaseClientConnection clientConnection, T data);
     }
 }

SimpleNetworkManager.NET/Messages/IMessageHandler.cs

@@ -1,10 +1,11 @@
 using Cysharp.Threading.Tasks;
+using Insthync.SimpleNetworkManager.NET.Network;
 
 namespace Insthync.SimpleNetworkManager.NET.Messages
 {
     public interface IMessageHandler
     {
         BaseMessage GetMessageInstance();
-        public UniTask HandleDataAsync(object? data);
+        public UniTask HandleDataAsync(BaseClientConnection clientConnection, object? data);
     }
 }

SimpleNetworkManager.NET/Services/MessageRouter.cs

@@ -76,7 +76,7 @@ public async UniTask RouteMessageAsync(BaseClientConnection clientConnection, by
 
             var messageInstance = handler.GetMessageInstance();
             var dataType = messageInstance.GetType();
-            await handler.HandleDataAsync(MessagePackSerializer.Deserialize(messageInstance.GetType(), data, messageInstance.GetMessagePackOptions()));
+            await handler.HandleDataAsync(clientConnection, MessagePackSerializer.Deserialize(messageInstance.GetType(), data, messageInstance.GetMessagePackOptions()));
         }
     }
 }