Promise 重写 （第一部分）

学习关键语句：
promise 重写

写在前面

重新学习了怎么重写 promise ， 我觉得最重要的就是要有思路，不然有些 A+ 规范是完全想不到的

开始

重写函数的过程中, 最重要的是有思路
我们从哪里获取重写思路? 从正常的代码中
我们先看正常的代码

const p1 = new Promise((resolve, reject) => {
  resolve();
  console.log("1");
}).then(() => {
  console.log("2");
});

我们的思路就来了:

1. promise 是一个类
2. 初始化的时候传入的是一个函数, 函数里面有两个参数, resolve, reject
3. 这说明类的构造器中有这两个函数声明
4. 实例化后的每一个 promise 都有自己的状态, 说明有实例属性 state

想到这里马上开始重写, 完成以上 1\2\3\4 点

// 1. promise 是一个类
class MyPromise {
  constructor(executor) {
    // 4. 实例化后的每一个 promise 都有自己的状态, 说明有实例属性 state
    this.state = "pending";
    // 3. 这说明类的构造器中有这两个函数声明
    let resolve = () => {};
    let reject = () => {};
    // 2. 初始化的时候传入的是一个函数, 函数里面有两个参数, resolve, reject
    executor(resolve, reject);
  }
}
let p = new MyPromise((resolve, reject) => {});

接下来我们知道 resolve 和 reject 函数中有参数 value 和 reason
所以我们马上就又有思路了:

5. 执行器函数中调用 resolve 会传入参数, reject 同样也是
6. 需要一个 then 方法来链式调用, 所以 then 方法是一个原型方法
7. then 方法中有两个参数, 一个成功回调一个失败回调

完成以上 5\6\7 点

class MyPromise {
  constructor(executor) {
    this.state = "pending";
    // 5. 执行器函数中调用 resolve 会传入参数, reject 同样也是
    let resolve = (value) => {};
    let reject = (reason) => {};

    executor(resolve, reject);
  }
  // 6. 需要一个 then 方法来链式调用, 所以 then 方法是一个原型方法
  // 7. then 方法中有两个参数, 一个成功回调一个失败回调
  then(onFulfilled, onRejected) {}
}
let p = new MyPromise((resolve, reject) => {});

我们知道只有在构造器函数中调用了 resolve 或者 reject 方法后才会进入 then 方法, 同时我们也知道一旦调用 resolve 或者 reject 方法就会改变 promise 实例的状态, 所以我们接下来的思路就是:

8. 调用 resolve 或者 reject 方法后改变状态
9. 成功或者失败回调需要拿到对应的 resolve 和 reject 函数中的参数, 这需要我们在实例上新增相应的属性存储
10. 在 resolve 和 reject 方法中要赋值给实例属性
11. 在 then 方法中判断状态调用方法

完成以上 8\9\10\11 点

class MyPromise {
  constructor(executor) {
    this.state = "pending";
    // 9. 成功或者失败回调需要拿到对应的 resolve 和 reject 函数中的参数, 这需要我们在实例上新增相应的属性存储
    this.value = undefined;
    this.reason = undefined;
    let resolve = (value) => {
      // 8. 调用 resolve 或者 reject 方法后改变状态
      this.state = "fulfilled";
      // 10. 在 resolve 和 reject 方法中要赋值给实例属性
      this.value = value;
    };
    let reject = (reason) => {
      // 8. 调用 resolve 或者 reject 方法后改变状态
      this.state = "rejected";
      // 10. 在 resolve 和 reject 方法中要赋值给实例属性
      this.reason = reason;
    };

    executor(resolve, reject);
  }
  then(onFulfilled, onRejected) {
    // 11. 在 then 方法中判断状态调用方法
    if (this.state === "fulfilled") {
      onFulfilled(this.value);
    }
    if (this.state === "rejected") {
      onRejected(this.reason);
    }
  }
}
let p = new MyPromise((resolve, reject) => {});

好完成到这里的时候, 不涉及异步的操作我们已经完成了, 现在我们尝试使用一下我们重写的 promise

const p1 = new MyPromise2((resolve, reject) => {
  resolve(2);
  console.log("1");
}).then((value) => {
  console.log(value);
});

接下来考虑异步的问题, 即如果执行器函数中存在异步的话, 进入 then 方法时状态其实还是 pending , 所以需要考虑当进入 then 方法时状态为 pending 就是异步, 我们要将异步时候的回调函数保存起来当真正执行的时候再调用 , 由于同一个 promise 可以多次调用 then 方法 , 所以需要用一个数组来保存所有可能的回调函数 , 我们从这出发 , 给出以下思路:

