Support clean localStorage when got excess error (#292)

subquery · May 28, 2024 · 63f207f · 63f207f
1 parent e8cc36a
commit 63f207f
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/packages/network-support/src/utils/store.ts b/packages/network-support/src/utils/store.ts
@@ -38,16 +38,23 @@ export class LocalStorageCache implements IStore {
   }
 
   async set<T>(key: string, value: T): Promise<void> {
-    const data = {
-      value,
-      expiry: this.ttl ? Date.now() + this.ttl : undefined,
-    };
-    localStorage.setItem(key, JSON.stringify(data));
+    const expiry = this.ttl ? Date.now() + this.ttl : undefined;
+    const data = { value, expiry };
+
+    try {
+      localStorage.setItem(key, JSON.stringify(data));
+    } catch {
+      await this.cleanExpiredLocalStorage();
+    }
   }
 
   async remove(key: string): Promise<void> {
     localStorage.removeItem(key);
   }
+
+  async cleanExpiredLocalStorage(): Promise<void> {
+    await Promise.all(Object.entries(localStorage).map(async ([key]) => this.get(key)));
+  }
 }
 
 export class LRUCache implements IStore {