lepu-test-platform-web/node_modules/run-queue/queue.js

96 lines
2.7 KiB
JavaScript

'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
}
}