从零手写Promise——then的链式调用

上一节一步步实现了一个Promise的基本功能,先回顾一下上节的代码

function changePromiseStatus(tar, key, value) {
    if(tar.status === "pending") {
        tar[key] = value;
        // 如果key为value代码是fulfilled,如果key是reason是rejected
        if(key === 'value') {
            tar.status = "fulfilled";
            // 开始执行搜集的方法
            tar.resolveCallbacks.forEach(fn => fn());
        } else {
            tar.status = "rejected";
            // 开始执行搜集的方法
            tar.rejectCallbacks.forEach(fn => fn());
        }
    }
}

class MyPromise {
    status = "pending"; // 状态
    value = undefined;  // 使用resove时记录的值
    reason = undefined; // 使用reject时记录的出错原因
    resolveCallbacks = []; // 用来收集状态为fulfilled时的执行方法
    rejectCallbacks = []; // 用来收集状态rejected时的执行方法

    constructor(executor) {
        // 创建resolve方法
        const resolve = res => {
            changePromiseStatus(this, 'value', res);            
        }
        // 创建reject方法
        const reject = err => {
            changePromiseStatus(this, 'reason', err);         
        }
        // 尝试执行executor,并将resolve,reject作为实参传给外部使用
        // 如果创建实例时没有传executor方法时则直接报错,走reject
        try {
            executor(resolve, reject);
        } catch(err) {
            reject(err);
        }
    }

    then(onFulfilled, onRejected) {
        switch(this.status) {
            case "fulfilled":
                onFulfilled(this.value);
                break;
            case "rejected":
                onRejected(this.reason);
                break;
            case "pending": {
                // 当状态为pending时收集待执行的方法
                this.resolveCallbacks.push(() => onFulfilled(this.value)); // 为了便于传参在外层包裹了一次方法              
                this.rejectCallbacks.push(() => onRejected(this.reason));
            }
        }
    }
}

但是光只实现基本功能还不行,因为使用Promise的原因很大程度上是为了避免回调函数嵌套太多产生回调地狱,先抛开async await不谈,Promise的链式调用在某一程序上大大避免了这种回调地狱的局面。如:

// 以jquery为例的ajax回调地狱
$.get("/getdata", res => {
    $.get("/getdata2", res2 => {
        $.get("/getdata3", res3 => {

        })
    })
})
// 链式调用
p.then(res => {
    console.log(res);
    return 400;
}).then(res => {
    console.log(res);
    return 300;
}).then(res => {
    console.log(res)
});

链式虽然看着还是有点不太舒服,但是已经比回调函数一层嵌一层看着舒服多了,接下来的内容就是开始实现这种链式调用了。

首先在写代码之前,先说明一下链式调用的特征:
1、要使then方法之后再使用then方法就得保证then方法里返回一个新的Promise对象
2、下一个then可以接收上一个then处理之后返回的结果
3、下一个then方法是要等待上一个then方法执行完之后才开始执行

根据以上特征下列就开始then方法的改造了,如下:

then(onFulfilled, onRejected) {
        // 需要返回一个新的promise对象
        return new MyPromise((resolve, reject) => {
            let x = null;
            switch(this.status) {
                case "fulfilled":
                    // 等待并获取当前执行的结果
                    x = onFulfilled(this.value);
                    // 将其结果传入下一次then
                    resolve(x);                
                    break;
                case "rejected":
                    x = onRejected(this.reason);
                    resolve(x);   
                    break;
                case "pending": {
                    // 当状态为pending时收集待执行的方法
                    this.resolveCallbacks.push(() => {
                        x = onFulfilled(this.value)
                        resolve(x);
                    }); // 为了便于值参在外层包裹了一次方法              
                    this.rejectCallbacks.push(() => {
                        x = onRejected(this.reason);
                        resolve(x);
                    });
                }
            }
        });
    }

接下来测试一下以下代码看能否按照预想打印出结果

// 使用
const p = new MyPromise((resolve, reject) => {
    resolve(500);
});

p.then(res => {
    console.log(res);
    return 400;
}).then(res => {
    console.log(res);
    return 300;
}).then(res => {
    console.log(res)
});

运行结果依次打印500 400 300,结果正确。
简单的链式调用已经完成,但是如果then方法中处理的有异步任务呢,如下:

p.then(res => {
    setTimeout(() => {
        console.log(res);
    }, 400);
    return 400;
}).then(res => {
    setTimeout(() => {
        console.log(res);
    }, 300);
    return 300;
}).then(res => {
    console.log(res)
});

结果为:300 400 500
这时就会结果虽然这几个值都打印出来了,但是顺序不符合我们的预期,预期的效果应该是上一次then执行完再执行下一次then,所以我们要的效果是500 400 300的结果。
既然处理过程是异步了,那我们将异步的过程使用promise包起来试试呢,如下:

p.then(res => {
    return new MyPromise((resolve, reject) => {
        console.log(res);
        resolve(400);
    });
}).then(res => {
    return new MyPromise((resolve, reject) => {
        console.log(res);
        resolve(300);
    });
}).then(res => {
    console.log(res)
});

理解很丰满,现实很骨感,运行结果如下:

