Prevent trying to kick the user multiple times

GeyserMC · Jul 16, 2024 · 983ee2a · 983ee2a
1 parent fd3cabf
commit 983ee2a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/main/java/org/geysermc/discordbot/listeners/AutoModHandler.java b/src/main/java/org/geysermc/discordbot/listeners/AutoModHandler.java
@@ -46,11 +46,15 @@
 
 public class AutoModHandler extends ListenerAdapter {
     private final Cache<Long, Integer> executionCache;
+    private final Cache<Long, Boolean> kickCache;
 
     public AutoModHandler() {
         this.executionCache = CacheBuilder.newBuilder()
             .expireAfterWrite(5, TimeUnit.MINUTES)
             .build();
+        this.kickCache = CacheBuilder.newBuilder()
+            .expireAfterWrite(10, TimeUnit.SECONDS)
+            .build();
     }
 
     @Override
@@ -68,6 +72,10 @@ public void onAutoModExecution(@Nonnull AutoModExecutionEvent event) {
 
         if (executions < 3) return;
 
+        // Prevent trying to kick the user multiple times
+        if (this.kickCache.getIfPresent(userId) != null) return;
+        this.kickCache.put(userId, true);
+
         Member member = event.getGuild().getMemberById(userId);
         if (member == null) return;