'use strict' module.exports = RunQueue var validate = require('aproba') function RunQueue (opts) { validate('Z|O', [opts]) if (!opts) opts = {} this.finished = false this.inflight = 0 this.maxConcurrency = opts.maxConcurrency || 1 this.queued = 0 this.queue = [] this.currentPrio = null this.currentQueue = null this.Promise = opts.Promise || global.Promise this.deferred = {} } RunQueue.prototype = {} RunQueue.prototype.run = function () { if (arguments.length !== 0) throw new Error('RunQueue.run takes no arguments') var self = this var deferred = this.deferred if (!deferred.promise) { deferred.promise = new this.Promise(function (resolve, reject) { deferred.resolve = resolve deferred.reject = reject self._runQueue() }) } return deferred.promise } RunQueue.prototype._runQueue = function () { var self = this while ((this.inflight < this.maxConcurrency) && this.queued) { if (!this.currentQueue || this.currentQueue.length === 0) { // wait till the current priority is entirely processed before // starting a new one if (this.inflight) return var prios = Object.keys(this.queue) for (var ii = 0; ii < prios.length; ++ii) { var prioQueue = this.queue[prios[ii]] if (prioQueue.length) { this.currentQueue = prioQueue this.currentPrio = prios[ii] break } } } --this.queued ++this.inflight var next = this.currentQueue.shift() var args = next.args || [] // we explicitly construct a promise here so that queue items can throw // or immediately return to resolve var queueEntry = new this.Promise(function (resolve) { resolve(next.cmd.apply(null, args)) }) queueEntry.then(function () { --self.inflight if (self.finished) return if (self.queued <= 0 && self.inflight <= 0) { self.finished = true self.deferred.resolve() } self._runQueue() }, function (err) { self.finished = true self.deferred.reject(err) }) } } RunQueue.prototype.add = function (prio, cmd, args) { if (this.finished) throw new Error("Can't add to a finished queue. Create a new queue.") if (Math.abs(Math.floor(prio)) !== prio) throw new Error('Priorities must be a positive integer value.') validate('NFA|NFZ', [prio, cmd, args]) prio = Number(prio) if (!this.queue[prio]) this.queue[prio] = [] ++this.queued this.queue[prio].push({cmd: cmd, args: args}) // if this priority is higher than the one we're currently processing, // switch back to processing its queue. if (this.currentPrio > prio) { this.currentQueue = this.queue[prio] this.currentPrio = prio } }