move classes around, name caches more consistently, add more caches

This commit is contained in:
sam 2024-08-19 16:12:28 +02:00
parent e86b37ce2a
commit e17dcf90a1
Signed by: sam
GPG key ID: 5F3C3C1B3166639D
30 changed files with 443 additions and 51 deletions

View file

@ -1,63 +0,0 @@
using System.Collections.Concurrent;
using System.Diagnostics.CodeAnalysis;
using Remora.Discord.API.Abstractions.Objects;
using Remora.Rest.Core;
namespace Catalogger.Backend.Cache;
public class RoleCacheService
{
private readonly ConcurrentDictionary<Snowflake, IRole> _roles = new();
private readonly ConcurrentDictionary<Snowflake, HashSet<Snowflake>> _guildRoles = new();
public int Size => _roles.Count;
public void Set(IRole role, Snowflake guildId)
{
_roles[role.ID] = role;
// Add to set of guild channels
_guildRoles.AddOrUpdate(guildId,
_ => [role.ID],
(_, l) =>
{
l.Add(role.ID);
return l;
});
}
public bool TryGet(Snowflake id, [NotNullWhen(true)] out IRole? role) =>
_roles.TryGetValue(id, out role);
public void Remove(Snowflake guildId, Snowflake id, out IRole? role)
{
_roles.Remove(id, out role);
// Remove from set of guild channels
_guildRoles.AddOrUpdate(guildId, _ => [], (_, s) =>
{
s.Remove(id);
return s;
});
}
public void RemoveGuild(Snowflake guildId)
{
if (!_guildRoles.TryGetValue(guildId, out var roleIds)) return;
foreach (var id in roleIds)
{
_roles.Remove(id, out _);
}
_guildRoles.Remove(guildId, out _);
}
/// <summary>
/// Gets all of a guild's cached roles.
/// </summary>
/// <param name="guildId">The guild to get the roles of</param>
/// <returns>A list of cached roles</returns>
public IEnumerable<IRole> GuildRoles(Snowflake guildId) =>
!_guildRoles.TryGetValue(guildId, out var roleIds)
? []
: roleIds.Select(id => _roles.GetValueOrDefault(id))
.Where(r => r != null).Select(r => r!);
}