2025-06-30 09:38:03 +08:00

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;