12. 在实例上添加属性来保存成功回调和失败回调
13. 在 then 方法中遇到 pending 状态就是有异步情况 , 真正的执行回调将不在这里发生 , 存到实例的属性上

完成以上 12\13 点

class MyPromise {
  constructor(executor) {
    this.state = "pending";
    this.value = undefined;
    this.reason = undefined;
    // 12. 在实例上添加属性来保存成功回调和失败回调
    this.onFulfilledCallbacks = [];
    this.onRejectedCallbacks = [];
    let resolve = (value) => {
      this.state = "fulfilled";
      this.value = value;
    };
    let reject = (reason) => {
      this.state = "rejected";
      this.reason = reason;
    };

    executor(resolve, reject);
  }
  then(onFulfilled, onRejected) {
    if (this.state === "fulfilled") {
      onFulfilled(this.value);
    }
    if (this.state === "rejected") {
      onRejected(this.reason);
    }
    // 13. 在 then 方法中遇到 pending 状态就是有异步情况, 真正的执行回调将不在这里发生, 存到实例的属性上
    if (this.state === "pending") {
      this.onFulfilledCallbacks.push(() => {
        onFulfilled(this.value);
      });
      this.onRejectedCallbacks.push(() => {
        onRejected(this.reason);
      });
    }
  }
}
let p = new MyPromise((resolve, reject) => {});

那么真正的回调函数执行到底在哪里呢? 我们来看一下一个出现了异步的情况代码, 看以下代码:

由于构造器函数中出现一个定时器, 所以会在 1 秒后才打印出 2

const p1 = new Promise((resolve, reject) => {
  setTimeout(() => {
    resolve(2);
  }, 1000);
  console.log("1");
}).then((value) => {
  console.log(value);
});

上面这个代码中能执行回调函数的地方只有在 resolve 函数中了, 所以我们就又有思路了:

14. 在 resolve 和 reject 方法中执行回调函数

完成以上 14 点

class MyPromise {
  constructor(executor) {
    this.state = "pending";
    this.value = undefined;
    this.reason = undefined;
    this.onFulfilledCallbacks = [];
    this.onRejectedCallbacks = [];
    let resolve = (value) => {
      this.state = "fulfilled";
      this.value = value;
      // 14. 在 resolve 和 reject 方法中执行回调函数
      this.onFulfilledCallbacks.forEach((fn) => fn());
    };
    let reject = (reason) => {
      this.state = "rejected";
      this.reason = reason;
      // 14. 在 resolve 和 reject 方法中执行回调函数
      this.onRejectedCallbacks.forEach((fn) => fn());
    };

    executor(resolve, reject);
  }
  then(onFulfilled, onRejected) {
    if (this.state === "fulfilled") {
      onFulfilled(this.value);
    }
    if (this.state === "rejected") {
      onRejected(this.reason);
    }
    if (this.state === "pending") {
      this.onFulfilledCallbacks.push(() => {
        onFulfilled(this.value);
      });
      this.onRejectedCallbacks.push(() => {
        onRejected(this.reason);
      });
    }
  }
}
let p = new MyPromise((resolve, reject) => {});

我们知道 promise 实例是可以链式调用的, 这意味每一个方法的最后都会返回一个新的 promise 来提供给下一次调用 promise 原型方法, 所以我们就考虑到在 then 方法中直接返回一个新的 promise 对象, 同时, 由于执行器函数执行是同步代码, 所以我们可以直接将 then 中的代码放入到新的 promise 中, 但是我们知道链式调用 then 方法时我们可以拿到上一次 then 返回的值, 所以在成功回调返回的值, 我们将再次使用新的 promise 中的 resolve 方法去调用, 以下就是这次的思路:

15. 在 then 方法中新声明一个 promise 对象并返回, 将 then 方法中代码放入执行器函数中
16. 将第一次 then 的回调的结果赋值给一个变量, 使用新声明的 promise 来调用 resolve 和 reject 方法

完成以上 15\16 点

