优化

‘liusuyi’

2023-10-21 ae92b6823499cfab77201b4033922fa373fb37a3

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
'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;