1
|
var MapCache = require('./_MapCache');
|
2
|
|
3
|
/** Error message constants. */
|
4
|
var FUNC_ERROR_TEXT = 'Expected a function';
|
5
|
|
6
|
/**
|
7
|
* Creates a function that memoizes the result of `func`. If `resolver` is
|
8
|
* provided, it determines the cache key for storing the result based on the
|
9
|
* arguments provided to the memoized function. By default, the first argument
|
10
|
* provided to the memoized function is used as the map cache key. The `func`
|
11
|
* is invoked with the `this` binding of the memoized function.
|
12
|
*
|
13
|
* **Note:** The cache is exposed as the `cache` property on the memoized
|
14
|
* function. Its creation may be customized by replacing the `_.memoize.Cache`
|
15
|
* constructor with one whose instances implement the
|
16
|
* [`Map`](http://ecma-international.org/ecma-262/7.0/#sec-properties-of-the-map-prototype-object)
|
17
|
* method interface of `clear`, `delete`, `get`, `has`, and `set`.
|
18
|
*
|
19
|
* @static
|
20
|
* @memberOf _
|
21
|
* @since 0.1.0
|
22
|
* @category Function
|
23
|
* @param {Function} func The function to have its output memoized.
|
24
|
* @param {Function} [resolver] The function to resolve the cache key.
|
25
|
* @returns {Function} Returns the new memoized function.
|
26
|
* @example
|
27
|
*
|
28
|
* var object = { 'a': 1, 'b': 2 };
|
29
|
* var other = { 'c': 3, 'd': 4 };
|
30
|
*
|
31
|
* var values = _.memoize(_.values);
|
32
|
* values(object);
|
33
|
* // => [1, 2]
|
34
|
*
|
35
|
* values(other);
|
36
|
* // => [3, 4]
|
37
|
*
|
38
|
* object.a = 2;
|
39
|
* values(object);
|
40
|
* // => [1, 2]
|
41
|
*
|
42
|
* // Modify the result cache.
|
43
|
* values.cache.set(object, ['a', 'b']);
|
44
|
* values(object);
|
45
|
* // => ['a', 'b']
|
46
|
*
|
47
|
* // Replace `_.memoize.Cache`.
|
48
|
* _.memoize.Cache = WeakMap;
|
49
|
*/
|
50
|
function memoize(func, resolver) {
|
51
|
if (typeof func != 'function' || (resolver != null && typeof resolver != 'function')) {
|
52
|
throw new TypeError(FUNC_ERROR_TEXT);
|
53
|
}
|
54
|
var memoized = function() {
|
55
|
var args = arguments,
|
56
|
key = resolver ? resolver.apply(this, args) : args[0],
|
57
|
cache = memoized.cache;
|
58
|
|
59
|
if (cache.has(key)) {
|
60
|
return cache.get(key);
|
61
|
}
|
62
|
var result = func.apply(this, args);
|
63
|
memoized.cache = cache.set(key, result) || cache;
|
64
|
return result;
|
65
|
};
|
66
|
memoized.cache = new (memoize.Cache || MapCache);
|
67
|
return memoized;
|
68
|
}
|
69
|
|
70
|
// Expose `MapCache`.
|
71
|
memoize.Cache = MapCache;
|
72
|
|
73
|
module.exports = memoize;
|