From 1be52fad08aa089ae20043465a4a8473e95c14cd Mon Sep 17 00:00:00 2001 From: Bruce Wayne Date: Fri, 3 Sep 2021 11:37:44 +0800 Subject: [PATCH] fix: Large buffer will cause FlushAsync block --- Pipelines.Extensions/SocketPipe/SocketPipeWriter.cs | 4 ++-- Pipelines.Extensions/WebSocketPipe/WebSocketPipeWriter.cs | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/Pipelines.Extensions/SocketPipe/SocketPipeWriter.cs b/Pipelines.Extensions/SocketPipe/SocketPipeWriter.cs index 906038b..0951d07 100644 --- a/Pipelines.Extensions/SocketPipe/SocketPipeWriter.cs +++ b/Pipelines.Extensions/SocketPipe/SocketPipeWriter.cs @@ -79,7 +79,7 @@ void CloseSocketIfNeeded() public override async ValueTask FlushAsync(CancellationToken cancellationToken = default) { - var flushResult = await Writer.FlushAsync(cancellationToken); + var flushTask = Writer.FlushAsync(cancellationToken); try { @@ -105,7 +105,7 @@ public override async ValueTask FlushAsync(CancellationToken cancel throw; } - return flushResult; + return await flushTask; } } } diff --git a/Pipelines.Extensions/WebSocketPipe/WebSocketPipeWriter.cs b/Pipelines.Extensions/WebSocketPipe/WebSocketPipeWriter.cs index ba5001c..c476496 100644 --- 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 FlushAsync(CancellationToken cancellationToken = default) { - var flushResult = await Writer.FlushAsync(cancellationToken); + var flushTask = Writer.FlushAsync(cancellationToken); try { @@ -76,7 +76,7 @@ public override async ValueTask FlushAsync(CancellationToken cancel throw; } - return flushResult; + return await flushTask; } } }