Как регулировать промисы: реализовать promise.map

Promise

Ссылка на эту статьюГора Специальность/код/про МИС…

выполнить

Предположим, что есть обещаниеgetИ ожидающий запрос на массивlist, используйте их для запроса данных. Однако, чтобы избежать чрезмерного ввода-вывода, необходимо ограничить количество одновременных

function get (i) {
  console.log('In ', i)
  return new Promise((resolve, reject) => {
    setTimeout(() => {
      resolve(i * 1000) 
      console.log('Out', i, 'Out')
    }, i * 1000)
  })
}

const list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

Написать фрагмент свободного кода, реализующего функциональность, очень просто, но также очень важно предоставить дизайнерские идеи для API. Простая реализация выглядит следующим образом, используяcountДостаточно вести счетчик количества одновременных

// 并发数量计数
let count = 0
function run () {
  if (count < 3 && list.length) {
    count+=1
    get(list.shift()).then(() => {
      count-=1 
      run()
    })
  }
}

// 限定三个并发数量
run()
run()
run()

код

Promise.map(
    Iterable<any>|Promise<Iterable<any>> input,
    function(any item, int index, int length) mapper,
    [Object {concurrency: int=Infinity} options]
) -> Promise

разработаны, чтобыBluebirdAPI относительно модульный и простой в использовании. Ключом к коду является поддержание очереди, и когда количество промисов превышает лимит, он передается в очередь для обслуживания. код показывает, как показано ниже

class Limit {
  constructor (n) {
    this.limit = n
    this.count = 0
    this.queue = []
  }

  enqueue (fn) {
    // 关键代码: fn, resolve, reject 统一管理
    return new Promise((resolve, reject) => {
      this.queue.push({ fn, resolve, reject })
    })
  }

  dequeue () {
    if (this.count < this.limit && this.queue.length) {
      // 等到 Promise 计数器小于阈值时,则出队执行
      const { fn, resolve, reject } = this.queue.shift()
      this.run(fn).then(resolve).catch(reject)
    }
  }

  // async/await 简化错误处理
  async run (fn) {
    this.count++
    // 维护一个计数器
    const value = await fn()
    this.count--
    // 执行完,看看队列有东西没
    this.dequeue()
    return value
  }

  build (fn) {
    if (this.count < this.limit) {
      // 如果没有到达阈值,直接执行
      return this.run(fn)
    } else {
      // 如果超出阈值,则先扔到队列中,等待有空闲时执行
      return this.enqueue(fn)
    }
  }
}

Promise.map = function (list, fn, { concurrency }) {
  const limit = new Limit(concurrency)
  return Promise.all(list.map((...args) => {
    return limit.build(() => fn(...args))
  }))
}

Ссылаться на

Bluebird.map

Bluebird.map(list, x => {
  return get(x)
}, {
  concurrency: 3
})

в основном для справкиconcurrencyреализация

featurist/promise-limit


Добро пожаловать, чтобы обратить внимание на мой общедоступный номерГорная Луна Путешествие, записывайте сюда мой технический рост, добро пожаловать в общение

欢迎关注公众号山月行,在这里记录我的技术成长,欢迎交流