思考题(面试真题)
实现一个带并发限制的异步调度器 Scheduler,保证同时运行的任务最多有N个。完善下面代码中的 Scheduler 类,使得以下程序能正确输出:
class Scheduler {add(promiseCreator) { ... }// ...
}const timeout = (time) => new Promise(resolve => {setTimeout(resolve, time)
})const scheduler = new Scheduler(n)
const addTask = (time, order) => {scheduler.add(() => timeout(time)).then(() => console.log(order))
}addTask(1000, '1')
addTask(500, '2')
addTask(300, '3')
addTask(400, '4')// 打印顺序是:2 3 1 4
当N=2时
流程分析:
整个的完整执行流程:
- 起始1、2两个任务开始执行;
- 500ms时,2任务执行完毕,输出2,任务3开始执行;
- 800ms时,3任务执行完毕,输出3,任务4开始执行;
- 1000ms时,1任务执行完毕,输出1,此时只剩下4任务在执行;
- 1200ms时,4任务执行完毕,输出4;
设置一个变量储存正在执行的任务个数和等待队列,当正在执行的任务个数少于并发器的最大值(资源),就执行新任务。当正在执行的任务执行完毕,等待队列里有待执行任务,就拎出来执行
关键点:
1、判断任务是执行还是进入等待
2、执行完一个任务后,加入新任务进行执行
3、add返回的是一个Promise,因为后面有then调用,所以需要用resolve返回结果
class Scheduler {constructor(n){//初始化this.max=n;//并发器的最大值this.exeCount= 0; //正在执行的任务个数this.awaitQueue = new Array(); // 表示等待队列}add(promiseCreator) { //接收一个函数,执行后,会返回一个promisevar returnpromise=new Promise(resolve => {//需要执行resolve,才能把Promise结果返回外面,给then调用,//什么时候执行resolve呢?//执行promiseCreator之后,因为我们需要执行promiseCreator,即执行当前任务之后,用把resolve结果返回给外面的thenpromiseCreator.resolve=resolve //将resolve挂到promiseCreator.resolve上if(this.exeCount<this.max){//判断当前执行任务的是否小于最大值,如果大于就进入等待队列this.exePromiseCreator(promiseCreator)//开始执行promiseCreator}else{this.awaitQueue.push(promiseCreator)}})return returnpromise}exePromiseCreator(promiseCreator){this.exeCount++promiseCreator().then(res=>{//关键点:当执行完当前任务,就将结果返回出去,结果给then,用resolve返回,同时执行完,加入下一个等待执行的任务// resolve() promiseCreator.resolve()//相当于执行add函数里的resolve()//执行完毕,exeCount减一this.exeCount--if(this.awaitQueue.length){//如果等待队列里有值let joinTask=this.awaitQueue.shift()this.exePromiseCreator(joinTask)}})}
}const timeout = (time) => new Promise(resolve => {setTimeout(resolve, time)
})const scheduler = new Scheduler(2)
const addTask = (time, order) => {scheduler.add(() => timeout(time)).then(() => console.log(order))
}addTask(10000, '1')
addTask(5000, '2')
addTask(300, '3')
addTask(400, '4')// output: 2 3 1 4