feat: adding option to use UniTaskVoid for message handlers

MirageNet · Apr 30, 2023 · 64b9cc7 · 64b9cc7
1 parent 5a2f0e0
commit 64b9cc7
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/Assets/Mirage/Runtime/INetworkPlayer.cs b/Assets/Mirage/Runtime/INetworkPlayer.cs
@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using Cysharp.Threading.Tasks;
 using Mirage.SocketLayer;
 
 namespace Mirage
@@ -17,6 +18,9 @@ public interface IMessageSender
     // delegates to give names to variables in handles
     public delegate void MessageDelegate<in T>(T message);
     public delegate void MessageDelegateWithPlayer<in T>(INetworkPlayer player, T message);
+    public delegate UniTaskVoid MessageDelegateAsync<in T>(T message);
+    public delegate UniTaskVoid MessageDelegateWithPlayerAsync<in T>(INetworkPlayer player, T message);
+
 
     /// <summary>
     /// An object that can receive messages

diff --git a/Assets/Mirage/Runtime/MessageHandler.cs b/Assets/Mirage/Runtime/MessageHandler.cs
@@ -66,6 +66,16 @@ public void RegisterHandler<T>(MessageDelegate<T> handler)
             RegisterHandler<T>((_, value) => handler.Invoke(value));
         }
 
+        public void RegisterHandler<T>(MessageDelegateAsync<T> handler)
+        {
+            RegisterHandler<T>((_, value) => handler.Invoke(value).Forget());
+        }
+
+        public void RegisterHandler<T>(MessageDelegateWithPlayerAsync<T> handler)
+        {
+            RegisterHandler<T>((player, value) => handler.Invoke(player, value).Forget());
+        }
+
         /// <summary>
         /// Unregister a handler for a particular message type.
         /// <para>Note: Messages dont need to be unregister when server or client stops as MessageHandler will be re-created next time server or client starts</para>