private async Task HandleAfterSocketAsync
(WebSocket nativeSocket, MariBaseWebSocketService service, HttpContext context)
{
using var socket = new MariWebSocket(nativeSocket, service.Cts.Token, service);
_logger.LogDebug($"The new WebSocket has the Id: {socket.Id}");
service.AddClient(socket);
await service.OnOpenAsync(socket, context)
.Try(_logger, service, socket, false);
try
{
await ReadAsync(socket, service, context);
}
catch (Exception ex)
{
if (ex is TaskCanceledException)
{
return;
}
await service.OnDisconnectedAsync(socket, WebSocketCloseStatus.ProtocolError, ex.Message)
.Try(_logger, service, socket, false);
if (socket.WebSocket.State.Equals(WebSocketState.Open))
{
await socket.WebSocket.CloseAsync(WebSocketCloseStatus.NormalClosure,
"Closed by remote", service.Cts.Token)
.Try(_logger, service, socket, false);
}
}
nativeSocket.Dispose();
}
private async Task ReadMessageAsync
(WebSocketReceiveResult result, byte[] buffer,
MariBaseWebSocketService service, MariWebSocket socket, HttpContext context)
{
if (!result.EndOfMessage)
{
return;
}
_logger.LogTrace($"Incoming WebSocket message from " +
$"{context.Connection.RemoteIpAddress}:{context.Connection.RemotePort}.");
_logger.LogDebug($"Received WebSocket message from client's id: {socket.Id}");
if (result.MessageType.Equals(WebSocketMessageType.Text))
{
await service.OnMessageAsync(socket, Encoding.UTF8.GetString(buffer))
.Try(_logger, service, socket, false);
}
else if (result.MessageType.Equals(WebSocketMessageType.Close))
{
_logger.LogTrace($"Incoming WebSocket disconnect from " +
$"{context.Connection.RemoteIpAddress}:{context.Connection.RemotePort}.");
_logger.LogDebug($"WebSocket with id {socket.Id} disconnected.");
await service.OnDisconnectedAsync(
socket, result.CloseStatus.Value, Encoding.UTF8.GetString(buffer))
.Try(_logger, service, socket, false);
service.RemoveClient(socket);
await socket.WebSocket.CloseAsync(WebSocketCloseStatus.NormalClosure,
"Closed by remote", service.Cts.Token)
.Try(_logger, service, socket, false);
socket.Dispose();
}
}