112 lines
3.4 KiB
C#
112 lines
3.4 KiB
C#
// Copyright (c) .NET Foundation. All rights reserved.
|
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
|
|
|
using System;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.AspNetCore.SignalR.Client;
|
|
using Microsoft.Extensions.CommandLineUtils;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace ClientSample
|
|
{
|
|
internal class HubSample
|
|
{
|
|
internal static void Register(CommandLineApplication app)
|
|
{
|
|
app.Command("hub", cmd =>
|
|
{
|
|
cmd.Description = "Tests a connection to a hub";
|
|
|
|
var baseUrlArgument = cmd.Argument("<BASEURL>", "The URL to the Chat Hub to test");
|
|
|
|
cmd.OnExecute(() => ExecuteAsync(baseUrlArgument.Value));
|
|
});
|
|
}
|
|
|
|
public static async Task<int> ExecuteAsync(string baseUrl)
|
|
{
|
|
var endpoint = new IPEndPoint(IPAddress.Loopback, 9001);
|
|
Console.WriteLine("Connecting to {0}", endpoint);
|
|
var connection = new HubConnectionBuilder()
|
|
.WithEndPoint(endpoint)
|
|
.WithLogging(logging =>
|
|
{
|
|
logging.AddConsole();
|
|
logging.SetMinimumLevel(LogLevel.Trace);
|
|
})
|
|
.Build();
|
|
|
|
try
|
|
{
|
|
var closeTcs = new TaskCompletionSource<object>();
|
|
connection.Closed += e => closeTcs.SetResult(null);
|
|
// Set up handler
|
|
connection.On<string>("Send", Console.WriteLine);
|
|
|
|
await ConnectAsync(connection);
|
|
|
|
Console.WriteLine("Connected to {0}", endpoint);
|
|
|
|
var sendCts = new CancellationTokenSource();
|
|
|
|
Console.CancelKeyPress += async (sender, a) =>
|
|
{
|
|
a.Cancel = true;
|
|
Console.WriteLine("Stopping loops...");
|
|
sendCts.Cancel();
|
|
await connection.DisposeAsync();
|
|
};
|
|
|
|
while (!closeTcs.Task.IsCompleted)
|
|
{
|
|
var completedTask = await Task.WhenAny(Task.Run(() => Console.ReadLine()), closeTcs.Task);
|
|
if (completedTask == closeTcs.Task)
|
|
{
|
|
break;
|
|
}
|
|
|
|
var line = await (Task<string>)completedTask;
|
|
|
|
if (line == null)
|
|
{
|
|
break;
|
|
}
|
|
|
|
await connection.InvokeAsync<object>("Send", line, sendCts.Token);
|
|
}
|
|
}
|
|
catch (AggregateException aex) when (aex.InnerExceptions.All(e => e is OperationCanceledException))
|
|
{
|
|
}
|
|
catch (OperationCanceledException)
|
|
{
|
|
}
|
|
finally
|
|
{
|
|
await connection.DisposeAsync();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
private static async Task ConnectAsync(HubConnection connection)
|
|
{
|
|
// Keep trying to until we can start
|
|
while (true)
|
|
{
|
|
try
|
|
{
|
|
await connection.StartAsync();
|
|
return;
|
|
}
|
|
catch (Exception)
|
|
{
|
|
await Task.Delay(1000);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|