99 lines
2.9 KiB
C#
99 lines
2.9 KiB
C#
using Gameboard.ShogiUI.Sockets.Extensions;
|
|
using Gameboard.ShogiUI.Sockets.ServiceModels.Socket;
|
|
using Microsoft.Extensions.Logging;
|
|
using Newtonsoft.Json;
|
|
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.Net.WebSockets;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace Gameboard.ShogiUI.Sockets.Managers
|
|
{
|
|
public interface ISocketConnectionManager
|
|
{
|
|
Task BroadcastToAll(ISocketResponse response);
|
|
void Subscribe(WebSocket socket, string playerName);
|
|
void Unsubscribe(string playerName);
|
|
Task BroadcastToPlayers(ISocketResponse response, params string?[] playerNames);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Retains all active socket connections and provides convenient methods for sending messages to clients.
|
|
/// </summary>
|
|
public class SocketConnectionManager : ISocketConnectionManager
|
|
{
|
|
/// <summary>Dictionary key is player name.</summary>
|
|
private readonly ConcurrentDictionary<string, WebSocket> connections;
|
|
/// <summary>Dictionary key is game name.</summary>
|
|
private readonly ILogger<SocketConnectionManager> logger;
|
|
|
|
public SocketConnectionManager(ILogger<SocketConnectionManager> logger)
|
|
{
|
|
this.logger = logger;
|
|
connections = new ConcurrentDictionary<string, WebSocket>();
|
|
}
|
|
|
|
public void Subscribe(WebSocket socket, string playerName)
|
|
{
|
|
connections.TryRemove(playerName, out var _);
|
|
connections.TryAdd(playerName, socket);
|
|
}
|
|
|
|
public void Unsubscribe(string playerName)
|
|
{
|
|
connections.TryRemove(playerName, out _);
|
|
}
|
|
|
|
public async Task BroadcastToPlayers(ISocketResponse response, params string?[] playerNames)
|
|
{
|
|
var tasks = new List<Task>(playerNames.Length);
|
|
foreach (var name in playerNames)
|
|
{
|
|
if (!string.IsNullOrEmpty(name) && connections.TryGetValue(name, out var socket))
|
|
{
|
|
var serialized = JsonConvert.SerializeObject(response);
|
|
logger.LogInformation("Response to {0} \n{1}\n", name, serialized);
|
|
tasks.Add(socket.SendTextAsync(serialized));
|
|
}
|
|
}
|
|
await Task.WhenAll(tasks);
|
|
}
|
|
public Task BroadcastToAll(ISocketResponse response)
|
|
{
|
|
var message = JsonConvert.SerializeObject(response);
|
|
logger.LogInformation($"Broadcasting\n{0}", message);
|
|
var tasks = new List<Task>(connections.Count);
|
|
foreach (var kvp in connections)
|
|
{
|
|
var socket = kvp.Value;
|
|
try
|
|
{
|
|
|
|
tasks.Add(socket.SendTextAsync(message));
|
|
}
|
|
catch (WebSocketException)
|
|
{
|
|
logger.LogInformation("Tried sending a message to socket connection for user [{user}], but found the connection has closed.", kvp.Key);
|
|
Unsubscribe(kvp.Key);
|
|
}
|
|
catch
|
|
{
|
|
logger.LogInformation("Tried sending a message to socket connection for user [{user}], but found the connection has closed.", kvp.Key);
|
|
Unsubscribe(kvp.Key);
|
|
}
|
|
}
|
|
try
|
|
{
|
|
var task = Task.WhenAll(tasks);
|
|
return task;
|
|
}
|
|
catch
|
|
{
|
|
Console.WriteLine("Yo");
|
|
}
|
|
return Task.FromResult(0);
|
|
}
|
|
}
|
|
}
|