-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathTaskQueuePC.js
55 lines (48 loc) · 1.36 KB
/
TaskQueuePC.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
export default class TaskQueuePC {
constructor(concurrency) {
this.taskQueue = []
this.consumerQueue = []
for (let i = 0; i < concurrency; i++) {
this.consumer()
}
}
consumer() {
const that = this;
new Promise((_, reject) => {
(function loop() {
that.getNextTask()
.then((task) => task())
.then(() => {
loop()
})
.catch((err) => {
reject(err)
})
})()
})
}
getNextTask() {
return new Promise((resolve) => {
if (this.taskQueue.length !== 0) {
resolve(this.taskQueue.shift())
} else {
this.consumerQueue.push(resolve)
}
})
}
runTask(task) {
return new Promise((resolve, reject) => {
const taskWrapper = () => {
const taskPromise = task()
taskPromise.then(resolve, reject)
return taskPromise;
}
if (this.consumerQueue.length !== 0) {
const consumer = this.consumerQueue.shift()
consumer(taskWrapper)
} else {
this.taskQueue.push(taskWrapper)
}
})
}
}