[面试题]基于promise的并发函数的实现

167 阅读1分钟

要求:实现一个 Parallel 函数,可以控制并发任务的数量,实现效果如下

function task (ms, name) {
  return new Promise((resolve, reject) => {
    setTimeout(_ => {
      console.log(new Date(), name);
      resolve();
    }, ms)
  })
}
console.log('task start-----------', new Date());
const para = new Parallel(3);
// 载入任务
para.addTask(task, 1000, 'a');
para.addTask(task, 2000, 'b');
para.addTask(task, 3000, 'c');
para.addTask(task, 4000, 'd');
// 开始执行
para.start();
// 并发数量为 3
// 分别于 1 2 3 5 秒后执行上述任务 
// 并发数量为 2
// 则分别于 1 2 4 6 秒后执行上述任务 

解答:

function task (ms, name) {
  return new Promise((resolve, reject) => {
    setTimeout(_ => {
      console.log(new Date(), name);
      resolve();
    }, ms)
  })
}

function Parallel (number) {
  this.number = number;
  this.currentNumber = 0;
  this.taskList = [];
  this.carryTaskList = [];
}
Parallel.prototype.addTask = function (task, ms, name) {
  this.taskList.push({
    fn: task,
    ms: ms,
    name: name
  })
}
Parallel.prototype.start = function () {
  if (this.taskList.length <= 0) {
    return;
  }
  while (this.currentNumber < this.number) {
    const newTask = this.taskList.shift();
    this.carryTaskList.push(newTask);
    this.currentNumber++;
  }
  this.carry();
}
Parallel.prototype.carry = function () {
  while (this.carryTaskList.length > 0) {
    const newTask = this.carryTaskList.shift();
    newTask.fn(newTask.ms, newTask.name).then(_ => {
      this.currentNumber--;
      this.start();
    })
  }
}

console.log('task start-----------', new Date());
const para = new Parallel(3);
para.addTask(task, 1000, 'a');
para.addTask(task, 2000, 'b');
para.addTask(task, 3000, 'c');
para.addTask(task, 4000, 'd');
para.start();