BirthdayBot/BackgroundServices/AutoUserDownload.cs

using BirthdayBot.Data;
using Microsoft.EntityFrameworkCore;

namespace BirthdayBot.BackgroundServices;
/// <summary>
/// Proactively fills the user cache for guilds in which any birthday data already exists.
/// </summary>
class AutoUserDownload : BackgroundService {
    public AutoUserDownload(ShardInstance instance) : base(instance) { }

    public override async Task OnTick(int tickCount, CancellationToken token) {
        // Take action if a guild's cache is incomplete...
        var incompleteCaches = ShardInstance.DiscordClient.Guilds.Where(g => !g.HasAllMembers).Select(g => g.Id).ToHashSet();
        // ...and if the guild contains any user data
        IEnumerable<ulong> mustFetch;
        try {
            await DbConcurrentOperationsLock.WaitAsync(token);
            using var db = new BotDatabaseContext();
            mustFetch = db.UserEntries.AsNoTracking()
                .Where(e => incompleteCaches.Contains(e.GuildId)).Select(e => e.GuildId).Distinct()
                .ToList();
        } finally {
            try {
                DbConcurrentOperationsLock.Release();
            } catch (ObjectDisposedException) { }
        }
        
        var processed = 0;
        foreach (var item in mustFetch) {
            // May cause a disconnect in certain situations. Cancel all further attempts until the next pass if it happens.
            if (ShardInstance.DiscordClient.ConnectionState != ConnectionState.Connected) break;

            var guild = ShardInstance.DiscordClient.GetGuild((ulong)item);
            if (guild == null) continue; // A guild disappeared...?
            await guild.DownloadUsersAsync(); // We're already on a seperate thread, no need to use Task.Run
            await Task.Delay(200, CancellationToken.None); // Must delay, or else it seems to hang...
            processed++;

            if (processed >= 150) break; // take a break (don't get killed by ShardManager for taking too long due to quantity)
        }

        if (processed > 25) Log($"Explicit user list request processed for {processed} guild(s).");
    }
}