手写一个Promise

2023-10-24  本文已影响0人  奋斗_登

通过手写 Promise,可以使我们深入理解它的内部工作原理。这有助于更好地理解异步编程、回调、事件循环等概念。
手写一个 Promise 的过程可以分为以下几个关键步骤:

1. 创建 Promise 构造函数

首先,你需要创建一个 Promise 构造函数。这个构造函数接受一个 executor 函数,这个函数有两个参数:resolve 和 reject,它们用于将 Promise 从 pending 状态转换为 fulfilled 或 rejected 状态。

2. 定义 Promise 的状态和值

在构造函数中,你需要定义 Promise 的状态(初始为 pending)和值。状态表示 Promise 的当前状态(pending、fulfilled、rejected),值表示 Promise 的最终结果值(成功时是结果值,失败时是错误原因)。

3. 实现 resolve 和 reject 函数

在构造函数中,你需要实现 resolve 和 reject 函数,它们用于将 Promise 从 pending 转换为 fulfilled 或 rejected 状态。当调用 resolve(value) 时,Promise 状态变为 fulfilled,并将 value 存储为结果值;当调用 reject(reason) 时,Promise 状态变为 rejected,并将 reason 存储为错误原因。

4. 处理异步操作

在 executor 函数中,通常会包含异步操作,例如网络请求、定时器等。你需要确保在异步操作完成后,调用 resolve 或 reject 函数来改变 Promise 的状态。

5. 实现 then 方法

then 方法是 Promise 实例的核心方法,它用于注册成功和失败的回调函数,以便在状态变为 fulfilled 或 rejected 时执行相应的操作。在 then 方法中,你需要考虑以下几点:

6. 支持链式调用

你需要返回一个新的 Promise 对象,以支持链式调用。这是通过在 then 方法内创建并返回一个新的 Promise 对象来实现的。

7. 处理回调返回值

在 then 方法中,你需要处理成功回调和失败回调的返回值。如果返回值是一个 Promise,需要等待这个 Promise 的状态变化,并根据其状态来决定新 Promise 的状态。如果返回值不是 Promise,直接将其作为新 Promise 的值。

8. 实现 catch 方法

catch 方法是 then 方法的一种简化形式,用于注册失败回调。你可以在 catch 方法内部调用 then(null, onRejected) 来实现。

9. 实现静态方法

你可以为 Promise 添加静态方法,如 Promise.resolve 和 Promise.reject,以便创建已解决或已拒绝的 Promise。另外,你还可以实现 Promise.all 和 Promise.race 等静态方法,用于处理多个 Promise 实例。

10. 错误处理

在所有涉及到异步操作的地方,要确保捕获异常并将其传递给 reject 函数,以便在 Promise 失败时能够正确处理错误。

