perf[cache]: check expire cache in all access method

zfoo-project · Aug 16, 2024 · d676bc4 · d676bc4
1 parent 96485d1
commit d676bc4
Showing 1 changed file with 15 additions and 12 deletions.
diff --git a/scheduler/src/main/java/com/zfoo/scheduler/util/LazyCache.java b/scheduler/src/main/java/com/zfoo/scheduler/util/LazyCache.java
@@ -111,9 +111,24 @@ public V get(K key) {
 
 
     public void remove(K key) {
+        checkExpire();
         removeForCause(key, RemovalCause.EXPLICIT);
     }
 
+    public void forEach(BiConsumer<K, V> biConsumer) {
+        checkExpire();
+        for (var entry : cacheMap.entrySet()) {
+            biConsumer.accept(entry.getKey(), entry.getValue().value);
+        }
+    }
+
+    public int size() {
+        checkExpire();
+        return cacheMap.size();
+    }
+
+
+    // -----------------------------------------------------------------------------------------------------------------
     private void removeForCause(K key, RemovalCause removalCause) {
         if (key == null) {
             return;
@@ -134,18 +149,6 @@ private void removeForCause(List<Pair<K, V>> list, RemovalCause removalCause) {
         removeListener.accept(removeList, removalCause);
     }
 
-    public void forEach(BiConsumer<K, V> biConsumer) {
-        for (var entry : cacheMap.entrySet()) {
-            biConsumer.accept(entry.getKey(), entry.getValue().value);
-        }
-    }
-
-    public int size() {
-        return cacheMap.size();
-    }
-
-
-    // -----------------------------------------------------------------------------------------------------------------
     private void checkMaximumSize() {
         if (cacheMap.size() > backPressureSize) {
             var removeList = cacheMap.entrySet()