浅谈Generator和Promise原理及实现

Generator

熟悉ES6语法的同学们肯定对Generator（生成器）函数不陌生，这是一个化异步为同步的利器。栗子：

function* abc() {
    let count = 0;
    while(true) {
        let msg = yield ++count;
        console.log(msg);
    }
}

let iter = abc();
console.log(iter.next().value);
// 1
console.log(iter.next('abc').value);
// 'abc'
// 2

首先，我们先简单回顾一下JS的运行规则：

JS是单线程的，只有一个主线程
函数内的代码从上到下顺序执行，遇到被调用的函数先进入被调用函数执行，待完成后继续执行
遇到异步事件，浏览器另开一个线程，主线程继续执行，待结果返回后，执行回调函数

那么，Generator函数是如何进行异步化为同步操作的呢？实质上很简单，* 和 yield 是一个标识符，在浏览器进行软编译的时候，遇到这两个符号，自动进行了代码转换：

// 异步函数
function asy() {
    $.ajax({
        url: 'test.txt',
        dataType: 'text',
        success() {
            console.log("我是异步代码");
        }
    })
}

function* gener() {
    let asy = yield asy();
    yield console.log("我是同步代码");
}
let it = gener().next();
it.then(function() {
    it.next();
})
// 我是异步代码
// 我是同步代码

// 浏览器编译之后
function gener() {
    // let asy = yield asy(); 替换为
    $.ajax({
        url: 'test.txt',
        dataType: 'text',
        success() {
            console.log("我是异步代码");
            // next 之后执行以下
            console.log("我是同步代码");
        }
    })
    // yield console.log("我是同步代码");
}

整个过程类似于，浏览器遇到标识符 * 之后，就明白这个函数是生成器函数，一旦遇到 yield 标识符，就会将以后的函数放入此异步函数之内，待异步返回结果后再进行执行。

更深一步，从内存上来讲：

普通函数在被调用时，JS 引擎会创建一个栈帧，在里面准备好局部变量、函数参数、临时值、代码执行的位置（也就是说这个函数的第一行对应到代码区里的第几行机器码），在当前栈帧里设置好返回位置，然后将新帧压入栈顶。待函数执行结束后，这个栈帧将被弹出栈然后销毁，返回值会被传给上一个栈帧。

当执行到 yield 语句时，Generator 的栈帧同样会被弹出栈外，但Generator在这里耍了个花招——它在堆里保存了栈帧的引用（或拷贝）！这样当 it.next 方法被调用时，JS引擎便不会重新创建一个栈帧，而是把堆里的栈帧直接入栈。因为栈帧里保存了函数执行所需的全部上下文以及当前执行的位置，所以当这一切都被恢复如初之时，就好像程序从原本暂停的地方继续向前执行了。

而因为每次 yield 和 it.next 都对应一次出栈和入栈，所以可以直接利用已有的栈机制，实现值的传出和传入。

至此，Generator 的魔力已经揭开。

Promise

Promise的用法大家应该都很熟悉：

let pr = new Promise(function(resolve, reject) {
    setTimeout(function() {
        resolve("成功执行啦");
    }, 2000)
})
pr.then(function(data) {
    console.log(data); // 成功执行啦
})

那么 Promise 是如何实现异步加载的呢？

Promise 并没有大家想的那么神秘，其本质就是一个状态机。

想要实现一个土生土长的 Promise 其实很简单，状态机，我们需要几个参数：

__success_res 用来存储成功时的参数
__error_res 用来存储失败时的参数
__status 用来存储状态
__watchList 用来存储执行队列

下面就手动实现一个 Promise

class Promise1 {
    constructor(fn) {
        // 执行队列
        this.__watchList = [];
        // 成功结果
        this.__success_res = null;
        // 失败结果
        this.__error_res = null;
        // 状态
        this.__status = "";
        fn((...args) => {
            // 保存成功数据
            this.__success_res = args;
            // 状态改为成功
            this.__status = "success";
            // 若为异步则回头执行then成功方法
            this.__watchList.forEach(element => {
                element.fn1(...args);
            });
        }, (...args) => {
            // 保存失败数据
            this.__error_res = args;
            // 状态改为失败
            this.__status = "error";
            // 若为异步则回头执行then失败方法
            this.__watchList.forEach(element => {
                element.fn2(...args);
            });
        });
    }

    // then 函数
    then(fn1, fn2) {
        if (this.__status === "success") {
            fn1(...this.__success_res);
        } else if (this.__status === "error") {
            fn2(...this.__error_res);
        } else {
            this.__watchList.push({
                fn1,
                fn2
            })
        }
    }
}

这样就简单实现了 Promise 的功能，在使用上和JS的 Promise 并无其他区别，若想实现 Promise.all 方法，则只需要进行小小的迭代：

Promise1.all = function(arr) {
    // 存放结果集
    let result = [];
    return Promise1(function(resolve, reject) {
        let i = 0;
        // 进行迭代执行
        function next() {
            arr[i].then(function(res) {
                // 存放每个方法的返回值
                result.push(res);
                i++;
                // 若全部执行完
                if (i === result.length) {
                    // 执行then回调
                    resolve(result);
                } else {
                    // 继续迭代
                    next();
                }
            }, reject)
        }
    })
}

至此，Generator 和 Promise 都已解析完成。

最后不好意思推广一下我基于 Taro 框架写的组件库：MP-ColorUI。

可以顺手 star 一下我就很开心啦，谢谢大家。

点这里是文档

点这里是 GitHUb 地址