public async Task Invoke(HttpContext context)
{
if (!context.WebSockets.IsWebSocketRequest)
{
Log.Info("Connection is not WebSocket, pass through");
await _next.Invoke(context).ConfigureAwait(false);
return;
}
var socket = await context.WebSockets.AcceptWebSocketAsync().ConfigureAwait(false);
await _webSocketHandler.OnConnected(socket).ConfigureAwait(false);
await Receive(socket, async (result, buffer) =>
{
if (result.MessageType == WebSocketMessageType.Text)
{
await _webSocketHandler.ReceiveAsync(socket, result, buffer).ConfigureAwait(false);
}
else if (result.MessageType == WebSocketMessageType.Close)
{
await _webSocketHandler.OnDisconnected(socket).ConfigureAwait(false);
}
}).ConfigureAwait(false);
}
