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

Clean up the logic in DoReceive

This commit is contained in:
David Fowler 2017-04-14 01:39:56 -07:00
parent aedd061865
commit 9e37272f06
1 changed files with 11 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/Microsoft.AspNetCore.Server.Kestrel.Transport.Sockets/SocketConnection.cs
View File

@ -84,30 +84,24 @@ namespace Microsoft.AspNetCore.Server.Kestrel.Transport.Sockets
// Ensure we have some reasonable amount of buffer space
var buffer = _input.Alloc(MinAllocBufferSize);
int bytesReceived;
try
{
bytesReceived = await _socket.ReceiveAsync(GetArraySegment(buffer.Buffer), SocketFlags.None);
var bytesReceived = await _socket.ReceiveAsync(GetArraySegment(buffer.Buffer), SocketFlags.None);
if (bytesReceived == 0)
{
// We receive a FIN so throw an exception so that we cancel the input
// with an error
throw new TaskCanceledException("The request was aborted");
}
buffer.Advance(bytesReceived);
}
catch (Exception)
finally
{
buffer.Commit();
throw;
}
if (bytesReceived == 0)
{
buffer.Commit();
// We receive a FIN so throw an exception so that we cancel the input
// with an error
throw new TaskCanceledException("The request was aborted");
}
// Record what data we filled into the buffer and push to pipe
buffer.Advance(bytesReceived);
var result = await buffer.FlushAsync();
if (result.IsCompleted)
{