fix: add isLoaded and lazy load

amplitude · Aug 16, 2024 · 6cfcbdc · 6cfcbdc
1 parent f26b06f
commit 6cfcbdc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sdk/src/main/java/com/amplitude/experiment/storage/Cache.kt b/sdk/src/main/java/com/amplitude/experiment/storage/Cache.kt
@@ -16,12 +16,15 @@ internal class LoadStoreCache<V>(
     private val onLoad: (() -> Unit)? = null,
 ) {
     private val cache: MutableMap<String, V> = mutableMapOf()
+    private var isLoaded = false
 
     fun get(key: String): V? {
+        if (!isLoaded) load()
         return cache[key]
     }
 
     fun getAll(): Map<String, V> {
+        if (!isLoaded) load()
         return HashMap(cache)
     }
 
@@ -62,6 +65,7 @@ internal class LoadStoreCache<V>(
         clear()
         putAll(values)
         onLoad?.invoke()
+        isLoaded = true
     }
 
     fun store(values: MutableMap<String, V> = cache) {