From 3d7c49a94a38baf34b2e132449aead5e88ac67d1 Mon Sep 17 00:00:00 2001
From: ALEXTANG <574809918@qq.com>
Date: Thu, 1 Sep 2022 11:37:50 +0800
Subject: [PATCH] Update Network.cs

---
 .../Runtime/Core/NetWork/Core/Network.cs      | 36 +++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/Assets/TEngine/Scripts/Runtime/Core/NetWork/Core/Network.cs b/Assets/TEngine/Scripts/Runtime/Core/NetWork/Core/Network.cs
index b156d585..5b286c8d 100644
--- a/Assets/TEngine/Scripts/Runtime/Core/NetWork/Core/Network.cs
+++ b/Assets/TEngine/Scripts/Runtime/Core/NetWork/Core/Network.cs
@@ -153,5 +153,41 @@ namespace TEngine.Runtime
         {
             GameEventMgr.Instance.Send(NetWorkEventId.NetworkCustomErrorEvent,channel,message);
         }
+
+        /// <summary>
+        /// 注册网络消息包处理函数。
+        /// </summary>
+        /// <param name="channelName"></param>
+        /// <param name="actionId"></param>
+        /// <param name="msgDelegate"></param>
+        /// <param name="checkRepeat"></param>
+        public void RegisterHandler(string channelName, int actionId, CsMsgDelegate msgDelegate,
+            bool checkRepeat = true)
+        {
+            var channel = this.GetNetworkChannel(channelName);
+            if (channel == null)
+            {
+                Log.Warning($"Channel is null :{channelName}");
+                return;
+            }
+            channel.RegisterHandler(actionId,msgDelegate);
+        }
+
+        /// <summary>
+        /// 注销网络消息包处理函数。
+        /// </summary>
+        /// <param name="channelName"></param>
+        /// <param name="actionId"></param>
+        /// <param name="msgDelegate"></param>
+        public void RmvHandler(string channelName, int actionId, CsMsgDelegate msgDelegate)
+        {
+            var channel = this.GetNetworkChannel(channelName);
+            if (channel == null)
+            {
+                Log.Warning($"Channel is null :{channelName}");
+                return;
+            }
+            channel.RmvHandler(actionId,msgDelegate);
+        }
     }
 }
\ No newline at end of file