wishthis/node_modules/p-map/index.js

82 lines
1.6 KiB
JavaScript
Raw Normal View History

2022-01-21 08:28:41 +00:00
'use strict';
2023-08-17 09:47:40 +00:00
const AggregateError = require('aggregate-error');
2022-01-21 08:28:41 +00:00
2023-08-17 09:47:40 +00:00
module.exports = async (
iterable,
mapper,
{
concurrency = Infinity,
stopOnError = true
} = {}
) => {
return new Promise((resolve, reject) => {
if (typeof mapper !== 'function') {
throw new TypeError('Mapper function is required');
}
2022-01-21 08:28:41 +00:00
2023-08-17 09:47:40 +00:00
if (!((Number.isSafeInteger(concurrency) || concurrency === Infinity) && concurrency >= 1)) {
throw new TypeError(`Expected \`concurrency\` to be an integer from 1 and up or \`Infinity\`, got \`${concurrency}\` (${typeof concurrency})`);
}
2022-01-21 08:28:41 +00:00
2023-08-17 09:47:40 +00:00
const result = [];
const errors = [];
const iterator = iterable[Symbol.iterator]();
let isRejected = false;
let isIterableDone = false;
let resolvingCount = 0;
let currentIndex = 0;
2022-01-21 08:28:41 +00:00
2023-08-17 09:47:40 +00:00
const next = () => {
if (isRejected) {
return;
}
2022-01-21 08:28:41 +00:00
2023-08-17 09:47:40 +00:00
const nextItem = iterator.next();
const index = currentIndex;
currentIndex++;
2022-01-21 08:28:41 +00:00
2023-08-17 09:47:40 +00:00
if (nextItem.done) {
isIterableDone = true;
2022-01-21 08:28:41 +00:00
2023-08-17 09:47:40 +00:00
if (resolvingCount === 0) {
if (!stopOnError && errors.length !== 0) {
reject(new AggregateError(errors));
} else {
resolve(result);
}
}
2022-01-21 08:28:41 +00:00
2023-08-17 09:47:40 +00:00
return;
2022-01-21 08:28:41 +00:00
}
2023-08-17 09:47:40 +00:00
resolvingCount++;
2022-01-21 08:28:41 +00:00
2023-08-17 09:47:40 +00:00
(async () => {
try {
const element = await nextItem.value;
result[index] = await mapper(element, index);
2022-01-21 08:28:41 +00:00
resolvingCount--;
next();
2023-08-17 09:47:40 +00:00
} catch (error) {
if (stopOnError) {
isRejected = true;
reject(error);
} else {
errors.push(error);
resolvingCount--;
next();
}
2022-01-21 08:28:41 +00:00
}
2023-08-17 09:47:40 +00:00
})();
};
2022-01-21 08:28:41 +00:00
2023-08-17 09:47:40 +00:00
for (let i = 0; i < concurrency; i++) {
next();
2022-01-21 08:28:41 +00:00
2023-08-17 09:47:40 +00:00
if (isIterableDone) {
break;
}
2022-01-21 08:28:41 +00:00
}
2023-08-17 09:47:40 +00:00
});
};