blob: 8e66b8505194b39c74ba5731d3eb730394280d50 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
'use strict';
import identity from 'lodash-es/identity';
import rest from 'lodash-es/rest';
import has from 'lodash-es/has';
import setImmediate from './internal/setImmediate';
export default function memoize(fn, hasher) {
var memo = Object.create(null);
var queues = Object.create(null);
hasher = hasher || identity;
var memoized = rest(function memoized(args) {
var callback = args.pop();
var key = hasher.apply(null, args);
if (has(memo, key)) {
setImmediate(function() {
callback.apply(null, memo[key]);
});
} else if (has(queues, key)) {
queues[key].push(callback);
} else {
queues[key] = [callback];
fn.apply(null, args.concat([rest(function(args) {
memo[key] = args;
var q = queues[key];
delete queues[key];
for (var i = 0, l = q.length; i < l; i++) {
q[i].apply(null, args);
}
})]));
}
});
memoized.memo = memo;
memoized.unmemoized = fn;
return memoized;
}
|