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
89
90
91
92
93
94
95
96
97
98
99
|
'use strict'
const defaultMaxRunning = 50
const limit = module.exports = function (func, maxRunning) {
const state = {running: 0, queue: []}
if (!maxRunning) maxRunning = defaultMaxRunning
return function limited () {
const args = Array.prototype.slice.call(arguments)
if (state.running >= maxRunning) {
state.queue.push({obj: this, args})
} else {
callFunc(this, args)
}
}
function callNext () {
if (state.queue.length === 0) return
const next = state.queue.shift()
callFunc(next.obj, next.args)
}
function callFunc (obj, args) {
const cb = typeof args[args.length - 1] === 'function' && args.pop()
try {
++state.running
func.apply(obj, args.concat(function () {
--state.running
process.nextTick(callNext)
if (cb) process.nextTick(() => cb.apply(obj, arguments))
}))
} catch (err) {
--state.running
if (cb) process.nextTick(() => cb.call(obj, err))
process.nextTick(callNext)
}
}
}
module.exports.method = function (classOrObj, method, maxRunning) {
if (typeof classOrObj === 'function') {
const func = classOrObj.prototype[method]
classOrObj.prototype[method] = limit(func, maxRunning)
} else {
const func = classOrObj[method]
classOrObj[method] = limit(func, maxRunning)
}
}
module.exports.promise = function (func, maxRunning) {
const state = {running: 0, queue: []}
if (!maxRunning) maxRunning = defaultMaxRunning
return function limited () {
const args = Array.prototype.slice.call(arguments)
if (state.running >= maxRunning) {
return new Promise(resolve => {
state.queue.push({resolve, obj: this, args})
})
} else {
return callFunc(this, args)
}
}
function callNext () {
if (state.queue.length === 0) return
const next = state.queue.shift()
next.resolve(callFunc(next.obj, next.args))
}
function callFunc (obj, args) {
return callFinally(() => {
++state.running
return func.apply(obj, args)
}, () => {
--state.running
process.nextTick(callNext)
})
}
function callFinally (action, fin) {
try {
return Promise.resolve(action()).then(value => {
fin()
return value
}, err => {
fin()
return Promise.reject(err)
})
} catch (err) {
fin()
return Promise.reject(err)
}
}
}
module.exports.promise.method = function (classOrObj, method, maxRunning) {
if (typeof classOrObj === 'function') {
const func = classOrObj.prototype[method]
classOrObj.prototype[method] = limit.promise(func, maxRunning)
} else {
const func = classOrObj[method]
classOrObj[method] = limit.promise(func, maxRunning)
}
}
|