Update the expiration time in OnDemandMuteCache#restart

A248 · Feb 29, 2024 · b75a4fa · b75a4fa
1 parent b680805
commit b75a4fa
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 11 deletions.
diff --git a/...re/src/main/java/space/arim/libertybans/core/selector/cache/AlwaysAvailableMuteCache.java b/...re/src/main/java/space/arim/libertybans/core/selector/cache/AlwaysAvailableMuteCache.java
@@ -44,7 +44,6 @@
 import java.util.UUID;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.TimeUnit;
-import java.util.function.Function;
 import java.util.function.Predicate;
 
 /**
@@ -84,8 +83,7 @@ public AlwaysAvailableMuteCache(Configs configs, FactoryOfTheFuture futuresFacto
 	@Override
 	public void startup() {
 		installCache((expirationTime, expirationSemantic) -> {
-			ConcurrentHashMap<MuteCacheKey, Entry> map = new ConcurrentHashMap<>();
-			Cache cache = new Cache(map, expirationTime);
+			Cache cache = new Cache(new ConcurrentHashMap<>(), expirationTime);
 			cache.startPurgeTask();
 			this.cache = cache;
 		});
@@ -127,18 +125,14 @@ private CentralisedFuture<MuteAndMessage> formatMessage(Punishment mute) {
 
 	@Override
 	public CentralisedFuture<Optional<Punishment>> getCachedMute(UUID uuid, NetworkAddress address) {
-		return cacheRequestTo(uuid, address, MuteAndMessage::mute);
+		return cacheRequest(new MuteCacheKey(uuid, address))
+				.thenApply((opt) -> opt.map(MuteAndMessage::mute));
 	}
 
 	@Override
 	public CentralisedFuture<Optional<Component>> getCachedMuteMessage(UUID uuid, NetworkAddress address) {
-		return cacheRequestTo(uuid, address, MuteAndMessage::message);
-	}
-
-	private <T> CentralisedFuture<Optional<T>> cacheRequestTo(UUID uuid, NetworkAddress address,
-															  Function<MuteAndMessage, T> toWhich) {
 		return cacheRequest(new MuteCacheKey(uuid, address))
-				.thenApply((muteAndMessage) -> muteAndMessage.map(toWhich));
+				.thenApply((opt) -> opt.map(MuteAndMessage::message));
 	}
 
 	private CentralisedFuture<Optional<MuteAndMessage>> cacheRequest(MuteCacheKey cacheKey) {

diff --git a/bans-core/src/main/java/space/arim/libertybans/core/selector/cache/OnDemandMuteCache.java b/bans-core/src/main/java/space/arim/libertybans/core/selector/cache/OnDemandMuteCache.java
@@ -86,7 +86,7 @@ public void startup() {
 
 	@Override
 	public void restart() {
-		cache.synchronous().invalidateAll();
+		startup();
 	}
 
 	@Override