fix: Large buffer will cause FlushAsync block

HMBSbige · Sep 3, 2021 · 1be52fa · 1be52fa
1 parent f3f12a7
commit 1be52fa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Pipelines.Extensions/SocketPipe/SocketPipeWriter.cs b/Pipelines.Extensions/SocketPipe/SocketPipeWriter.cs
@@ -79,7 +79,7 @@ void CloseSocketIfNeeded()
 
 		public override async ValueTask<FlushResult> FlushAsync(CancellationToken cancellationToken = default)
 		{
-			var flushResult = await Writer.FlushAsync(cancellationToken);
+			var flushTask = Writer.FlushAsync(cancellationToken);
 
 			try
 			{
@@ -105,7 +105,7 @@ public override async ValueTask<FlushResult> FlushAsync(CancellationToken cancel
 				throw;
 			}
 
-			return flushResult;
+			return await flushTask;
 		}
 	}
 }
diff --git a/Pipelines.Extensions/WebSocketPipe/WebSocketPipeWriter.cs b/Pipelines.Extensions/WebSocketPipe/WebSocketPipeWriter.cs
@@ -51,7 +51,7 @@ public override void Complete(Exception? exception = null)
 
 		public override async ValueTask<FlushResult> FlushAsync(CancellationToken cancellationToken = default)
 		{
-			var flushResult = await Writer.FlushAsync(cancellationToken);
+			var flushTask = Writer.FlushAsync(cancellationToken);
 
 			try
 			{
@@ -76,7 +76,7 @@ public override async ValueTask<FlushResult> FlushAsync(CancellationToken cancel
 				throw;
 			}
 
-			return flushResult;
+			return await flushTask;
 		}
 	}
 }