【字节】如何实现一个 Promise.all

转载

mb60efb0123ab42 2021-07-16 11:08:45

文章标签 面试 文章分类 代码人生

摇篮山-圣克莱尔湖国家公园，澳大利亚塔斯马尼亚州
(© Paparwin Tanupatarachai/Getty Images)

大家好，我是山月，为大家进大厂而操碎了心的小编。本文收录于 GitHub 日问: DailyQuestion[1]，每天学习五分钟，一年进入大厂中。可在右下角打开原文查看

有一次头条面试，一道手写题目是：如何手写实现 promise.all。

我从来没有想过要手写实现 promise.all 函数，稍微一想，大概就是维护一个数组，把所有 promise 给 resolve 了之后都扔进去，这有啥子好问的。没想到，一上手还稍微有点棘手。

先来看一个示例吧:

await Promise.all([1, Promise.resolve(2)])
//-> [1, 2]

await Promise.all([1, Promise.reject(2)])
//-> Throw Error: 2

传入一个数组，可包含 Promise，也可包含普通数据
数组中 Prmise 并行执行
但凡有一个 Promise 被 Reject 掉，Promise.all 失败

function pAll (promises) {
  return new Promise((resolve, reject) => {
    // 结果用一个数组维护
    const r = []
    const len = promises.length
    let count = 0
    for (let i = 0; i < len; i++) {
      // Promise.resolve 确保把所有数据都转化为 Promise
      Promise.resolve(promises[i]).then(o => {
        // 因为 promise 是异步的，保持数组一一对应
        r[i] = o;

        // 如果数组中所有 promise 都完成，则返回结果数组
        if (++count === len) {
          resolve(r)
        }
        // 当发生异常时，直接 reject
      }).catch(e => reject(e))
    }
  })
}

为了测试，实现一个 sleep 函数

const sleep = (seconds) => new Promise(resolve => setTimeout(() => resolve(seconds), seconds))

以下示例进行测试，没有问题

pAll([1, 2, 3]).then(o => console.log(o))
pAll([
  sleep(3000),
  sleep(2000),
  sleep(1000)
]).then(o => console.log(o))
pAll([
  sleep(3000),
  sleep(2000),
  sleep(1000),
  Promise.reject(10000)
]).then(o => console.log(o)).catch(e => console.log(e, '<- Error'))