npm
Version:
a package manager for JavaScript
60 lines • 1.81 kB
JavaScript
import * as os from 'node:os';
// availableParallelism available only since node v19, for older versions use
// cpus() cpus() can return an empty list if /proc is not mounted, use 1 in
// this case
/* c8 ignore start */
const defLimit = 'availableParallelism' in os ?
Math.max(1, os.availableParallelism() - 1)
: Math.max(1, os.cpus().length - 1);
export const callLimit = (queue, { limit = defLimit, rejectLate } = {}) => new Promise((res, rej) => {
let active = 0;
let current = 0;
const results = [];
// Whether or not we rejected, distinct from the rejection just in case the rejection itself is falsey
let rejected = false;
let rejection;
const reject = (er) => {
if (rejected)
return;
rejected = true;
rejection ??= er;
if (!rejectLate)
rej(rejection);
};
let resolved = false;
const resolve = () => {
if (resolved || active > 0)
return;
resolved = true;
res(results);
};
const run = () => {
const c = current++;
if (c >= queue.length)
return rejected ? reject() : resolve();
active++;
const step = queue[c];
/* c8 ignore start */
if (!step)
throw new Error('walked off queue');
/* c8 ignore stop */
results[c] = step()
.then(result => {
active--;
results[c] = result;
return result;
}, er => {
active--;
reject(er);
})
.then(result => {
if (rejected && active === 0)
return rej(rejection);
run();
return result;
});
};
for (let i = 0; i < limit; i++)
run();
});
//# sourceMappingURL=index.js.map