private void StartListener()
{
var server = new AsyncNetworkStream(_listener.Accept());
HandshakeMessage handshake = server.ReceiveHandshake().Result;
server.SendHandshake(HubId.Generate());
using (var buffer = new PooledMemoryStream())
{
while (true)
{
buffer.Reset();
Receive(server, buffer, WireHeader.Size);
buffer.Position = 0;
WireHeader header = WireHeader.ReadFrom(buffer);
//Console.WriteLine($"Received {header.MessageHeader} ({header.PayloadSize.Value.ToString(CultureInfo.InvariantCulture)})");
Receive(server, buffer, header.PayloadSize.Value);
_serverReceived++;
if (_serverReceived == _count + 1)
{
_allMessagesReceived.Set();
}
}
}
}
private async ValueTask Receive()
{
var buffer = new PooledMemoryStream();
Message message;
try
{
await Receive(buffer, WireHeader.Size).ConfigureAwait(false);
buffer.Position = 0;
WireHeader wireHeader = WireHeader.ReadFrom(buffer);
buffer.Reset();
await Receive(buffer, wireHeader.PayloadSize.Value).ConfigureAwait(false);
buffer.Position = 0;
message = new Message(wireHeader.MessageHeader, buffer);
}
catch (Exception ex)
{
if (!(ex is OperationCanceledException))
{
//if (_logger.IsEnabled(LogLevel.Error))
//{
// _logger.LogError(ex, "Error during receive");
//}
}
//if(ex is SocketException sex && sex.ErrorCode == SocketError.TimedOut)
buffer.Dispose();
return;
}
_session.EnqueueIncoming(message);
}