image.png

顺序好像是对了,但是值又不对了,后面两次明显是一个promise对象,那看来还要对返回的结果做判断了,如果是一个promise对象,则自行then之后才把结果传给后面去,既然这样那这部分代码干脆就抽离用一个函数完成比较好。

// 处理then的处理结果
function handleThenResult(result, resolve, reject) {
    // 判断是否为MyPromise对象 
    // 如果是就将then之后的结果分别用resolve,reject传到后面的then中
    // 如果是一个普通的数据,则直接使用resolve传到后面then中
    if(result instanceof MyPromise) {
        // 便于理解的写法
        //result.then(res => resolve(res), err => reject(err))
        result.then(resolve, reject); // 上面代码精简后的写法,上面本质都是使用resolve,reject方法传值
    } else {
        resolve(result);
    }
}
then(onFulfilled, onRejected) {
        // 需要返回一个新的promise对象
        return new MyPromise((resolve, reject) => {
            let x = null;
            switch(this.status) {
                case "fulfilled":
                    // 等待并获取当前执行的结果
                    handleThenResult(onFulfilled(this.value), resolve, reject);
                    break;
                case "rejected":
                    handleThenResult(onRejected(this.reason), resolve, reject);
                    break;
                case "pending": {
                    // 当状态为pending时收集待执行的方法
                    this.resolveCallbacks.push(() => {
                        handleThenResult(onFulfilled(this.value), resolve, reject);
                    }); // 为了便于传参在外层包裹了一次方法              
                    this.rejectCallbacks.push(() => {
                        handleThenResult(onRejected(this.reason), resolve, reject);
                    });
                }
            }
        });
    }

此时再尝试运行如下代码

p.then(res => {
    return new MyPromise((resolve, reject) => {
        console.log(res);
        resolve(400);
    });
}).then(res => {
    return new MyPromise((resolve, reject) => {
        console.log(res);
        resolve(300);
    });
}).then(res => {
    console.log(res)
});

结果是依次等待并输出:500 400 300,已达到预期效果了,下面将完整的代码展示出来

// 处理promise对象实例状态改变的时候
function changePromiseStatus(tar, key, value) {
    if(tar.status === "pending") {
        tar[key] = value;
        // 如果key为value代码是fulfilled,如果key是reason是rejected
        if(key === 'value') {
            tar.status = "fulfilled";
            // 开始执行搜集的方法
            tar.resolveCallbacks.forEach(fn => fn());
        } else {
            tar.status = "rejected";
            // 开始执行搜集的方法
            tar.rejectCallbacks.forEach(fn => fn());
        }
    }
}

// 处理then的处理结果
function handleThenResult(result, resolve, reject) {
    // 判断是否为MyPromise对象 
    // 如果是就将then之后的结果分别用resolve,reject传到后面的then中
    // 如果是一个普通的数据,则直接使用resolve传到后面then中
    if(result instanceof MyPromise) {
        // 便于理解的写法
        //result.then(res => resolve(res), err => reject(err))
        result.then(resolve, reject); // 上面代码精简后的写法,上面本质都是使用resolve,reject方法传值
    } else {
        resolve(result);
    }
}

class MyPromise {
    status = "pending"; // 状态
    value = undefined;  // 使用resove时记录的值
    reason = undefined; // 使用reject时记录的出错原因
    resolveCallbacks = []; // 用来收集状态为fulfilled时的执行方法
    rejectCallbacks = []; // 用来收集状态rejected时的执行方法

    constructor(executor) {
        // 创建resolve方法
        const resolve = res => {
            changePromiseStatus(this, 'value', res);            
        }
        // 创建reject方法
        const reject = err => {
            changePromiseStatus(this, 'reason', err);         
        }
        // 尝试执行executor,并将resolve,reject作为实参传给外部使用
        // 如果创建实例时没有传executor方法时则直接报错,走reject
        try {
            executor(resolve, reject);
        } catch(err) {
            reject(err);
        }
    }

    then(onFulfilled, onRejected) {
        // 需要返回一个新的promise对象
        return new MyPromise((resolve, reject) => {
            let x = null;
            switch(this.status) {
                case "fulfilled":
                    // 等待并获取当前执行的结果
                    handleThenResult(onFulfilled(this.value), resolve, reject);
                    break;
                case "rejected":
                    handleThenResult(onRejected(this.reason), resolve, reject);
                    break;
                case "pending": {
                    // 当状态为pending时收集待执行的方法
                    this.resolveCallbacks.push(() => {
                        handleThenResult(onFulfilled(this.value), resolve, reject);
                    }); // 为了便于传参在外层包裹了一次方法              
                    this.rejectCallbacks.push(() => {
                        handleThenResult(onRejected(this.reason), resolve, reject);
                    });
                }
            }
        });
    }
}

到此then的链式调用已经处理完成了,下一节将带上promise的四个静态方法(all, race, reject, resolve)的处理。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,245评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,749评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,960评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,575评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,668评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,670评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,664评论 3 415
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,422评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,864评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,178评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,340评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,015评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,646评论 3 323
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,265评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,494评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,261评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,206评论 2 352

推荐阅读更多精彩内容