|
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152 |
- 'use strict';
- const pLimit = require('p-limit');
-
- class EndError extends Error {
- constructor(value) {
- super();
- this.value = value;
- }
- }
-
- // The input can also be a promise, so we await it
- const testElement = async (element, tester) => tester(await element);
-
- // The input can also be a promise, so we `Promise.all()` them both
- const finder = async element => {
- const values = await Promise.all(element);
- if (values[1] === true) {
- throw new EndError(values[0]);
- }
-
- return false;
- };
-
- const pLocate = async (iterable, tester, options) => {
- options = {
- concurrency: Infinity,
- preserveOrder: true,
- ...options
- };
-
- const limit = pLimit(options.concurrency);
-
- // Start all the promises concurrently with optional limit
- const items = [...iterable].map(element => [element, limit(testElement, element, tester)]);
-
- // Check the promises either serially or concurrently
- const checkLimit = pLimit(options.preserveOrder ? 1 : Infinity);
-
- try {
- await Promise.all(items.map(element => checkLimit(finder, element)));
- } catch (error) {
- if (error instanceof EndError) {
- return error.value;
- }
-
- throw error;
- }
- };
-
- module.exports = pLocate;
- // TODO: Remove this for the next major release
- module.exports.default = pLocate;
|