class MyPromise {
  constructor(executor) {
    this.state = "pending";
    this.value = undefined;
    this.reason = undefined;
    this.onFulfilledCallbacks = [];
    this.onRejectedCallbacks = [];
    let resolve = (value) => {
      this.state = "fulfilled";
      this.value = value;
      this.onFulfilledCallbacks.forEach((fn) => fn());
    };
    let reject = (reason) => {
      this.state = "rejected";
      this.reason = reason;
      this.onRejectedCallbacks.forEach((fn) => fn());
    };

    executor(resolve, reject);
  }
  then(onFulfilled, onRejected) {
    // 15. 在 then 方法中新声明一个 promise 对象并返回, 将 then 方法中代码放入执行器函数中
    const p2 = new MyPromise((resolve, reject) => {
      let x;
      if (this.state === "fulfilled") {
        // 16. 将第一次 then 的回调的结果赋值给一个变量, 使用新声明的 promise 来调用 resolve 方法
        x = onFulfilled(this.value);
        resolve(x);
      }
      if (this.state === "rejected") {
        // 16. 将第一次 then 的回调的结果赋值给一个变量, 使用新声明的 promise 来调用 resolve 方法
        x = onRejected(this.reason);
        reject(x);
      }
      if (this.state === "pending") {
        this.onFulfilledCallbacks.push(() => {
          // 16. 将第一次 then 的回调的结果赋值给一个变量, 使用新声明的 promise 来调用 resolve 方法
          x = onFulfilled(this.value);
          resolve(x);
        });
        this.onRejectedCallbacks.push(() => {
          // 16. 将第一次 then 的回调的结果赋值给一个变量, 使用新声明的 promise 来调用 resolve 方法
          x = onRejected(this.reason);
          reject(x);
        });
      }
    });
    return p2;
  }
}
let p = new MyPromise((resolve, reject) => {});

好的这样就已经完成了链式调用的问题了, 但是现在又有一个新的问题了, 那就是如果第一次的回调函数返回的是一个 promise 对象呢? 那就需要对 x 进行判断和处理, 我们思路如下:

17. 声明一个 resolvePromise 专门处理判断 x 的问题, 根据 promise A+ 规范, 返回的 promise 不能与新增的 promise 相同, 在 resolvePromise 方法中需要传入 p2 进行判断
18. 由于 p2 此时没有初始化完毕所以无法调用, 使用定时器包裹延后执行
19. 由于 resolvePromise 中可能会抛出错误, 所以使用 try.catch 包裹
20. 同样的, 在构造器中, 对执行器函数使用 try.catch 包裹

完成以上 17\18\19\20 点

class MyPromise {
  constructor(executor) {
    this.state = "pending";
    this.value = undefined;
    this.reason = undefined;
    this.onFulfilledCallbacks = [];
    this.onRejectedCallbacks = [];
    let resolve = (value) => {
      this.state = "fulfilled";
      this.value = value;
      this.onFulfilledCallbacks.forEach((fn) => fn());
    };
    let reject = (reason) => {
      this.state = "rejected";
      this.reason = reason;
      this.onRejectedCallbacks.forEach((fn) => fn());
    };
    // 20. 同样的, 在构造器中, 对执行器函数使用 try.catch 包裹
    try {
      executor(resolve, reject);
    } catch (err) {
      reject(err);
    }
  }
  then(onFulfilled, onRejected) {
    const p2 = new MyPromise((resolve, reject) => {
      let x;
      if (this.state === "fulfilled") {
        // 18. 由于 p2 此时没有初始化完毕所以无法调用, 使用定时器包裹延后执行
        setTimeout(() => {
          // 19. 由于 resolvePromise 中可能会抛出错误, 所以使用 try.catch 包裹
          try {
            x = onFulfilled(this.value);
            // 17. 声明一个 resolvePromise 专门处理判断 x 的问题, 根据 promise A+ 规范, 返回的 promise 不能与新增的 promise 相同, 在 resolvePromise 方法中需要传入 p2 进行判断
            resolvePromise(p2, x, resolve, reject);
          } catch (err) {
            reject(err);
          }
        }, 0);
      }
      if (this.state === "rejected") {
        // 18. 由于 p2 此时没有初始化完毕所以无法调用, 使用定时器包裹延后执行
        setTimeout(() => {
          // 19. 由于 resolvePromise 中可能会抛出错误, 所以使用 try.catch 包裹
          try {
            x = onRejected(this.reason);
            // 17. 声明一个 resolvePromise 专门处理判断 x 的问题, 根据 promise A+ 规范, 返回的 promise 不能与新增的 promise 相同, 在 resolvePromise 方法中需要传入 p2 进行判断
            resolvePromise(p2, x, resolve, reject);
          } catch (err) {
            reject(err);
          }
        }, 0);
      }
      if (this.state === "pending") {
        this.onFulfilledCallbacks.push(() => {
          // 18. 由于 p2 此时没有初始化完毕所以无法调用, 使用定时器包裹延后执行
          setTimeout(() => {
            // 19. 由于 resolvePromise 中可能会抛出错误, 所以使用 try.catch 包裹
            try {
              x = onFulfilled(this.value);
              // 17. 声明一个 resolvePromise 专门处理判断 x 的问题, 根据 promise A+ 规范, 返回的 promise 不能与新增的 promise 相同, 在 resolvePromise 方法中需要传入 p2 进行判断
              resolvePromise(p2, x, resolve, reject);
            } catch (err) {
              reject(err);
            }
          }, 0);
        });
        this.onRejectedCallbacks.push(() => {
          // 18. 由于 p2 此时没有初始化完毕所以无法调用, 使用定时器包裹延后执行
          setTimeout(() => {
            // 19. 由于 resolvePromise 中可能会抛出错误, 所以使用 try.catch 包裹
            try {
              x = onRejected(this.reason);
              // 17. 声明一个 resolvePromise 专门处理判断 x 的问题, 根据 promise A+ 规范, 返回的 promise 不能与新增的 promise 相同, 在 resolvePromise 方法中需要传入 p2 进行判断
              resolvePromise(p2, x, resolve, reject);
            } catch (err) {
              reject(err);
            }
          }, 0);
        });
      }
    });
    return p2;
  }
}
// 17. 声明一个 resolvePromise 专门处理判断 x 的问题, 根据 promise A+ 规范, 返回的 promise 不能与新增的 promise 相同, 在 resolvePromise 方法中需要传入 p2 进行判断
function resolvePromise(p2, x, resolve, reject) {
  if (p2 === x) {
    return reject(new Error("Error: p2 is x"));
  }
}
let p = new MyPromise((resolve, reject) => {});

