GH-1153 Rework delay system with per-entry Instant TTL

buildSrc/src/main/kotlin/Versions.kt

@@ -49,5 +49,7 @@ object Versions {
     // tests
     const val JUNIT_BOM = "5.13.4"
     const val MOCKITO_CORE = "5.20.0"
+    const val ASSERTJ_CORE = "3.26.3"
+    const val AWAITILITY = "4.2.1"
 
 }

eternalcore-core/build.gradle.kts

@@ -25,6 +25,8 @@ dependencies {
     annotationProcessor("org.projectlombok:lombok:${Versions.LOMBOK}")
 
     testImplementation("com.eternalcode:eternalcode-commons-bukkit:${Versions.ETERNALCODE_COMMONS}")
+    testImplementation("org.assertj:assertj-core:${Versions.ASSERTJ_CORE}")
+    testImplementation("org.awaitility:awaitility:${Versions.AWAITILITY}")
 }
 
 eternalShadow {

eternalcore-core/src/main/java/com/eternalcode/core/delay/Delay.java

@@ -0,0 +1,57 @@
+package com.eternalcode.core.delay;
+
+import com.github.benmanes.caffeine.cache.Cache;
+import com.github.benmanes.caffeine.cache.Caffeine;
+
+import java.time.Duration;
+import java.time.Instant;
+import java.util.function.Supplier;
+
+public class Delay<T> {
+
+    private final Cache<T, Instant> cache;
+    private final Supplier<Duration> defaultDelay;
+
+    private Delay(Supplier<Duration> defaultDelay) {
+        if (defaultDelay == null) {
+            throw new IllegalArgumentException("defaultDelay cannot be null");
+        }
+
+        this.defaultDelay = defaultDelay;
+        this.cache = Caffeine.newBuilder()
+            .expireAfter(new InstantExpiry<T>())
+            .build();
+    }
+
+    public void markDelay(T key, Duration delay) {
+        if (delay.isZero() || delay.isNegative()) {
+            this.cache.invalidate(key);
+        }
+
+        this.cache.put(key, Instant.now().plus(delay));
+    }
+    public void markDelay(T key) {
+        this.markDelay(key, this.defaultDelay.get());
+    }
+
+    public void unmarkDelay(T key) {
+        this.cache.invalidate(key);
+    }
+
+    public boolean hasDelay(T key) {
+        Instant delayExpireMoment = this.getExpireAt(key);
+        return Instant.now().isBefore(delayExpireMoment);
+    }
+
+    public Duration getRemaining(T key) {
+        return Duration.between(Instant.now(), this.getExpireAt(key));
+    }
+
+    private Instant getExpireAt(T key) {
+        return this.cache.asMap().getOrDefault(key, Instant.MIN);
+    }
+
+    public static <T> Delay<T> withDefault(Supplier<Duration> defaultDelay) {
+        return new Delay<>(defaultDelay);
+    }
+}

eternalcore-core/src/main/java/com/eternalcode/core/delay/InstantExpiry.java

@@ -0,0 +1,39 @@
+package com.eternalcode.core.delay;
+
+import com.github.benmanes.caffeine.cache.Expiry;
+import java.time.Duration;
+import java.time.Instant;
+import org.jetbrains.annotations.NotNull;
+
+class InstantExpiry<T> implements Expiry<@NotNull T, @NotNull Instant> {
+
+    @Override
+    public long expireAfterCreate(@NotNull T key, @NotNull Instant expireTime, long currentTime) {
+        return timeToExpire(expireTime);
+    }
+
+    @Override
+    public long expireAfterUpdate(@NotNull T key, @NotNull Instant newExpireTime, long currentTime, long currentDuration) {
+        return timeToExpire(newExpireTime);
+    }
+
+    @Override
+    public long expireAfterRead(@NotNull T key, @NotNull Instant value, long currentTime, long currentDuration) {
+        return currentDuration;
+    }
+
+    private static long timeToExpire(Instant expireTime) {
+        Duration toExpire = Duration.between(Instant.now(), expireTime);
+        if (toExpire.isNegative()) {
+            return 0;
+        }
+
+        long nanos = toExpire.toNanos();
+        if (nanos == 0) {
+            return 1;
+        }
+
+        return nanos;
+    }
+
+}

eternalcore-core/src/main/java/com/eternalcode/core/feature/afk/AfkCommand.java

@@ -37,7 +37,7 @@ class AfkCommand {
         this.noticeService = noticeService;
         this.afkSettings = afkSettings;
         this.afkService = afkService;
-        this.delay = new Delay<>(() -> this.afkSettings.afkCommandDelay());
+        this.delay = Delay.withDefault(() -> this.afkSettings.afkCommandDelay());
     }
 
     @Execute
@@ -51,7 +51,7 @@ void execute(@Sender Player player) {
         }
 
         if (this.delay.hasDelay(uuid)) {
-            Duration time = this.delay.getDurationToExpire(uuid);
+            Duration time = this.delay.getRemaining(uuid);
 
             this.noticeService
                 .create()
@@ -64,6 +64,6 @@ void execute(@Sender Player player) {
         }
 
         this.afkService.switchAfk(uuid, AfkReason.COMMAND);
-        this.delay.markDelay(uuid, this.afkSettings.afkCommandDelay());
+        this.delay.markDelay(uuid);
     }
 }

