尝试修复一个CancellationToken.Register导致内存泄漏问题

Author: aaasoft

Quick.Protocol/QpServerChannel.cs

@@ -13,6 +13,7 @@ public class QpServerChannel : QpChannel
     {
         private readonly Stream stream;
         private readonly CancellationTokenSource cts;
+        private readonly CancellationToken serverCancellationToken;
         private readonly QpServerOptions options;
         private readonly string channelName;
         //通过认证后，才允许使用的命令执行管理器列表
@@ -38,9 +39,9 @@ public QpServerChannel(Stream stream, string channelName, CancellationToken canc
             this.options = options;
             this.authedCommandExecuterManagerList = options.CommandExecuterManagerList;
             this.authedNoticeHandlerManagerList = options.NoticeHandlerManagerList;
+            serverCancellationToken = cancellationToken;
 
             cts = new CancellationTokenSource();
-            cancellationToken.Register(Stop);
 
             //初始化连接相关指令处理器
             var connectAndAuthCommandExecuterManager = new CommandExecuterManager();
@@ -57,6 +58,8 @@ public QpServerChannel(Stream stream, string channelName, CancellationToken canc
             BeginReadPackage(token);
             //开始统计网络数据
             _ = BeginNetstat(token);
+            //开始检查服务端的取消令牌
+            _ = BeginCheckServerCancellationToken(token);
 
             //如果认证超时时间后没有通过认证，则断开连接
             if (options.AuthenticateTimeout > 0)
@@ -79,6 +82,16 @@ public QpServerChannel(Stream stream, string channelName, CancellationToken canc
                 });
         }
 
+        private async Task BeginCheckServerCancellationToken(CancellationToken cancellationToken)
+        {
+            while (!cancellationToken.IsCancellationRequested)
+            {
+                await Task.Delay(1000, cancellationToken);
+                if (serverCancellationToken.IsCancellationRequested)
+                    Stop();
+            }
+        }
+
         private Commands.Connect.Response connect(QpChannel handler, Commands.Connect.Request request)
         {
             if (request.InstructionIds != null)