private async Task <TcpNetworkConnector> StartClient(CancellationToken cancellationToken, int port)
{
TcpNetworkConnector tcpNetworkConnector = new TcpNetworkConnector(MessageTypeCache, MessageSerializer, MessageProcessor, Logger, Host, port);
await tcpNetworkConnector.ConnectAsync(cancellationToken);
tcpNetworkConnector.Start();
return(tcpNetworkConnector);
}
public async Task ConnectAsync_Should_Set_IsConnected_To_True()
{
using CancellationTokenSource cts = new CancellationTokenSource(TimeSpan.FromSeconds(DefaultTimeOut * 3));
_ = StartServer(cts.Token, 9989);
TcpNetworkConnector tcpNetworkConnector = await StartClient(cts.Token, 9989);
Assert.IsTrue(tcpNetworkConnector.IsConnected);
cts.Cancel();
}
/// <inheritdoc cref="IRegistryService.StartReceivingServiceEndPointsAsync(CancellationToken)"/>
public async Task StartReceivingServiceEndPointsAsync(CancellationToken cancellationToken)
{
_tcpListener.Start();
while (!cancellationToken.IsCancellationRequested)
{
TcpClient tcpClient = await _tcpListener.AcceptTcpClientAsync();
_ = Task.Run(() =>
{
INetworkConnector networkConnector = new TcpNetworkConnector(_messageTypeCache, _messageSerializer, _registryServiceMessageProcessor, _tcpNetworkConnectorLogger, tcpClient);
networkConnector.Start();
}, cancellationToken);
_registryServiceLogger.LogInformation("Accepted a new RegistryService!");
}
}
public async Task SendMessageAsync_Should_Invoke_MessageProcessor()
{
using CancellationTokenSource cts = new CancellationTokenSource(TimeSpan.FromSeconds(DefaultTimeOut * 3));
_ = StartServer(cts.Token, 9977);
TcpNetworkConnector tcpNetworkConnector = await StartClient(cts.Token, 9977);
AuthenticateRequest message = new AuthenticateRequest()
{
CredentialTypeCode = "Name", Password = "******", Username = "******"
};
await tcpNetworkConnector.SendMessageAsync(message, cts.Token);
IMessage messageInMessageProcessor = await MessageProcessor.GetMessageAsync(cts.Token);
Assert.AreEqual(message.Id, messageInMessageProcessor?.Id);
cts.Cancel();
}
public async Task SendMessageAsync_Should_Invoke_MessageProcessor()
{
using CancellationTokenSource cts = new CancellationTokenSource(TimeSpan.FromSeconds(DefaultTimeOut * 3));
_ = StartServer(cts.Token, 9988);
TcpNetworkConnector tcpNetworkConnector = await StartClient(cts.Token, 9988);
TestRequestMessage message = new TestRequestMessage()
{
Message = "Test123"
};
await tcpNetworkConnector.SendMessageAsync(message, cts.Token);
IMessage messageInMessageProcessor = await MessageProcessor.GetMessageAsync(cts.Token);
Assert.AreEqual(message.Id, messageInMessageProcessor?.Id);
cts.Cancel();
}
private async Task StartServer(CancellationToken cancellationToken, int port)
{
Console.WriteLine("Starting listener...");
TcpListener tcpListener = new TcpListener(IPAddress.Loopback, port);
tcpListener.Start();
Console.WriteLine("Stared listener!");
while (!cancellationToken.IsCancellationRequested)
{
Console.WriteLine("Awaiting new connection...");
TcpClient tcpClient = await tcpListener.AcceptTcpClientAsync();
Console.WriteLine("Accepted connection!");
_ = Task.Run(() =>
{
TcpNetworkConnector networkConnector = new TcpNetworkConnector(MessageTypeCache, MessageSerializer, MessageProcessor, Logger, tcpClient);
networkConnector.Start();
Console.WriteLine("Started tcp network connector");
}, cancellationToken);
}
}
public async Task RunAsync(Action <string> log)
{
Stopwatch stopwatch = new Stopwatch();
stopwatch.Start();
List <Task> tasks = new List <Task>();
for (int j = 0; j < ClientCount; j++)
{
Task clientTask = Task.Run(async() =>
{
IMessageProcessor messageProcessor = new ClientMessageProcessor();
IMessageSerializer messageSerializer = new MessagePackMessageSerializer();
INetworkConnector networkConnector = new TcpNetworkConnector(messageSerializer, messageProcessor, Host, Port);
await networkConnector.ConnectAsync(CancellationToken.None);
networkConnector.Start();
MessageListener <LoginResponse> loginResponseListener = new MessageListener <LoginResponse>();
loginResponseListener.Subscribe(messageProcessor);
for (int i = 0; i < MessageCount; i++)
{
LoginRequest loginRequest = new LoginRequest(Guid.NewGuid(), "Mario", "password");
await networkConnector.SendMessageAsync(loginRequest, CancellationToken.None);
LoginResponse loginResponse = await loginResponseListener.ReceiveMessageAsync();
log.Invoke($"Success: {loginResponse.IsSuccess}, Username: {loginResponse.UserDto.Name}, RequestId: {loginResponse.RequestId}, ResponseId: {loginResponse.Id}");
}
loginResponseListener.Unsubscribe();
});
tasks.Add(clientTask);
}
await Task.WhenAll(tasks).ContinueWith(action =>
{
stopwatch.Stop();
log.Invoke($"Messages per second: {TotalMessages / stopwatch.Elapsed.TotalSeconds:N0}");
});
}
