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

Use TryRead and TryWrite (#113) 

* Use TryRead and TryWrite
- Use TryWrite to avoid errors on channel close for /send requests
- Use TryRead until it returns false for all transports but long polling
This commit is contained in:
David Fowler 2017-01-11 12:27:18 -08:00 committed by GitHub
parent 5d374b7dbe
commit 8dc68cb798
5 changed files with 66 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/Microsoft.AspNetCore.SignalR/HubEndPoint.cs
View File

@ -145,16 +145,13 @@ namespace Microsoft.AspNetCore.SignalR
while (await connection.Transport.Input.WaitToReadAsync())
{
Message message;
if (!connection.Transport.Input.TryRead(out message))
Message incomingMessage;
while (connection.Transport.Input.TryRead(out incomingMessage))
{
continue;
}
InvocationDescriptor invocationDescriptor;
using (message)
using (incomingMessage)
{
var inputStream = new MemoryStream(message.Payload.Buffer.ToArray());
var inputStream = new MemoryStream(incomingMessage.Payload.Buffer.ToArray());
// TODO: Handle receiving InvocationResultDescriptor
invocationDescriptor = await invocationAdapter.ReadMessageAsync(inputStream, this) as InvocationDescriptor;
@ -194,9 +191,15 @@ namespace Microsoft.AspNetCore.SignalR
await invocationAdapter.WriteMessageAsync(result, outStream);
var buffer = ReadableBuffer.Create(outStream.ToArray()).Preserve();
if (await connection.Transport.Output.WaitToWriteAsync())
var outMessage = new Message(buffer, connection.Metadata.Format, endOfMessage: true);
while (await connection.Transport.Output.WaitToWriteAsync())
{
connection.Transport.Output.TryWrite(new Message(buffer, connection.Metadata.Format, endOfMessage: true));
if (connection.Transport.Output.TryWrite(outMessage))
{
break;
}
}
}
}
}

10
src/Microsoft.AspNetCore.Sockets/HttpConnectionDispatcher.cs
View File

@ -225,8 +225,14 @@ namespace Microsoft.AspNetCore.Sockets
format,
endOfMessage: true);
await state.Application.Output.WriteAsync(message);
// REVIEW: Do we want to return a specific status code here if the connection has ended?
while (await state.Application.Output.WaitToWriteAsync())
{
if (state.Application.Output.TryWrite(message))
{
break;
}
}
}
else
{

15
src/Microsoft.AspNetCore.Sockets/Transports/LongPollingTransport.cs
View File

@ -24,20 +24,17 @@ namespace Microsoft.AspNetCore.Sockets.Transports
public async Task ProcessRequestAsync(HttpContext context)
{
if (_application.Completion.IsCompleted)
{
// Client should stop if it receives a 204
_logger.LogInformation("Terminating Long Polling connection by sending 204 response.");
context.Response.StatusCode = 204;
return;
}
try
{
// TODO: We need the ability to yield the connection without completing the channel.
// This is to force ReadAsync to yield without data to end to poll but not the entire connection.
// This is for cases when the client reconnects see issue #27
await _application.WaitToReadAsync(context.RequestAborted);
if (!await _application.WaitToReadAsync(context.RequestAborted))
{
_logger.LogInformation("Terminating Long Polling connection by sending 204 response.");
context.Response.StatusCode = 204;
return;
}
Message message;
if (_application.TryRead(out message))

2
src/Microsoft.AspNetCore.Sockets/Transports/ServerSentEventsTransport.cs
View File

@ -33,7 +33,7 @@ namespace Microsoft.AspNetCore.Sockets.Transports
while (await _application.WaitToReadAsync(context.RequestAborted))
{
Message message;
if (_application.TryRead(out message))
while (_application.TryRead(out message))
{
using (message)
{

2
src/Microsoft.AspNetCore.Sockets/Transports/WebSocketsTransport.cs
View File

@ -156,7 +156,7 @@ namespace Microsoft.AspNetCore.Sockets.Transports
{
// Get a frame from the application
Message message;
if (_application.Input.TryRead(out message))
while (_application.Input.TryRead(out message))
{
using (message)
{