办法add承受一个返回Promise的函数,同时执行的工作数量不超过两个。
这个一道面试题,限流器保障同时最多只有两个工作在执行,如果工作是异步的,会期待异步工作实现之后,才去执行其余未执行工作。
class Scheduler {
constructor(maxNum) {
this.taskList = []; //{1}
this.count = 0;
this.maxNum = maxNum;
}
async add(promiseCreator) {
if (this.count >= this.maxNum) {
await new Promise((resolve) => { //{2}
this.taskList.push(resolve)
})
}
this.count ++;
const result = await promiseCreator();
this.count --;
if (this.taskList.length > 0) {
this.taskList.shift()(); //{3}
}
return result;
}
}
const scheduler = new Scheduler(2)
const timeout = (time)=>{
return new Promise(r=>setTimeout(r, time))
}
const addTask = (time, order) => {
scheduler.add(()=>timeout(time))
.then(()=>console.log(order))
}
addTask(1000, 1)
addTask(500, 2)
addTask(300, 3)
addTask(400, 4)
//log:2 3 1 4
以上是解答,要害在add办法要怎么实现。
在{1}处定义了一个工作队列。执行add办法,如果此时运行的工作超过了最大限度,就会在{2}处await一个promise,并且把promise的resolve放进工作队列中,这样这个办法就会阻塞在这里,期待promise的实现。而只有在之前的工作实现之后,会去taskList中拿resolve去执行({3}处)。执行完resolve,方才await的promise才会持续往下执行,去执行被阻塞的工作,如此循环,直到所有工作实现。
发表回复