login/node_modules/asyncc/lib/Queue.js
2024-01-17 19:45:14 +08:00

264 lines
No EOL
7.7 KiB
JavaScript

"use strict";
Object.defineProperty(exports, "__esModule", {
value: true
});
exports["default"] = Queue;
exports.queue = queue;
var _setImmediate2 = require("./_setImmediate");
var _PrioArray = _interopRequireDefault(require("./PrioArray"));
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) { arr2[i] = arr[i]; } return arr2; }
function _iterableToArrayLimit(arr, i) { var _i = arr == null ? null : typeof Symbol !== "undefined" && arr[Symbol.iterator] || arr["@@iterator"]; if (_i == null) return; var _arr = []; var _n = true; var _d = false; var _s, _e; try { for (_i = _i.call(arr); !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"] != null) _i["return"](); } finally { if (_d) throw _e; } } return _arr; }
function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
/**
* Run queued `items` through an asynchronous `task`.
*
* Once finishing the `task` an optional callback is called.
* While pushing to the queue, you may define a priority for execution.
* Lower values means faster execution.
*
* @name Queue
* @methodOf: module:parallel
* @class
* @param {Function} task - iterator function of type `function (item: any, cb: Function, index: Number)`
* @param {Number} concurrency - max. number of tasks running in parallel
* @example <caption>Default usage</caption>
* var arr = []
* var q = new Queue((item, cb) => {
* arr.push(item)
* cb(null, item)
* })
* // push item "one" at end of queue
* q.push('one', (err, res) => {
* console.log(res + ' finished')
* })
* // add item "two" at start of queue
* q.unshift('two', () => {
* console.log('two finished')
* })
* // called when all items in queue where processed
* q.drain(() => {
* console.log(arr)
* //> arr = ['one', 'two']
* })
* @example <caption>Using priorities</caption>
* let arr = []
*
* let q = new Queue(function (item, cb) {
* arr.push(item)
* cb()
* }, 2)
*
* q.concat([100, 101, 102], 3) // priority = 3 - last (but 2 items already processed)
* q.concat([0, 1, 2], 1) // priority = 1 - first
* q.concat([10, 11, 12], 2) // priority = 2 - second
*
* q.drain(() => {
* //> arr = [ 100, 101, 0, 1, 2, 10, 11, 12, 102 ])
* })
*/
function Queue(task, concurrency) {
this._task = task;
this._concurrency = Math.abs(concurrency || 1);
this._worker = 0;
this._paused = false;
this._items = new _PrioArray["default"]();
}
Queue.prototype = {
/**
* process items in queue
* @private
*/
_run: function _run() {
var _this = this;
var _items = this._items,
_drain = this._drain;
this._worker -= 1;
if (_items.length === 0) {
if (this._worker <= 0) {
this._worker = 0;
_drain && _drain();
}
} else {
this._worker += 1;
var _items$shift = _items.shift(),
_items$shift2 = _slicedToArray(_items$shift, 2),
item = _items$shift2[0],
cb = _items$shift2[1];
this._task(item, function (err, res) {
cb && cb(err, res);
(0, _setImmediate2._setImmediate)(function () {
// prevent RangeError: Maximum call stack size exceeded for sync tasks
_this._run();
});
});
}
},
/**
* start processing queue or add workers up to concurrency
* @private
*/
_start: function _start() {
while (!this._paused && this._worker < Math.min(this._concurrency, this._items.length)) {
this._worker += 1;
this._run();
}
return this;
},
/**
* Check if queue is paused
* @return {Boolean} `true` if paused
*/
get paused() {
return this._paused;
},
/**
* Check if queue is idle - means no items in queue and no workers running
* @return {Boolean} `true` if idle
*/
get idle() {
return !this.length && this._worker === 0;
},
/**
* Number of items waiting in the queue to get processed
* @return {Number} number of items in queue
*/
get length() {
return this._items.length;
},
/**
* Pause processing
* @return {this} for chaining
*/
pause: function pause() {
this._paused = true;
return this;
},
/**
* Resume processing
* @return {this} for chaining
*/
resume: function resume() {
this._paused = false;
return this._start();
},
/**
* Reset the queue by removing all pending items from the queue
* @return {this} for chaining
*/
reset: function reset() {
this._items.reset();
return this;
},
/**
* Number of items being processed
* @return {Number} number of items processed
*/
running: function running() {
return this._worker;
},
/**
* push `item` onto queue
* @param {Any} item
* @param {Function} [callback] - optional callback if item was processed
* @param {Number} [priority] - priority `0 ... Infinity` of the item to process. Smaller values, faster processing
* @return {this} for chaining
*/
push: function push(item, callback, priority) {
return this.concat([item], callback, priority);
},
/**
* concat `items` onto queue - fills the queue first with `items` before starting processing
* @param {Any[]} items
* @param {Function} [callback] - optional callback if single item was processed
* @param {Number} [priority] - priority `0 ... Infinity` of the item to process. Smaller values, faster processing
* @return {this} for chaining
*/
concat: function concat(items, callback, priority) {
var _this2 = this;
if (typeof callback === 'number') {
priority = callback;
callback = undefined;
}
items.forEach(function (item) {
_this2._items.push([item, callback], priority);
});
return this._start();
},
/**
* put `item` at the very beginnning of the queue
* @param {Any} item
* @param {Function} [callback] - optional callback if item was processed
* @return {this} for chaining
*/
unshift: function unshift(item, callback) {
this._items.unshift([item, callback]);
return this._start();
},
/**
* @param {Function} [callback] - optional callback called if all queue items got processed
* @return {this} for chaining
*/
drain: function drain(callback) {
this._drain = callback;
return this;
}
};
/**
* Run queued `items` through an asynchronous `task`.
*
* Once finishing the `task` an optional callback is called.
* While pushing to the queue, you may define a priority for execution.
* Lower values means faster execution.
*
* See full API here {@link Queue}.
*
* @name queue
* @memberOf module:parallel
* @static
* @method
* @param {Function} task - iterator function of type `function (item: any, cb: Function, index: Number)`
* @param {Number} concurrency - max. number of tasks running in parallel
* @return {Queue}
*/
function queue(task, concurrency) {
return new Queue(task, concurrency);
}