接下来我们要继续完善 resolvePromise 中的逻辑, 我们需要判断 x 到底是不是 promise 对象, 我们使用是否是带 then 属性的对象来判断, 是就是, 不是就不是, 当是 promise 对象后, 直接调用 x 的 then 方法, 在成功回调和失败回调中分别调用 resolvePromise 传入的 resolve 和 reject

21. x 需要是个对象或者函数
22. x 为普通数据类型就直接 resolve
23. 获取 x 的 then 属性, 是函数才是 promise 对象
24. 获取 x.then 可能会报错, 使用 try.catch 包裹

完成以上 21\22\23\24 点

class MyPromise {
  constructor(executor) {
    this.state = "pending";
    this.value = undefined;
    this.reason = undefined;
    this.onFulfilledCallbacks = [];
    this.onRejectedCallbacks = [];
    let resolve = (value) => {
      this.state = "fulfilled";
      this.value = value;
      this.onFulfilledCallbacks.forEach((fn) => fn());
    };
    let reject = (reason) => {
      this.state = "rejected";
      this.reason = reason;
      this.onRejectedCallbacks.forEach((fn) => fn());
    };
    try {
      executor(resolve, reject);
    } catch (err) {
      reject(err);
    }
  }
  then(onFulfilled, onRejected) {
    const p2 = new MyPromise((resolve, reject) => {
      let x;
      if (this.state === "fulfilled") {
        setTimeout(() => {
          try {
            x = onFulfilled(this.value);
            resolvePromise(p2, x, resolve, reject);
          } catch (err) {
            reject(err);
          }
        }, 0);
      }
      if (this.state === "rejected") {
        setTimeout(() => {
          try {
            x = onRejected(this.reason);
            resolvePromise(p2, x, resolve, reject);
          } catch (err) {
            reject(err);
          }
        }, 0);
      }
      if (this.state === "pending") {
        this.onFulfilledCallbacks.push(() => {
          setTimeout(() => {
            try {
              x = onFulfilled(this.value);
              resolvePromise(p2, x, resolve, reject);
            } catch (err) {
              reject(err);
            }
          }, 0);
        });
        this.onRejectedCallbacks.push(() => {
          setTimeout(() => {
            try {
              x = onRejected(this.reason);
              resolvePromise(p2, x, resolve, reject);
            } catch (err) {
              reject(err);
            }
          }, 0);
        });
      }
    });
    return p2;
  }
}
function resolvePromise(p2, x, resolve, reject) {
  if (p2 === x) {
    return reject(new Error("Error: p2 is x"));
  }
  // 21. x 需要是个对象或者函数
  if ((typeof x === "object" && x !== null) || typeof x === "function") {
    // 24. 获取 x.then 可能会报错, 使用 try.catch 包裹
    try {
      // 23. 获取 x 的 then 属性, 是函数才是 promise 对象
      let then = x.then;
      if (typeof then === "function") {
        then.call(
          x,
          (y) => {
            resolve(y);
          },
          (r) => {
            reject(r);
          }
        );
      } else {
        resolve(x);
      }
    } catch (err) {
      reject(err);
    }
  } else {
    // 22. x 为普通数据类型就直接 resolve
    resolve(x);
  }
}
let p = new MyPromise((resolve, reject) => {});

