add weakMemoizeObj

theKashey · May 6, 2019 · c78198a · c78198a
1 parent e7f0ad0
commit c78198a
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/src/objectTrie.js b/src/objectTrie.js
@@ -1,8 +1,8 @@
-import {weakMemoizeArray} from "./weakMemoize";
+import {weakMemoizeObj} from "./weakMemoize";
 
 export const EDGE = 'EDGE';
 
-export const pushObjTrie = (root, line) => {
+const pushObjTrie = (root, line) => {
   const path = line.split('.');
   let node = root;
   const lastIndex = path.length - 1;
@@ -24,4 +24,4 @@ const buildObjTrie = (lines) => {
   return root;
 };
 
-export const memoizedBuildTrie = weakMemoizeArray(buildObjTrie);
+export const memoizedBuildTrie = weakMemoizeObj(buildObjTrie);
diff --git a/src/weakMemoize.js b/src/weakMemoize.js
@@ -16,6 +16,18 @@ export const weakMemoizeArray = fn => {
   }
 }
 
+export const weakMemoizeObj = fn => {
+  let cache = new WeakMap();
+  return arg => {
+    if (cache.has(arg)) {
+      return cache.get(arg);
+    }
+    const ret = fn(arg);
+    cache.set(arg, ret);
+    return ret;
+  };
+};
+
 export const weakMemoizeWalk = fn => {
   let cache = new WeakMap();
   return (a, b, node) => {
@@ -35,5 +47,5 @@ export const weakMemoizeWalk = fn => {
       b,
     });
     return ret;
-  }
+  };
 };