mirror of
https://gitee.com/myxzgzs/boyue_jnpf.git
synced 2025-08-08 15:22:43 +08:00
25 lines
670 B
JavaScript
25 lines
670 B
JavaScript
function asyncPool(poolLimit, iterable, iteratorFn) {
|
|
let i = 0;
|
|
const ret = [];
|
|
const executing = new Set();
|
|
const enqueue = function() {
|
|
if (i === iterable.length) {
|
|
return Promise.resolve();
|
|
}
|
|
const item = iterable[i++];
|
|
const p = Promise.resolve().then(() => iteratorFn(item, iterable));
|
|
ret.push(p);
|
|
executing.add(p);
|
|
const clean = () => executing.delete(p);
|
|
p.then(clean).catch(clean);
|
|
let r = Promise.resolve();
|
|
if (executing.size >= poolLimit) {
|
|
r = Promise.race(executing);
|
|
}
|
|
return r.then(() => enqueue());
|
|
};
|
|
return enqueue().then(() => Promise.all(ret));
|
|
}
|
|
|
|
module.exports = asyncPool;
|