fix(memoize): 主动清理过期数据

planjs · Mar 17, 2022 · 67438fe · 67438fe
1 parent b5398a6
commit 67438fe
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/src/promise/memoize.ts b/src/promise/memoize.ts
@@ -24,10 +24,17 @@ function memoize<FnType extends AnyPromiseFN>(
 
   const returnFn = (async (...args: Parameters<FnType>): Promise<ThenReturn<FnType>> => {
     const key = harsher(...args);
-    if (memos.has(key)) {
-      if (!timeoutMs || Date.now() < memos.get(key)!.expiration) {
-        return memos.get(key)!.value;
-      }
+
+    if (timeoutMs) {
+      memos.forEach((item, k) => {
+        if (item.expiration > Date.now()) {
+          memos.delete(k);
+        }
+      });
+    }
+
+    if (!timeoutMs || memos.has(key)) {
+      return memos.get(key)!.value;
     }
 
     if (queues.has(key)) {