From a5687208788bf54491e1341dbd5b2e99ee408221 Mon Sep 17 00:00:00 2001
From: Empa <42304516+ItsEmpa@users.noreply.github.com>
Date: Wed, 5 Jun 2024 13:43:45 +0200
Subject: Fix: Sea Creature Warning (#1985)

Co-authored-by: hannibal2 <24389977+hannibal00212@users.noreply.github.com>
---
 src/main/java/at/hannibal2/skyhanni/utils/TimeLimitedCache.kt | 2 ++
 src/main/java/at/hannibal2/skyhanni/utils/TimeLimitedSet.kt   | 6 ++++++
 2 files changed, 8 insertions(+)

(limited to 'src/main/java/at/hannibal2/skyhanni/utils')

diff --git a/src/main/java/at/hannibal2/skyhanni/utils/TimeLimitedCache.kt b/src/main/java/at/hannibal2/skyhanni/utils/TimeLimitedCache.kt
index caf5b7734..e17389c52 100644
--- a/src/main/java/at/hannibal2/skyhanni/utils/TimeLimitedCache.kt
+++ b/src/main/java/at/hannibal2/skyhanni/utils/TimeLimitedCache.kt
@@ -25,6 +25,8 @@ class TimeLimitedCache<K: Any, V: Any>(
 
     fun clear() = cache.invalidateAll()
 
+    fun remove(key: K) = cache.invalidate(key)
+
     fun entries(): Set<Map.Entry<K, V>> = cache.asMap().entries
 
     fun values(): Collection<V> = cache.asMap().values
diff --git a/src/main/java/at/hannibal2/skyhanni/utils/TimeLimitedSet.kt b/src/main/java/at/hannibal2/skyhanni/utils/TimeLimitedSet.kt
index aab5595da..242d06cf0 100644
--- a/src/main/java/at/hannibal2/skyhanni/utils/TimeLimitedSet.kt
+++ b/src/main/java/at/hannibal2/skyhanni/utils/TimeLimitedSet.kt
@@ -13,6 +13,12 @@ class TimeLimitedSet<T : Any>(
         cache[element] = Unit
     }
 
+    fun addIfAbsent(element: T) {
+        if (!contains(element)) add(element)
+    }
+
+    fun remove(element: T) = cache.remove(element)
+
     operator fun contains(element: T): Boolean = cache.containsKey(element)
 
     fun clear() = cache.clear()
-- 
cgit