Update FrameResponseStream Cancellation

This commit is contained in:
Ben Adams 2016-01-23 10:11:46 +00:00
parent 735c0fbbef
commit 0ca8ca0328
1 changed files with 2 additions and 5 deletions

View File

@ -138,13 +138,10 @@ namespace Microsoft.AspNetCore.Server.Kestrel.Http
case FrameStreamState.Closed: case FrameStreamState.Closed:
throw new ObjectDisposedException(nameof(FrameResponseStream)); throw new ObjectDisposedException(nameof(FrameResponseStream));
case FrameStreamState.Aborted: case FrameStreamState.Aborted:
if (cancellationToken.CanBeCanceled) if (cancellationToken.IsCancellationRequested)
{ {
// Aborted state only throws on write if cancellationToken requests it // Aborted state only throws on write if cancellationToken requests it
return TaskUtilities.GetCancelledTask( return TaskUtilities.GetCancelledTask(cancellationToken);
cancellationToken.IsCancellationRequested ?
cancellationToken :
new CancellationToken(true));
} }
break; break;
} }