You've already forked Epicnabbo-Catalogus-Updated-Daily
154 lines
3.8 KiB
JavaScript
154 lines
3.8 KiB
JavaScript
'use strict';
|
|
|
|
const { AigleProxy } = require('aigle-core');
|
|
|
|
const Aigle = require('./aigle');
|
|
const { DEFAULT_LIMIT, INTERNAL, PENDING } = require('./internal/util');
|
|
const { execute, setLimit } = require('./internal/collection');
|
|
|
|
class EachLimit extends AigleProxy {
|
|
constructor(collection, limit, iterator, set = setDefault) {
|
|
super();
|
|
if (typeof limit === 'function') {
|
|
iterator = limit;
|
|
limit = DEFAULT_LIMIT;
|
|
}
|
|
this._iterator = iterator;
|
|
this._promise = new Aigle(INTERNAL);
|
|
this._index = 0;
|
|
this._limit = limit;
|
|
this._coll = undefined;
|
|
this._rest = undefined;
|
|
this._size = undefined;
|
|
this._keys = undefined;
|
|
this._result = undefined;
|
|
this._iterate = undefined;
|
|
this._callRest = undefined;
|
|
if (collection === PENDING) {
|
|
this._set = set;
|
|
this._iterate = this._callResolve;
|
|
this._callResolve = execute;
|
|
} else {
|
|
set.call(this, collection);
|
|
}
|
|
}
|
|
|
|
_execute() {
|
|
if (this._rest === 0) {
|
|
this._promise._resolve(this._result);
|
|
} else {
|
|
while (this._limit--) {
|
|
this._iterate();
|
|
}
|
|
}
|
|
return this._promise;
|
|
}
|
|
|
|
_callResolve(value) {
|
|
if (value === false) {
|
|
this._callRest = 0;
|
|
this._promise._resolve(this._result);
|
|
} else if (--this._rest === 0) {
|
|
this._promise._resolve(this._result);
|
|
} else if (this._callRest-- > 0) {
|
|
this._iterate();
|
|
}
|
|
}
|
|
|
|
_callReject(reason) {
|
|
this._callRest = 0;
|
|
this._promise._reject(reason);
|
|
}
|
|
}
|
|
|
|
module.exports = { eachLimit, EachLimit };
|
|
|
|
function setDefault(collection) {
|
|
setLimit.call(this, collection);
|
|
this._result = collection;
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* `Aigle.eachLimit` is almost same as [`Aigle.each`](https://suguru03.github.io/aigle/docs/Aigle.html#each)
|
|
* and [`Aigle.eachSeries`](https://suguru03.github.io/aigle/docs/Aigle.html#eachSeries),
|
|
* but it will work with concurrency.
|
|
* `limit` is concurrency, if it is not defined, concurrency is 8.
|
|
* @param {Array|Object} A - collection to iterate over
|
|
* @param {integer} [limit=8] - It is concurrncy, default is 8
|
|
* @param {Function} iterator
|
|
* @return {Aigle} Returns an Aigle instance
|
|
* @example
|
|
* const order = [];
|
|
* const collection = [1, 5, 3, 4, 2];
|
|
* const iterator = (num, index) => {
|
|
* return Aigle.delay(num * 10)
|
|
* .then(() => {
|
|
* order.push(num);
|
|
* return num;
|
|
* });
|
|
* };
|
|
* Aigle.eachLimit(collection, 2, iterator)
|
|
* .then(value => {
|
|
* console.log(value); // undefined
|
|
* console.log(order); // [1, 3, 5, 2, 4];
|
|
* });
|
|
*
|
|
* @example
|
|
* const order = [];
|
|
* const collection = {
|
|
* task1: 1,
|
|
* task2: 5,
|
|
* task3: 3,
|
|
* task4: 4,
|
|
* task5: 2
|
|
* };
|
|
* const iterator = (num, key) => {
|
|
* return Aigle.delay(num * 10)
|
|
* .then(() => {
|
|
* order.push(num);
|
|
* return num;
|
|
* });
|
|
* };
|
|
* Aigle.eachLimit(collection, 2, iterator)
|
|
* .then(value => {
|
|
* console.log(value); // undefined
|
|
* console.log(order); // [1, 3, 5, 2, 4];
|
|
* });
|
|
*
|
|
* @example
|
|
* const order = [];
|
|
* const collection = [1, 5, 3, 4, 2];
|
|
* const iterator = num => {
|
|
* return Aigle.delay(num * 10)
|
|
* .then(() => {
|
|
* order.push(num);
|
|
* return num;
|
|
* });
|
|
* };
|
|
* Aigle.eachLimit(collection, iterator)
|
|
* .then(value => {
|
|
* console.log(value); // undefined
|
|
* console.log(order); // [1, 2, 3, 4, 5];
|
|
* });
|
|
*
|
|
* @example
|
|
* const order = [];
|
|
* const collection = [1, 5, 3, 4, 2];
|
|
* const iterator = num => {
|
|
* return Aigle.delay(num * 10)
|
|
* .then(() => {
|
|
* order.push(num);
|
|
* return num !== 3;
|
|
* });
|
|
* };
|
|
* Aigle.eachLimit(collection, iterator)
|
|
* .then(value => {
|
|
* console.log(value); // undefined
|
|
* console.log(order); // [1, 2, 3];
|
|
* });
|
|
*/
|
|
function eachLimit(collection, limit, iterator) {
|
|
return new EachLimit(collection, limit, iterator)._execute();
|
|
}
|