File size: 713 Bytes
bc20498 |
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 38 |
export const memoize = ( fn, keyFn ) => {
if( !keyFn ){
keyFn = function(){
if( arguments.length === 1 ){
return arguments[0];
} else if( arguments.length === 0 ){
return 'undefined';
}
let args = [];
for( let i = 0; i < arguments.length; i++ ){
args.push( arguments[ i ] );
}
return args.join( '$' );
};
}
let memoizedFn = function(){
let self = this;
let args = arguments;
let ret;
let k = keyFn.apply( self, args );
let cache = memoizedFn.cache;
if( !(ret = cache[ k ]) ){
ret = cache[ k ] = fn.apply( self, args );
}
return ret;
};
memoizedFn.cache = {};
return memoizedFn;
};
|