minster586/nadekobot
1
0
Fork 0
mirror of https://gitlab.com/Kwoth/nadekobot.git synced 2025-09-11 01:38:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Kwoth
2024-08-27 01:38:36 +00:00
parent 8b2ed0dbdc
commit 52b87c7776
3 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/NadekoBot/Modules/Administration/DangerousCommands/CleanupCommands.cs
View File

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

13
src/NadekoBot/Modules/Administration/DangerousCommands/CleanupService.cs
View File

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

2
src/NadekoBot/Modules/Administration/DangerousCommands/_common/ICleanupService.cs
View File

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