stop reading output pipe if the previous read is cancelled

kerryjiang · Jan 1, 2025 · 7deb9a5 · 7deb9a5
1 parent 7305207
commit 7deb9a5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/SuperSocket.Connection/PipeConnection.cs b/src/SuperSocket.Connection/PipeConnection.cs
@@ -53,9 +53,9 @@ protected virtual async Task ProcessSends()
 
             while (true)
             {
-                var completed = await ProcessOutputRead(output).ConfigureAwait(false);
+                var completedOrCancelled = await ProcessOutputRead(output).ConfigureAwait(false);
 
-                if (completed)
+                if (completedOrCancelled)
                 {
                     break;
                 }
@@ -153,7 +153,7 @@ protected async ValueTask<bool> ProcessOutputRead(PipeReader reader)
         {
             var result = await reader.ReadAsync(CancellationToken.None).ConfigureAwait(false);
 
-            var completed = result.IsCompleted;
+            var completedOrCancelled = result.IsCompleted || result.IsCanceled;
 
             var buffer = result.Buffer;
             var end = buffer.End;
@@ -178,7 +178,7 @@ protected async ValueTask<bool> ProcessOutputRead(PipeReader reader)
             }
 
             reader.AdvanceTo(end);
-            return completed;
+            return completedOrCancelled;
         }
 
         protected abstract ValueTask<int> SendOverIOAsync(ReadOnlySequence<byte> buffer, CancellationToken cancellationToken);