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

src/Microsoft.AspNetCore.SignalR/HubEndPoint.cs

@@ -145,16 +145,13 @@ namespace Microsoft.AspNetCore.SignalR
 
             while (await connection.Transport.Input.WaitToReadAsync())
             {
-                Message message;
+                Message incomingMessage;
-                if (!connection.Transport.Input.TryRead(out message))
+                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;
+                        }
+                    }
                 }
             }
         }

src/Microsoft.AspNetCore.Sockets/HttpConnectionDispatcher.cs

@@ -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
             {

src/Microsoft.AspNetCore.Sockets/Transports/LongPollingTransport.cs

@@ -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))

src/Microsoft.AspNetCore.Sockets/Transports/ServerSentEventsTransport.cs

@@ -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)
                         {

src/Microsoft.AspNetCore.Sockets/Transports/WebSocketsTransport.cs

@@ -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)
                     {