111 lines
3.6 KiB
C#
111 lines
3.6 KiB
C#
using MareSynchronosShared.Protos;
|
|
using Microsoft.Extensions.Logging;
|
|
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace MareSynchronosServices.Identity;
|
|
|
|
internal class IdentityHandler
|
|
{
|
|
private readonly ConcurrentDictionary<string, ServerIdentity> cachedIdentities = new();
|
|
private readonly ConcurrentDictionary<string, ConcurrentQueue<IdentChange>> identChanges = new();
|
|
private readonly ILogger<IdentityHandler> _logger;
|
|
|
|
public IdentityHandler(ILogger<IdentityHandler> logger)
|
|
{
|
|
_logger = logger;
|
|
}
|
|
|
|
internal Task<string> GetUidForCharacterIdent(string ident, string serverId)
|
|
{
|
|
var exists = cachedIdentities.Any(f => f.Value.CharacterIdent == ident && f.Value.ServerId == serverId);
|
|
return Task.FromResult(exists ? cachedIdentities.FirstOrDefault(f => f.Value.CharacterIdent == ident && f.Value.ServerId == serverId).Key : string.Empty);
|
|
}
|
|
|
|
internal Task<ServerIdentity> GetIdentForuid(string uid)
|
|
{
|
|
ServerIdentity result;
|
|
if (!cachedIdentities.TryGetValue(uid, out result))
|
|
{
|
|
result = new ServerIdentity();
|
|
}
|
|
|
|
return Task.FromResult(result);
|
|
}
|
|
|
|
internal void SetIdent(string uid, string serverId, string ident)
|
|
{
|
|
cachedIdentities[uid] = new ServerIdentity() { ServerId = serverId, CharacterIdent = ident };
|
|
}
|
|
|
|
internal void RemoveIdent(string uid, string serverId)
|
|
{
|
|
if (cachedIdentities.ContainsKey(uid) && cachedIdentities[uid].ServerId == serverId)
|
|
{
|
|
cachedIdentities.TryRemove(uid, out _);
|
|
}
|
|
}
|
|
|
|
internal int GetOnlineUsers(string serverId)
|
|
{
|
|
if (string.IsNullOrEmpty(serverId))
|
|
return cachedIdentities.Count;
|
|
return cachedIdentities.Count(c => c.Value.ServerId == serverId);
|
|
}
|
|
|
|
internal Dictionary<string, ServerIdentity> GetIdentsForAllExcept(string serverId)
|
|
{
|
|
return cachedIdentities.Where(k => k.Value.ServerId != serverId).ToDictionary(k => k.Key, k => k.Value);
|
|
}
|
|
|
|
internal Dictionary<string, ServerIdentity> GetIdentsForServer(string serverId)
|
|
{
|
|
return cachedIdentities.Where(k => k.Value.ServerId == serverId).ToDictionary(k => k.Key, k => k.Value);
|
|
}
|
|
|
|
internal void ClearIdentsForServer(string serverId)
|
|
{
|
|
var serverIdentities = cachedIdentities.Where(i => i.Value.ServerId == serverId);
|
|
foreach (var identity in serverIdentities)
|
|
{
|
|
cachedIdentities.TryRemove(identity.Key, out _);
|
|
}
|
|
}
|
|
|
|
internal void EnqueueIdentChange(IdentChange identchange)
|
|
{
|
|
_logger.LogInformation("Enqueued " + identchange.UidWithIdent.Uid.Uid + ":" + identchange.IsOnline + " from " + identchange.UidWithIdent.Ident.ServerId);
|
|
|
|
foreach (var k in identChanges.Keys)
|
|
{
|
|
if (string.Equals(k, identchange.UidWithIdent.Ident.ServerId, System.StringComparison.Ordinal)) continue;
|
|
identChanges[k].Enqueue(identchange);
|
|
}
|
|
}
|
|
|
|
internal bool DequeueIdentChange(string server, out IdentChange? cur)
|
|
{
|
|
if (!(identChanges.ContainsKey(server) && identChanges[server].TryDequeue(out cur)))
|
|
{
|
|
cur = null;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
internal void RegisterServerForQueue(string serverId)
|
|
{
|
|
identChanges[serverId] = new ConcurrentQueue<IdentChange>();
|
|
}
|
|
}
|
|
|
|
internal record ServerIdentity
|
|
{
|
|
public string ServerId { get; set; } = string.Empty;
|
|
public string CharacterIdent { get; set; } = string.Empty;
|
|
}
|