Fix in-memory cache implementation

DreamExposure · Sep 9, 2023 · 9d598ba · 9d598ba
1 parent 0eed2a7
commit 9d598ba
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/core/src/main/kotlin/org/dreamexposure/discal/core/cache/JdkCacheRepository.kt b/core/src/main/kotlin/org/dreamexposure/discal/core/cache/JdkCacheRepository.kt
@@ -16,12 +16,12 @@ class JdkCacheRepository<K : Any, V>(override val ttl: Duration) : CacheReposito
     }
 
     override suspend fun put(key: K, value: V) {
-        cache[key] = Pair(Instant.now(), value)
+        cache[key] = Pair(Instant.now().plus(ttl), value)
     }
 
     override suspend fun get(key: K): V? {
         val cached = cache[key] ?: return null
-        if (Instant.now().isAfter(cached.first)) {
+        if (cached.first.isExpiredTtl()) {
             evict(key)
             return null
         }
@@ -41,6 +41,6 @@ class JdkCacheRepository<K : Any, V>(override val ttl: Duration) : CacheReposito
     }
 
     private fun evictOld() {
-        cache.forEach { (key, pair) -> if (Duration.between(pair.first, Instant.now()) >= ttl) cache.remove(key) }
+        cache.forEach { (key, pair) -> if (pair.first.isExpiredTtl()) cache.remove(key) }
     }
 }