Fixing build

This commit is contained in:
moozzyk 2017-02-09 10:59:41 -08:00
parent 0c8df245de
commit 633c572a22
1 changed files with 12 additions and 0 deletions

View File

@ -104,6 +104,7 @@ namespace Microsoft.AspNetCore.Sockets.Client
{
Output.TryComplete();
await _transport.StopAsync();
await DrainMessages();
}
public void Dispose()
@ -112,6 +113,17 @@ namespace Microsoft.AspNetCore.Sockets.Client
_transport.Dispose();
}
private async Task DrainMessages()
{
while (await Input.WaitToReadAsync())
{
if (Input.TryRead(out Message message))
{
message.Dispose();
}
}
}
public static Task<Connection> ConnectAsync(Uri url, ITransport transport) => ConnectAsync(url, transport, null, null);
public static Task<Connection> ConnectAsync(Uri url, ITransport transport, ILoggerFactory loggerFactory) => ConnectAsync(url, transport, null, loggerFactory);
public static Task<Connection> ConnectAsync(Uri url, ITransport transport, HttpClient httpClient) => ConnectAsync(url, transport, httpClient, null);