You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					101 lines
				
				3.2 KiB
			
		
		
			
		
	
	
					101 lines
				
				3.2 KiB
			| 
								 
											3 years ago
										 
									 | 
							
								'use strict';
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Object.defineProperty(exports, "__esModule", {
							 | 
						||
| 
								 | 
							
								    value: true
							 | 
						||
| 
								 | 
							
								});
							 | 
						||
| 
								 | 
							
								exports.default = memoize;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _identity = require('lodash/identity');
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _identity2 = _interopRequireDefault(_identity);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _slice = require('./internal/slice');
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _slice2 = _interopRequireDefault(_slice);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _setImmediate = require('./internal/setImmediate');
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _setImmediate2 = _interopRequireDefault(_setImmediate);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _initialParams = require('./internal/initialParams');
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _initialParams2 = _interopRequireDefault(_initialParams);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _wrapAsync = require('./internal/wrapAsync');
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var _wrapAsync2 = _interopRequireDefault(_wrapAsync);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function has(obj, key) {
							 | 
						||
| 
								 | 
							
								    return key in obj;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/**
							 | 
						||
| 
								 | 
							
								 * Caches the results of an async function. When creating a hash to store
							 | 
						||
| 
								 | 
							
								 * function results against, the callback is omitted from the hash and an
							 | 
						||
| 
								 | 
							
								 * optional hash function can be used.
							 | 
						||
| 
								 | 
							
								 *
							 | 
						||
| 
								 | 
							
								 * If no hash function is specified, the first argument is used as a hash key,
							 | 
						||
| 
								 | 
							
								 * which may work reasonably if it is a string or a data type that converts to a
							 | 
						||
| 
								 | 
							
								 * distinct string. Note that objects and arrays will not behave reasonably.
							 | 
						||
| 
								 | 
							
								 * Neither will cases where the other arguments are significant. In such cases,
							 | 
						||
| 
								 | 
							
								 * specify your own hash function.
							 | 
						||
| 
								 | 
							
								 *
							 | 
						||
| 
								 | 
							
								 * The cache of results is exposed as the `memo` property of the function
							 | 
						||
| 
								 | 
							
								 * returned by `memoize`.
							 | 
						||
| 
								 | 
							
								 *
							 | 
						||
| 
								 | 
							
								 * @name memoize
							 | 
						||
| 
								 | 
							
								 * @static
							 | 
						||
| 
								 | 
							
								 * @memberOf module:Utils
							 | 
						||
| 
								 | 
							
								 * @method
							 | 
						||
| 
								 | 
							
								 * @category Util
							 | 
						||
| 
								 | 
							
								 * @param {AsyncFunction} fn - The async function to proxy and cache results from.
							 | 
						||
| 
								 | 
							
								 * @param {Function} hasher - An optional function for generating a custom hash
							 | 
						||
| 
								 | 
							
								 * for storing results. It has all the arguments applied to it apart from the
							 | 
						||
| 
								 | 
							
								 * callback, and must be synchronous.
							 | 
						||
| 
								 | 
							
								 * @returns {AsyncFunction} a memoized version of `fn`
							 | 
						||
| 
								 | 
							
								 * @example
							 | 
						||
| 
								 | 
							
								 *
							 | 
						||
| 
								 | 
							
								 * var slow_fn = function(name, callback) {
							 | 
						||
| 
								 | 
							
								 *     // do something
							 | 
						||
| 
								 | 
							
								 *     callback(null, result);
							 | 
						||
| 
								 | 
							
								 * };
							 | 
						||
| 
								 | 
							
								 * var fn = async.memoize(slow_fn);
							 | 
						||
| 
								 | 
							
								 *
							 | 
						||
| 
								 | 
							
								 * // fn can now be used as if it were slow_fn
							 | 
						||
| 
								 | 
							
								 * fn('some name', function() {
							 | 
						||
| 
								 | 
							
								 *     // callback
							 | 
						||
| 
								 | 
							
								 * });
							 | 
						||
| 
								 | 
							
								 */
							 | 
						||
| 
								 | 
							
								function memoize(fn, hasher) {
							 | 
						||
| 
								 | 
							
								    var memo = Object.create(null);
							 | 
						||
| 
								 | 
							
								    var queues = Object.create(null);
							 | 
						||
| 
								 | 
							
								    hasher = hasher || _identity2.default;
							 | 
						||
| 
								 | 
							
								    var _fn = (0, _wrapAsync2.default)(fn);
							 | 
						||
| 
								 | 
							
								    var memoized = (0, _initialParams2.default)(function memoized(args, callback) {
							 | 
						||
| 
								 | 
							
								        var key = hasher.apply(null, args);
							 | 
						||
| 
								 | 
							
								        if (has(memo, key)) {
							 | 
						||
| 
								 | 
							
								            (0, _setImmediate2.default)(function () {
							 | 
						||
| 
								 | 
							
								                callback.apply(null, memo[key]);
							 | 
						||
| 
								 | 
							
								            });
							 | 
						||
| 
								 | 
							
								        } else if (has(queues, key)) {
							 | 
						||
| 
								 | 
							
								            queues[key].push(callback);
							 | 
						||
| 
								 | 
							
								        } else {
							 | 
						||
| 
								 | 
							
								            queues[key] = [callback];
							 | 
						||
| 
								 | 
							
								            _fn.apply(null, args.concat(function () /*args*/{
							 | 
						||
| 
								 | 
							
								                var args = (0, _slice2.default)(arguments);
							 | 
						||
| 
								 | 
							
								                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;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								module.exports = exports['default'];
							 |