小而美的Promise库——promiz源码浅析

时间 2020-02-10

标签小而美的 promise promiz 源码浅析繁體版

原文原文链接

背景

在上一篇博客[[译]前端基础知识储备——Promise/A+规范](https://segmentfault.com/a/11...，咱们介绍了Promise/A+规范的具体条目。在本文中，咱们来选择了promiz，让你们来看下一个具体的Promise库的内部代码是如何运做的。javascript

promiz是一个体积很小的promise库（官方介绍约为913 bytes (gzip)），做为一个ES2015标准中的Promise的polyfill，实现了诸如resolve、all和race等API。前端

知识储备

咱们在这里简单回顾一下Promise/A+的主要关键点，若是须要了解详细内容的同窗，能够阅读个人上一篇博客。java

Promise有三个状态，分别为pending、fulfilled和rejected，且只能从pending到fulfilled或者rejected，没有其余的流转方式。
Promise的返回值是一个新的Promise，缘由见上一条。
传递给then函数的两个回调函数，有且仅有一次机会被执行（即执行了onfulfilled就不会执行onrejected函数，且只执行一次）。

代码实现与分析

异步执行器

在介绍Promise以前，咱们先介绍一下异步执行器。在Promise中，咱们须要一个异步的执行器来异步执行咱们的回调函数。在规范中提到，一般状况下，咱们可使用微任务（nextTick）或者宏任务（setTimeout）来实现。可是，若是咱们须要兼容Web Worker这种状况的话，咱们可能还须要一些更多的方式来处理。具体代码以下：git

var queueId = 1
var queue = {}
var isRunningTask = false

// 使用postMessage来执行异步函数
if (!global.setImmediate)
    global.addEventListener('message', function (e) {
        if (e.source == global) {
            if (isRunningTask)
                nextTick(queue[e.data])
            else {
                isRunningTask = true
                try {
                    queue[e.data]()
                } catch (e) {}

                delete queue[e.data]
                isRunningTask = false
            }
        }
    })

/**
 * 异步执行方法
 * @param {function} fn 须要执行的回调函数
 */
function nextTick(fn) {
    if (global.setImmediate) setImmediate(fn)
    // 若是在Web Worker中使用如下方法
    else if (global.importScripts) setTimeout(fn)
    else {
        queueId++
        queue[queueId] = fn
        global.postMessage(queueId, '*')
    }
}

以上代码比较简单，咱们简单说明下：github

在代码中，promiz使用了setImmediate、setTimeout和postMessage这三个方法来执行异步函数，其中：segmentfault
- setImmedeate，只有IE实现了该方法，在执行完队列中的代码后当即执行。
- PostMessage，新增的H5中的方法。
- setTimeout，兼容性最佳，能够适用各类场景。

所以，在promiz的这段代码中，有必定的兼容性问题，应该把setTimeout放到最后做为一个兜底策略，不然没法在老浏览器中执行。数组

构造函数

说完了异步函数执行器，咱们来看下promise的构造函数。promise

首先咱们来看下内存数据，咱们须要存储当前promise的状态、成功的值或者失败的缘由、下一个promise的引用和成功与失败的回调函数。所以，咱们须要如下变量：浏览器

// states
// 0: pending
// 1: resolving
// 2: rejecting
// 3: resolved
// 4: rejected
var self = this,
    state = 0, // promise状态
    val = 0, // success callback返回值
    next = [], // 返回的新的promise对象
    fn, er; // then方法中的成功回调函数和失败回调函数

在存储完相关数据后，咱们来看下构造函数。异步

function Deferred(resolver) {
    ...
    self = this;
    try {
        if (typeof resolver == 'function')
            resolver(self['resolve'], self['reject'])
    } catch (e) {
        self['reject'](e)
    }
}

构造函数很是简单，除了声明相关的函数，就只有执行传入的callback而已。固然，若是咱们不是链式调用的第一个promise，那么咱们会没有resolver参数，所以不须要在此执行，咱们会在then函数执行resolve方法。

下面咱们来看下上面提到的处理函数resovle和reject。

self['resolve'] = function (v) {
    fn = self.fn
    er = self.er
    if (!state) {
        val = v
        state = 1

        nextTick(fire)
    }
    return self
}

self['reject'] = function (v) {
    fn = self.fn
    er = self.er
    if (!state) {
        val = v
        state = 2

        nextTick(fire)

    }
    return self
}

self['then'] = function (_fn, _er) {
    if (!(this._d == 1))
        throw TypeError()

    var d = new Deferred()

    d.fn = _fn
    d.er = _er
    if (state == 3) {
        d.resolve(val)
    }
    else if (state == 4) {
        d.reject(val)
    }
    else {
        next.push(d)
    }

    return d
}

在resolve和reject这两个函数中，都是改变了内部promise的状态，给定了参数值，同时异步触发了fire函数。而then方法，则是生成了一个新的Deferred对象，而且完成了相关的初始化（执行完then方法咱们就会获得这个新生成的Deferred对象，也就是一个新的Promise）；当前一个promise到达resolved状态时，不须要等待则直接出发resolve方法，rejected状态时也同样。那么，让咱们来看下fire方法究竟是作什么的呢？

function fire() {

    // 检测是否是一个thenable对象
    var ref;
    try {
        ref = val && val.then
    } catch (e) {
        val = e
        state = 2
        return fire()
    }

    thennable(ref, function () {
        state = 1
        fire()
    }, function () {
        state = 2
        fire()
    }, function () {
        try {
            if (state == 1 && typeof fn == 'function') {
                val = fn(val)
            }

            else if (state == 2 && typeof er == 'function') {
                val = er(val)
                state = 1
            }
        } catch (e) {
            val = e
            return finish()
        }

        if (val == self) {
            val = TypeError()
            finish()
        } else thennable(ref, function () {
            finish(3)
        }, finish, function () {
            finish(state == 1 && 3)
        })

    })
}

从上面的代码来看，fire函数只是判断了ref是否是一个thenable对象，而后调用了thenable函数，传递了3个回调函数。那么这些回调函数究竟是作什么用的呢？咱们须要来看下thenable函数的实现代码。

// ref:指向thenable对象的`then`函数
// cb, ec, cn : successCallback, failureCallback, notThennableCallback
function thennable(ref, cb, ec, cn) {
    // Promises can be rejected with other promises, which should pass through
    if (state == 2) {
        return cn()
    }
    if ((typeof val == 'object' || typeof val == 'function') && typeof ref == 'function') {
        try {

            // cnt变量用来保证成功和失败的回调函数总共只会被执行一次
            var cnt = 0
            ref.call(val, function (v) {
                if (cnt++) return
                val = v
                cb()
            }, function (v) {
                if (cnt++) return
                val = v
                ec()
            })
        } catch (e) {
            val = e
            ec()
        }
    } else {
        cn()
    }
};

在thenable函数中，若是判断当前的promise的状态是处于rejecting时，会直接执行cn，也就是将reject状态传递下去。而若是当ref不是一个thenable对象的then函数时（那么此时值为undefined），那么就会直接执行cn。

经过fire函数传递的三个callback咱们能够看到，cn是在promise的状态改变时，针对特定的状态来触发相对应的onfulfilled或者onrejected回调函数。

只有当ref是一个thenable时（传递给resolve的是一个promise），代码才会进入上面的try catch逻辑中。

Promise执行流程

看完了上面的各部分代码，我相信你们可能对整个执行流程仍然不够熟悉，下面，咱们将这些流程拼接起来，经过几个完整的流程来讲明下。

链式调用第一个Promise

当咱们声明一个promise式，咱们会传入一个resolver。此时，整个Deferred对象的state是0。若是咱们在resolver里面调用了resolve方法，那么咱们的state就会变成1，而后出发fire函数注册到thenable函数里面的第三个回调函数，从而将值传递给下一个thenable。当thenable的then函数执行完成（即咱们看到的Promise后面跟着的then函数执行完成之后），咱们的state才会变成3，也就是说上一个Promise才会结束，返回一个新的Promise。

链式调用非第一个Promise

若是不是第一个Promise，那么咱们就没有resolver参数。所以，咱们的resolve方法并非经过在resolver中进行调用的，而是将回调函数fn注册进来，在上一个Promise完成后主动调用执行的。也就是说，咱们在上一个Promise执行完then函数而且返回一个新的Promise时，咱们这个返回的Promise就已经进入了resolving的状态。

给`resolve`传递一个Promise

在Promise/A+规范中，若是咱们给resolve传递一个promise，那么咱们的经过resolve获取到的值就是传递进去的这个promise返回的值。固然，咱们也必须等待做为参数的这个promise处理完成后，才会处理外面的这个promise。

在promiz的代码中，咱们若是经过resolve接收到一个promise，那么咱们在fire函数中就会吧promise.then的引用传递给thenable函数。在thenable函数中，咱们会将咱们当前promise须要执行的onfulfilled和onrejected封装成一个函数，传递给做为参数的promise的then函数。所以，看成为参数的promise执行任意结果的回调函数时，就会将参数传递给外层的promise，执行对应的回调函数。

全局执行方法

Promise.all

让咱们先看代码。

Deferred.all = function (arr) {
    if (!(this._d == 1))
        throw TypeError()

    if (!(arr instanceof Array))
        return Deferred.reject(TypeError())

    var d = new Deferred()

    function done(e, v) {
        if (v)
            return d.resolve(v)

        if (e)
            return d.reject(e)

        var unresolved = arr.reduce(function (cnt, v) {
            if (v && v.then)
                return cnt + 1
            return cnt
        }, 0)

        if (unresolved == 0)
            d.resolve(arr)

        arr.map(function (v, i) {
            if (v && v.then)
                v.then(function (r) {
                    arr[i] = r
                    done()
                    return r
                }, done)
        })
    }

    done()

    return d
}

在Promise.all中，咱们使用了一个计数器来进行统计，在每个Promise后面都增长一个then函数用于增长计数。当Promise成功时则计数+1。当整个数组中的Promise都已经进入resolved状态时，咱们才会执行thenable的then函数。若是有一个失败的话，则当即进入reject流程。

总结

从代码设计层面来看，promiz的代码量较少，阅读也较为简单。可是，在某些细节的设计上，promiz仍是体现出了较为巧妙的思路，如在处理做为入参的promise时，可以在这个promise后面动态的添加一个then函数，从而获取数据给外面的promise。

若是你们有兴趣，建议本身根据本文的说明阅读一遍源码，配合Promise/A+规范来看下是如何实现每一条规范的。

下一篇博客，咱们将为你们从头开始，来实现一个Promise库。