并行处理async / await函数队列

时间:2018-05-01 03:10:41

标签: javascript async-await es6-promise

我正在尝试并行处理队列。我有一个对象数组,我需要并行地将函数func应用于每个元素。我正在制作链的并行度级别如下:

async function() {
  const pickUpNextTask = () => {
    if (this.internalQueue.length) {
      return this.func(this.internalQueue.shift())
    }
  }
  const startChain = () => {
    return Promise.resolve().then(function next() {
      console.log('before then(next)')
      return pickUpNextTask().then(next)
    })
  }
  let chains = []
  for (let k = 0; k < this.parallelism; k += 1) {
    chains.push(startChain())
  }
  await Promise.all(chains)
  this.drain()
}

它不像我想的那样工作。当队列为空时,pickUpNextTask()最终返回undefined,因此没有。

如何解决这个问题?

1 个答案:

答案 0 :(得分:0)

原来,这很容易:

  const pickUpNextTask = () => {
    if (this.internalQueue.length) {
      return this.func(this.internalQueue.shift())
    } else {
      return Promise.reject()
    }
  }
  const startChain = () => {
    return Promise.resolve()
            .then(function next() {
              return pickUpNextTask()
                .then(next)
                .catch(() => {
                  return Promise.resolve()
                })
            })
  }