feat: add values

wopjs · Dec 30, 2024 · daf6929 · daf6929
1 parent 1de04a0
commit daf6929
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/src/weak-cache.test.ts b/src/weak-cache.test.ts
@@ -226,4 +226,24 @@ describe("WeakCache", () => {
       cache.set("key", "value");
     }).toThrowError();
   });
+
+  it("should get alive values of cache", async () => {
+    const value = { value: "value" };
+    const cache = new WeakCache<string | { key: string }, { value: string }>([
+      ["key", { value: "value" }],
+      [{ key: "key" }, { value: "value" }],
+      [{ key: "key" }, { value: "value" }],
+      [{ key: "key" }, value],
+      ["key2", value],
+    ]);
+
+    expect(cache.size).toBeGreaterThan(0);
+    expect([...cache.values()].length).toBeGreaterThan(0);
+
+    await waitGC();
+
+    expect(cache.size).toBe(1);
+    expect([...cache.values()].length).toBe(1);
+    expect(cache.values().next().value).toBe(value);
+  });
 });
diff --git a/src/weak-cache.ts b/src/weak-cache.ts
@@ -97,4 +97,13 @@ export class WeakCache<K extends {}, V extends WeakKey = WeakKey> {
     }
     return this;
   }
+
+  public *values(): IterableIterator<V> {
+    for (const ref of this._refs_.values()) {
+      const value = ref.deref();
+      if (value) {
+        yield value;
+      }
+    }
+  }
 }