22

✏️ Editing: limit.js

"use strict";

var ensureNaturalNumber  = require("type/natural-number/ensure")
  , ensurePlainFunction  = require("type/plain-function/ensure")
  , ensure               = require("type/ensure")
  , defineFunctionLength = require("../lib/private/define-function-length");

module.exports = function (limit, callback) {
	limit = ensure(
		["limit", limit, ensureNaturalNumber, { min: 1 }],
		["callback", callback, ensurePlainFunction]
	)[0];

var Promise = this, ongoingCount = 0, pending = [];
	var onSuccess, onFailure;

onSuccess = function (value) {
		release();
		return value;
	};

onFailure = function (exception) {
		release();
		throw exception;
	};

return defineFunctionLength(callback.length, function () {
		if (ongoingCount >= limit) {
			var context = this, args = arguments;
			return new Promise(function (resolve, reject) {
				pending.push({
					context: context,
					arguments: args,
					resolve: resolve,
					reject: reject
				});
			});
		}
		++ongoingCount;
		try {
			return Promise.resolve(callback.apply(this, arguments)).then(onSuccess, onFailure);
		} catch (exception) { return onFailure(exception); }
	});
};

MR SKB