Generalize unaryMemoize to efficiently support undefined results

tc39 · Jul 12, 2023 · 80ab7d6 · 80ab7d6
1 parent c1c31de
commit 80ab7d6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/emitter.ts b/src/emitter.ts
@@ -185,10 +185,10 @@ function unaryMemoize<K, V>(fn: (arg: K) => V, prepopulate: K[] = []) {
   const cache = new Map(prepopulate.map(arg => [arg, fn(arg)]));
   return (arg: K) => {
     let value = cache.get(arg);
-    if (!value) {
+    if (value === undefined && !cache.has(arg)) {
       value = fn(arg);
       cache.set(arg, value);
     }
-    return value;
+    return value as V;
   };
 }