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))
{
Task<int> result;
var firstLoop = true;
do
{
var send100Continue = 0;
result = SkipAsyncImplementation(cancellationToken);
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);
}