关于javascript:JS实现一个限流器

办法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才会持续往下执行,去执行被阻塞的工作,如此循环,直到所有工作实现。

评论

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

这个站点使用 Akismet 来减少垃圾评论。了解你的评论数据如何被处理