blob: a093c2481c451de5b1597c66db0d02305f533113 (
plain)
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
|
const defLimit = require('os').cpus().length
const callLimit = (queue, limit = defLimit) => new Promise((res, rej) => {
let active = 0
let current = 0
const results = []
let rejected = false
const reject = er => {
if (rejected)
return
rejected = true
rej(er)
}
let resolved = false
const resolve = () => {
if (resolved || active > 0)
return
resolved = true
res(results)
}
const run = () => {
const c = current++
if (c >= queue.length) {
return resolve()
}
active ++
results[c] = queue[c]().then(result => {
active --
results[c] = result
run()
return result
}, reject)
}
for (let i = 0; i < limit; i++) {
run()
}
})
module.exports = callLimit
|