public async Task OnConnectedAsync(FramingConnection connection) { var inputPipe = connection.Input; var modeDecoder = new ServerModeDecoder(); try { await modeDecoder.ReadModeAsync(inputPipe); } catch (CommunicationException e) { // see if we need to send back a framing fault string framingFault; if (FramingEncodingString.TryGetFaultString(e, out framingFault)) { // TODO: Timeouts await connection.SendFaultAsync(framingFault, Timeout.InfiniteTimeSpan /*GetRemainingTimeout()*/, ConnectionOrientedTransportDefaults.MaxViaSize + ConnectionOrientedTransportDefaults.MaxContentTypeSize); } return; // Completing the returned Task causes the connection to be closed if needed and cleans everything up. } connection.FramingMode = modeDecoder.Mode; await _next(connection); // One .NET Framework, with the way that AspNetCore closes a connection, it sometimes doesn't send the // final bytes if those bytes haven't been sent yet. Delaying completeing the connection to compensate. await Task.Delay(5); // AspNetCore 2.1 doesn't close the connection. 2.2+ does so these lines can eventually be rmoved. connection.RawTransport.Input.Complete(); connection.RawTransport.Output.Complete(); }
public async Task OnConnectedAsync(FramingConnection connection) { var inputPipe = connection.Input; var modeDecoder = new ServerModeDecoder(); await modeDecoder.ReadModeAsync(inputPipe); connection.FramingMode = modeDecoder.Mode; await _next(connection); }
public async Task OnConnectedAsync(FramingConnection connection) { using (_appLifetime.ApplicationStopped.Register(() => { connection.RawTransport.Input.Complete(); connection.RawTransport.Output.Complete(); })) { IConnectionReuseHandler reuseHandler = null; do { var inputPipe = connection.Input; var modeDecoder = new ServerModeDecoder(); try { if (!await modeDecoder.ReadModeAsync(inputPipe)) { break; // Input pipe closed } } catch (CommunicationException e) { // see if we need to send back a framing fault string framingFault; if (FramingEncodingString.TryGetFaultString(e, out framingFault)) { // TODO: Timeouts await connection.SendFaultAsync(framingFault, Timeout.InfiniteTimeSpan /*GetRemainingTimeout()*/, ConnectionOrientedTransportDefaults.MaxViaSize + ConnectionOrientedTransportDefaults.MaxContentTypeSize); } return; // Completing the returned Task causes the connection to be closed if needed and cleans everything up. } connection.FramingMode = modeDecoder.Mode; await _next(connection); // Unwrap the connection. // TODO: Investigate calling Dispose on the wrapping stream to improve cleanup. nb: .NET Framework does not call Dispose. connection.Transport = connection.RawTransport; // connection.ServiceDispatcher is null until later middleware layers are executed. if (reuseHandler == null) { reuseHandler = connection.ServiceDispatcher.Binding.GetProperty <IConnectionReuseHandler>(new BindingParameterCollection()); } } while (await reuseHandler.ReuseConnectionAsync(connection, _appLifetime.ApplicationStopping)); // On .NET Framework, with the way that AspNetCore closes a connection, it sometimes doesn't send the // final bytes if those bytes haven't been sent yet. Delaying completeing the connection to compensate. await Task.Delay(5); } // AspNetCore 2.1 doesn't close the connection. 2.2+ does so these lines can eventually be removed. connection.RawTransport.Input.Complete(); connection.RawTransport.Output.Complete(); }