EllieBotDevs/elliebot
1
1
Fork 1
Code Issues Pull requests Projects Releases 32 Packages 1 Wiki Activity

Changed how leaving unkept servers work. It only works per-shard now

Browse source
This commit is contained in:
Toastie 2024-08-27 22:05:37 +12:00
parent 7f935a72c1
commit 7874008da2
Signed by: toastie_t0ast
GPG key ID: 27F3B6855AFD40A4
3 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/EllieBot/Modules/Administration/DangerousCommands/CleanupCommands.cs
View file

@ -42,7 +42,7 @@ public partial class Administration
[Cmd]
[OwnerOnly]
public async Task LeaveUnkeptServers()
public async Task LeaveUnkeptServers(int shardId = 0)
{
var keptGuildCount = await _svc.GetKeptGuildCount();
@ -58,7 +58,7 @@ public partial class Administration
if (!response)
return;
await _svc.LeaveUnkeptServers();
await _svc.LeaveUnkeptServers(shardId);
await ctx.OkAsync();
}
}

13
src/EllieBot/Modules/Administration/DangerousCommands/CleanupService.cs
View file

@ -14,7 +14,7 @@ public sealed class CleanupService : ICleanupService, IReadyExecutor, IEService
private TypedKey<KeepReport> _cleanupReportKey = new("cleanup:report");
private TypedKey<bool> _cleanupTriggerKey = new("cleanup:trigger");
private TypedKey<bool> _keepTriggerKey = new("keep:trigger");
private TypedKey<int> _keepTriggerKey = new("keep:trigger");
private readonly IPubSub _pubSub;
private readonly DiscordSocketClient _client;
@ -47,8 +47,11 @@ public sealed class CleanupService : ICleanupService, IReadyExecutor, IEService
private bool keepTriggered = false;
private async ValueTask InternalTriggerKeep(bool arg)
private async ValueTask InternalTriggerKeep(int shardId)
{
if (_client.ShardId != shardId)
return;
if (keepTriggered)
return;
@ -80,7 +83,7 @@ public sealed class CleanupService : ICleanupService, IReadyExecutor, IEService
continue;
// 1 leave per 20 seconds per shard
await Task.Delay(RandomNumberGenerator.GetInt32(18_000, 22_000));
await Task.Delay(500);
SocketGuild? guild = null;
try
@ -236,8 +239,8 @@ public sealed class CleanupService : ICleanupService, IReadyExecutor, IEService
return await table.CountAsync();
}
public async Task LeaveUnkeptServers()
=> await _pubSub.Pub(_keepTriggerKey, true);
public async Task LeaveUnkeptServers(int shardId)
=> await _pubSub.Pub(_keepTriggerKey, shardId);
private ValueTask OnKeepReport(KeepReport report)
{

2
src/EllieBot/Modules/Administration/DangerousCommands/_common/ICleanupService.cs
View file

@ -5,5 +5,5 @@ public interface ICleanupService
Task<KeepResult?> DeleteMissingGuildDataAsync();
Task<bool> KeepGuild(ulong guildId);
Task<int> GetKeptGuildCount();
Task LeaveUnkeptServers();
Task LeaveUnkeptServers(int shardId);
}