以下是完整的代码示例:

    class MyPromise {
        constructor(executor) {
            // 初始化 Promise 的状态为 pending
            this.state = 'pending';
            // 初始化 Promise 的结果值
            this.value = undefined;
            // 初始化 Promise 的拒因(错误原因)
            this.reason = undefined;
            // 初始化成功和失败的回调队列
            this.onFulfilledCallbacks = [];
            this.onRejectedCallbacks = [];

            // 定义 resolve 函数,用于将 Promise 状态从 pending 变为 fulfilled
            const resolve = (value) => {
                if (this.state === 'pending') {
                    this.state = 'fulfilled';
                    this.value = value;
                    // 依次执行成功回调函数
                    this.onFulfilledCallbacks.forEach(callback => callback(this.value));
                }
            };

            // 定义 reject 函数,用于将 Promise 状态从 pending 变为 rejected
            const reject = (reason) => {
                if (this.state === 'pending') {
                    this.state = 'rejected';
                    this.reason = reason;
                    // 依次执行失败回调函数
                    this.onRejectedCallbacks.forEach(callback => callback(this.reason));
                }
            };

            try {
                // 执行用户传入的执行器函数,并传入 resolve 和 reject 函数
                executor(resolve, reject);
            } catch (error) {
                // 如果执行器函数抛出异常,将 Promise 状态置为 rejected
                reject(error);
            }
        }

        then(onFulfilled, onRejected) {
            // 创建一个新的 Promise 对象
            return new MyPromise((resolve, reject) => {
                // 封装成功时执行的函数
                const onFulfilledHandler = (value) => {
                    try {
                        // 如果 onFulfilled 不是函数,则将结果传递给下一个 Promise
                        if (typeof onFulfilled === 'function') {
                            const result = onFulfilled(value);
                            this.resolvePromise(result, resolve, reject);
                        } else {
                            resolve(value);
                        }
                    } catch (error) {
                        reject(error);
                    }
                };

                // 封装失败时执行的函数
                const onRejectedHandler = (reason) => {
                    try {
                        // 如果 onRejected 不是函数,则将错误传递给下一个 Promise
                        if (typeof onRejected === 'function') {
                            const result = onRejected(reason);
                            this.resolvePromise(result, resolve, reject);
                        } else {
                            reject(reason);
                        }
                    } catch (error) {
                        reject(error);
                    }
                };

                // 根据当前 Promise 的状态执行不同的处理函数
                if (this.state === 'fulfilled') {
                    // 当前 Promise 已经是 fulfilled 状态,执行成功处理函数
                    onFulfilledHandler(this.value);
                } else if (this.state === 'rejected') {
                    // 当前 Promise 已经是 rejected 状态,执行失败处理函数
                    onRejectedHandler(this.reason);
                } else if (this.state === 'pending') {
                    // 当前 Promise 是 pending 状态,将成功和失败处理函数分别添加到回调队列
                    this.onFulfilledCallbacks.push(onFulfilledHandler);
                    this.onRejectedCallbacks.push(onRejectedHandler);
                }
            });
        }

        catch(onRejected) {
            return this.then(null, onRejected);
        }

        resolvePromise(result, resolve, reject) {
            if (result === this) {
                return reject(new TypeError('Chaining cycle detected'));
            }

            let called = false;
            if (result instanceof MyPromise) {
                // 如果返回值是一个 Promise,则等待其状态变化
                result.then(
                    value => {
                        if (called) return;
                        called = true;
                        this.resolvePromise(value, resolve, reject);
                    },
                    reason => {
                        if (called) return;
                        called = true;
                        reject(reason);
                    }
                );
            } else {
                // 如果返回值不是 Promise,则直接将其作为新 Promise 的结果值
                resolve(result);
            }
        }

        static resolve(value) {
            if (value instanceof MyPromise) return value;
            return new MyPromise(resolve => resolve(value));
        }

        static reject(reason) {
            return new MyPromise((resolve, reject) => reject(reason));
        }

        static all(promises) {
            return new MyPromise((resolve, reject) => {
                const results = [];
                let completed = 0;

                if (promises.length === 0) {
                    resolve(results);
                } else {
                    promises.forEach((promise, index) => {
                        MyPromise.resolve(promise).then(result => {
                            results[index] = result;
                            completed++;

                            if (completed === promises.length) {
                                resolve(results);
                            }
                        }).catch(reject);
                    });
                }
            });
        }

        static race(promises) {
            return new MyPromise((resolve, reject) => {
                promises.forEach(promise => {
                    MyPromise.resolve(promise).then(resolve).catch(reject);
                });
            });
        }
    }

示例代码

    // 示例用法
    const promise = new MyPromise((resolve, reject) => {
        // 异步操作
        setTimeout(() => {
            resolve('Promise resolved');
            // 或者
            //reject('Promise rejected');
        }, 500);
    });

    promise
        .then(value => {
            console.log(value);
            return 'New value';
        })
        .then(value => {
            console.log(value);
        })
        .catch(reason => {
            console.error(reason);
        });
上一篇下一篇

猜你喜欢

热点阅读