-
Notifications
You must be signed in to change notification settings - Fork 13
/
index.js
88 lines (75 loc) · 1.51 KB
/
index.js
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
function limiter (count) {
var outstanding = 0
var jobs = []
function remove () {
outstanding--
if (outstanding < count) {
dequeue()
}
}
function dequeue () {
var job = jobs.shift()
semaphore.queue = jobs.length
if (job) {
run(job.fn).then(job.resolve).catch(job.reject)
}
}
function queue (fn) {
return new Promise(function (resolve, reject) {
jobs.push({fn: fn, resolve: resolve, reject: reject})
semaphore.queue = jobs.length
})
}
function run (fn) {
outstanding++
try {
return Promise.resolve(fn()).then(function (result) {
remove()
return result
}, function (error) {
remove()
throw error
})
} catch (err) {
remove()
return Promise.reject(err)
}
}
var semaphore = function (fn) {
if (outstanding >= count) {
return queue(fn)
} else {
return run(fn)
}
}
return semaphore
}
function map (items, mapper) {
var failed = false
var limit = this
return Promise.all(items.map(function () {
var args = arguments
return limit(function () {
if (!failed) {
return mapper.apply(undefined, args).catch(function (e) {
failed = true
throw e
})
}
})
}))
}
function addExtras (fn) {
fn.queue = 0
fn.map = map
return fn
}
module.exports = function (count) {
if (count) {
return addExtras(limiter(count))
} else {
return addExtras(function (fn) {
return fn()
})
}
}