-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathindex.js
72 lines (62 loc) · 1.52 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
'use strict'
const { EventEmitter } = require('events')
class TaskQueue extends EventEmitter {
constructor(concurrent) {
super();
this.concurrent = concurrent;
}
queue = [];
concurrent = 0;
running = 0;
addTask(task) {
return new Promise((resolve, reject) => {
this.queue.push(async () => {
try {
const curr = await task()
return resolve(curr)
}
catch(err) {
return reject(err)
}
});
this.next()
});
}
async next() {
while (this.running < this.concurrent && this.queue.length !== 0) {
const curr = this.queue.shift();
try {
await curr()
}
finally {
this.running--;
this.next();
}
this.running++;
}
}
}
const delay = (ms) => {
return new Promise((resolve, reject) => {
setTimeout(() => resolve(), ms)
})
}
const taskA = async () => {
console.log("starting task a")
await delay(5000)
return Promise.resolve('a')
}
const taskB = async () => {
console.log("starting task b")
await delay(1000)
return Promise.resolve('b')
}
const taskC = async () => {
console.log("starting task c")
await delay(2000)
return Promise.resolve('c')
}
const queue = new TaskQueue(2)
queue.addTask(taskA).then(data => console.log(data))
queue.addTask(taskB).then(data => console.log(data))
queue.addTask(taskC).then(data => console.log(data))