Memoize a curried function

Question

const f = (arg1) => (arg2) => { /* returns something */ } 

Is it possible to memoize f with regard to the 2 arguments, namely:

f(1)(2); f(1)(3); // Cache not hit f(4)(2); // Cache not hit f(1)(2); // Cache hit 

Answer 1

You could take a Map as cache and take nested maps for all following arguments.

This cache works for arbitrary count of arguments and reuses the values from the former calls.

It works by taking a curried function and an optional Map. If the map is not supplied, a new map is created which serves as base cache for all other calls of the returned closure or the final result.

The inner function takes a single argument and checks if this value is in the map.

• If not, call the curried function and check the returned value

  • if function, create a new closure over the function and a new map,

  • if no function take the result,

  as value for a new element of the map.

• Finally return the value from the map.

const cached = (fn, map = new Map()) => arg => {      const inCache = map.has(arg);      const hint = inCache ? 'in cache' : 'not in cache';        console.log(arg, hint);        if (!inCache) {          const value = fn(arg);          const result = typeof value === 'function' ? cached(value, new Map()) : value;            map.set(arg, result);      }        return map.get(arg);  };    const f = a => b => c => a * b * c; // the original curried function  const g = cached(f); // its cached variant    console.log(g(1)(2)(5)); // not not not 10  console.log(g(1)(3)(4)); //  in not not 12  console.log(g(4)(2)(3)); // not not not 24  console.log(g(1)(2)(6)); //  in  in not 12  console.log(g(4)(2)(3)); //  in  in  in 24

.as-console-wrapper { max-height: 100% !important; top: 0; }