好, 这样看起来已经很完美了, 返回的是 promise 对象我们也能解决了, 但是紧接着又有一个问题, 要是里面的 promise 再返回一个 promise 呢? 我们就想到用递归来解决这个问题, 同时我们还要解决一个问题, 那就是一个 promise 的执行器函数中只会执行遇到的第一个 resolve 或者 reject 方法, 我们用新增变量来避免重复调用

25. 当每次返回的都是 promise 对象时, 我们递归调用判断 x 的类型的函数
26. 声明一个变量来防止构造器函数中重复执行 resolve 和 reject 方法
27. 每次进入 then 方法时, 如果没有参数, 就直接返回一个返回参数的函数

class MyPromise {
  constructor(executor) {
    this.state = "pending";
    this.value = undefined;
    this.reason = undefined;
    this.onFulfilledCallbacks = [];
    this.onRejectedCallbacks = [];
    let resolve = (value) => {
      this.state = "fulfilled";
      this.value = value;
      this.onFulfilledCallbacks.forEach((fn) => fn());
    };
    let reject = (reason) => {
      this.state = "rejected";
      this.reason = reason;
      this.onRejectedCallbacks.forEach((fn) => fn());
    };
    try {
      executor(resolve, reject);
    } catch (err) {
      reject(err);
    }
  }
  then(onFulfilled, onRejected) {
    // 27. 每次进入 then 方法时, 如果没有参数, 就直接返回一个返回参数的函数
    onFulfilled =
      typeof onFulfilled === "function" ? onFulfilled : (value) => value;
    onRejected =
      typeof onRejected === "function"
        ? onRejected
        : (reason) => {
            throw reason;
          };
    const p2 = new MyPromise((resolve, reject) => {
      let x;
      if (this.state === "fulfilled") {
        setTimeout(() => {
          try {
            x = onFulfilled(this.value);
            resolvePromise(p2, x, resolve, reject);
          } catch (err) {
            reject(err);
          }
        }, 0);
      }
      if (this.state === "rejected") {
        setTimeout(() => {
          try {
            x = onRejected(this.reason);
            resolvePromise(p2, x, resolve, reject);
          } catch (err) {
            reject(err);
          }
        }, 0);
      }
      if (this.state === "pending") {
        this.onFulfilledCallbacks.push(() => {
          setTimeout(() => {
            try {
              x = onFulfilled(this.value);
              resolvePromise(p2, x, resolve, reject);
            } catch (err) {
              reject(err);
            }
          }, 0);
        });
        this.onRejectedCallbacks.push(() => {
          setTimeout(() => {
            try {
              x = onRejected(this.reason);
              resolvePromise(p2, x, resolve, reject);
            } catch (err) {
              reject(err);
            }
          }, 0);
        });
      }
    });
    return p2;
  }
}
function resolvePromise(p2, x, resolve, reject) {
  // 26. 声明一个变量来防止构造器函数中重复执行 resolve 和 reject 方法
  let called = false;
  if (p2 === x) {
    return reject(new Error("Error: p2 is x"));
  }
  if ((typeof x === "object" && x !== null) || typeof x === "function") {
    try {
      let then = x.then;
      if (typeof then === "function") {
        then.call(
          x,
          (y) => {
            if (called) return;
            called = true;
            // 25. 当每次返回的都是 promise 对象时, 我们递归调用判断 x 的类型的函数
            resolvePromise(p2, y, resolve, reject);
          },
          (r) => {
            if (called) return;
            called = true;
            reject(r);
          }
        );
      } else {
        resolve(x);
      }
    } catch (err) {
      if (called) return;
      called = true;
      reject(err);
    }
  } else {
    resolve(x);
  }
}
let p = new MyPromise((resolve, reject) => {});

总结

ok 这样就重写完了最基本的 promise , 之后我们再来补充上 promise 上面的方法包括 catch \ race \ all 等等

我觉得虽然挺难的 ， 但是如果你有思路的话，顺着思路写下来其实是没问题的

结束

学习之路漫漫啊~