eternalcore-core/src/main/java/com/eternalcode/core/feature/helpop/HelpOpCommand.java

@@ -42,7 +42,7 @@ class HelpOpCommand {
         this.helpOpSettings = helpOpSettings;
         this.eventCaller = eventCaller;
         this.server = server;
-        this.delay = new Delay<>(() -> this.helpOpSettings.helpOpDelay());
+        this.delay = Delay.withDefault(() -> this.helpOpSettings.helpOpDelay());
     }
 
     @Execute
@@ -58,7 +58,7 @@ void execute(@Sender Player player, @Join String message) {
         }
 
         if (this.delay.hasDelay(uuid)) {
-            Duration time = this.delay.getDurationToExpire(uuid);
+            Duration time = this.delay.getRemaining(uuid);
 
             this.noticeService.create()
                 .notice(translation -> translation.helpOp().helpOpDelay())
@@ -91,7 +91,7 @@ void execute(@Sender Player player, @Join String message) {
             .notice(translation -> translation.helpOp().send())
             .send();
 
-        this.delay.markDelay(uuid, this.helpOpSettings.helpOpDelay());
+        this.delay.markDelay(uuid);
     }
 
 }

eternalcore-core/src/main/java/com/eternalcode/core/feature/randomteleport/RandomTeleportCommand.java

@@ -40,7 +40,7 @@ class RandomTeleportCommand {
         this.randomTeleportService = randomTeleportService;
         this.randomTeleportTaskService = randomTeleportTaskService;
         this.randomTeleportSettings = randomTeleportSettings;
-        this.cooldown = new Delay<>(() -> this.randomTeleportSettings.cooldown());
+        this.cooldown = Delay.withDefault(() -> this.randomTeleportSettings.cooldown());
     }
 
     @Execute
@@ -68,7 +68,7 @@ void executeSelf(@Sender Player player) {
                 this.handleTeleportSuccess(player);
             });
 
-        this.cooldown.markDelay(uuid, this.randomTeleportSettings.cooldown());
+        this.cooldown.markDelay(uuid);
     }
 
     @Execute
@@ -96,7 +96,7 @@ void executeOther(@Sender Viewer sender, @Arg Player player) {
                 this.handleAdminTeleport(sender, player);
             });
 
-        this.cooldown.markDelay(uuid, this.randomTeleportSettings.cooldown());
+        this.cooldown.markDelay(uuid);
     }
 
     private void handleTeleportSuccess(Player player) {
@@ -129,7 +129,7 @@ private boolean hasRandomTeleportDelay(Player player) {
         }
 
         if (this.cooldown.hasDelay(uniqueId)) {
-            Duration time = this.cooldown.getDurationToExpire(uniqueId);
+            Duration time = this.cooldown.getRemaining(uniqueId);
 
             this.noticeService.create()
                 .notice(translation -> translation.randomTeleport().randomTeleportDelay())

eternalcore-core/src/main/java/com/eternalcode/core/feature/repair/RepairCommand.java

@@ -29,7 +29,7 @@ class RepairCommand {
     RepairCommand(NoticeService noticeService, RepairSettings repairSettings) {
         this.noticeService = noticeService;
         this.repairSettings = repairSettings;
-        this.delay = new Delay<>(() -> this.repairSettings.repairDelay());
+        this.delay = Delay.withDefault(() -> this.repairSettings.repairDelay());
     }
 
     @Execute
@@ -73,7 +73,7 @@ void repair(@Sender Player player) {
             .player(player.getUniqueId())
             .send();
 
-        this.delay.markDelay(uuid, this.repairSettings.repairDelay());
+        this.delay.markDelay(uuid);
     }
 
     @Execute(name = "all")
@@ -117,7 +117,7 @@ void repairAll(@Sender Player player) {
             .player(player.getUniqueId())
             .send();
 
-        this.delay.markDelay(uuid, this.repairSettings.repairDelay());
+        this.delay.markDelay(uuid);
     }
 
     @Execute(name = "armor")
@@ -161,12 +161,12 @@ void repairArmor(@Sender Player player) {
             .player(player.getUniqueId())
             .send();
 
-        this.delay.markDelay(uuid, this.repairSettings.repairDelay());
+        this.delay.markDelay(uuid);
     }
 
     private boolean hasRepairDelay(UUID uuid) {
         if (this.delay.hasDelay(uuid)) {
-            Duration time = this.delay.getDurationToExpire(uuid);
+            Duration time = this.delay.getRemaining(uuid);
 
             this.noticeService
                 .create()