huahaofeng
/
aspnetcore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Early bail, completed tasks+Interlocked

This commit is contained in:
Ben Adams 2015-11-15 21:56:20 +00:00
parent f089abd337
commit 1589b54018
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/Microsoft.AspNet.Server.Kestrel/Http/MessageBody.cs
View File

@ -41,17 +41,27 @@ namespace Microsoft.AspNet.Server.Kestrel.Http
public async Task Consume(CancellationToken cancellationToken = default(CancellationToken)) public async Task Consume(CancellationToken cancellationToken = default(CancellationToken))
{ {
Task<int> result; Task<int> result;
var firstLoop = true;
do do
{ {
var send100Continue = 0;
result = SkipAsyncImplementation(cancellationToken); result = SkipAsyncImplementation(cancellationToken);
if (!result.IsCompleted) if (!result.IsCompleted)
{ {
send100Continue = Interlocked.Exchange(ref _send100Continue, 0); if (firstLoop && Interlocked.Exchange(ref _send100Continue, 0) == 1)
{
firstLoop = false;
_context.FrameControl.ProduceContinue();
}
} }
if (send100Continue == 1) else if (result.GetAwaiter().GetResult() == 0)
{ {
_context.FrameControl.ProduceContinue(); // Completed Task, end of stream
return;
}
else
{
// Completed Task, get next Task rather than await
continue;
} }
} while (await result != 0); } while (await result != 0);
} }