前言

大名鼎鼎的co正是TJ大神基于ES6的一些新特性开发的异步流程控制库,基于它所开发的koa更是被视为未来主流的web框架。之前在论坛也看了不少大神们关于co源码的分析,不过co在升级为4.X版本时,代码进行了一次颇有规模的重构,从先前的基于thunkify函数,改变成了现在的基于Promise,并且可能在未来版本移除对thunkify函数的支持。正好小弟最近也在看TJ大神的源码,也来分享一下co(4.X)的实现思路以及源码注解。

进入正题

以下是两段co(4.X)的经典使用示例:

1
2
3
4
5
6
7
8
co(function* () {
var result = yield Promise.resolve(true);
return result;
}).then(function (value) {
console.log(value);
}, function (err) {
console.error(err.stack);
});

1
2
3
4
5
6
7
8
9
co(function *(){
// resolve multiple promises in parallel
var a = Promise.resolve(1);
var b = Promise.resolve(2);
var c = Promise.resolve(3);
var res = yield [a, b, c];
console.log(res);
// => [1, 2, 3]
}).catch(onerror);

再结合co的文档,我们可以把co的主要功能分为:

  • 异步流程控制,依次执行generator函数内的每个位于yield后的Promise对象,并在Promise的状态改变后,把其将要传递给reslove函数的结果传递给reject函数的错误返回出来,可供外部来进行传递值等操作,这些Promise是串行执行的。
  • yield后是Promise对象的数组属性值是Promise对象的对象,则返回出结构相同的Promise执行结果数组/对象,并且这些Promise是并行执行的。
  • co自身的返回值也是一个Promise对象,可供继续使用。

好了,我们要实现以上这几个功能,其实就是解决以下几个问题:

1,确保每一个yield动作的串行执行,并正确的返回异步结果。
2,若在yield后的是Promise数组属性值为Promise对象的对象,则并行执行这些Promise。
3,检查每一个yield后的对象是否符合要求,若不,则尝试进行转换。
4,最后整体返回一个Promise。

我们来逐个击破。

解决1,4

以下是co(4.X)源码中的解决1,4的方案:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
function co(gen) {
var ctx = this;

/**
* 我们看到,co函数整个的返回值便是一个Promise实例,包装了
* 传递的generator函数内所有Promise的执行,
* 我们暂且称它为"外壳Promise"吧,而在传递进来的generator函数内的所有Promise我们都先称为“内部Promise”,
* 用以区分。
*/
return new Promise(function(resolve, reject) {

//判断传递给co的参数,若是一个generator函数,则执行之,得到一个generator对象
if (typeof gen === 'function') gen = gen.call(ctx);
//此时gen对象还不是一个generator对象的话,
//则调用“外壳Promise”的resolve,直接结束整个“外壳Promise”,把参数值作为结果传出
if (!gen || typeof gen.next !== 'function') return resolve(gen);

//入口,这个函数的详细分析请往下看两行。。
onFulfilled();

//这个onFulfilled函数主要有两个用途:
//第一个用途就是上面的入口函数的功能,将generator执行到第一个yield处开启第一个异步调用
//第二个用途便是当作所有”内部Promise“的resolve方法,处理异步结果,并继续调用下一个Promise
function onFulfilled(res) {
var ret;
try {
//当作为”内部Promise“的resolve方法时,res参数自然便是本次Promise的执行结果了
//利用generator函数的特性,调用next方法时的参数,会当做yield的返回值,这样我们就做到了将异步的结果返回出来
ret = gen.next(res);
} catch (e) {
//若报错,则直接调用"外壳Promise"的reject方法,直接结束整个“外壳Promise”,把错误对象作为结果传出
return reject(e);
}

//将generator.next的执行结果传入next函数,实现串行调用。关于next函数的分析也请往下看。。
next(ret);
}

//这个onRejected函数的用途自然便是当作"内部Promise"的reject方法啦
function onRejected(err) {
var ret;
try {
ret = gen.throw(err);
} catch (e) {
return reject(e);
}
next(ret);
}

//这个next函数便是用来执行串行调用
function next(ret) {
//如果“内部Promise”,全部执行完毕,done的值便已经是generator函数中的return出的值了,
//把这个结果传递给“外壳Promise”的resolve函数,暴露给处理整个co结果的后续调用
if (ret.done) return resolve(ret.value);

//若“内部Promise”尚未执行完毕,那么确保ret.value是一个Promise对象,并进而调用它
var value = toPromise.call(ctx, ret.value);
if (value && isPromise(value)) return value.then(onFulfilled, onRejected);
return onRejected(new TypeError('You may only yield a function, promise, generator, array, or object, '
+ 'but the following object was passed: "' + String(ret.value) + '"'));
}
});
}

好,到这儿我们已经把co4.X的核心串行调用的实现过程给搞定了。整个的执行流程可以归结为:

  • 进入“外壳Promise
  • 通过入口onFulfilled(),取得第一个Promise对象
  • 将每一个“内部Promise”通过then(onFulfilled, onRejected)开始执行,并将resolve函数都封装为onFulfilled(),reject函数都封装为onRejected()
  • 通过在onFulfilled()onRejected()内调用next方法,实现在一个异步调用结果返回后继续执行下一个,通过将结果作为参数传递给next方法,实现将异步结果返回给外部,依次往复。
  • 所有“内部Promise”执行完毕,将最后结果暴露给“外壳Promise”的处理函数,结束

解决2,3

在TJ大神的源码中,2其实伴随在3(即上面代码中的toPromise方法)中解决的,我们来看一下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
/**
* 主要任务便是将传入的参数对象转换为Promise对象
*/
function toPromise(obj) {
//确保obj有意义
if (!obj) return obj;
//如果已经是Promise对象,则直接把obj返回出去
if (isPromise(obj)) return obj;
//若是generator函数或现成的generator对象,则直接把obj作为参数传入co函数,并把这个co函数
//返回出来的"外壳Promise"作为return出来的Promise
if (isGeneratorFunction(obj) || isGenerator(obj)) return co.call(this, obj);
//若obj是函数,则直接视为符合thunk规范的函数(thunk函数是啥这里就不细说了哈。。),直接转换
if ('function' == typeof obj) return thunkToPromise.call(this, obj);
//若是Promise数组,则调用arrayToPromise方法
if (Array.isArray(obj)) return arrayToPromise.call(this, obj);
//若是属性值是Promise对象的对象,则调用objectToPromise方法
if (isObject(obj)) return objectToPromise.call(this, obj);
return obj;
}

5个if其实就是一堆的判断并执行响应转换咯(废话。。),看来已经解决了问题3,问题2的精髓自然就是arrayToPromiseobjectToPromise,好,我们来攻克这最后一道壁垒:

先是arrayToPromise

1
2
3
4
5
6
7
8
9
10
/**
* 利用ES6规范中的Promise.all方法,充当全部结果的返回者
* 利用Array.map方法,实现了并行操作,分别对数组中的每一个元素递归执行toPromise方法,把这些子Promise接着
* 返回co中来获取执行结果,最后等待这些子Promise全部得到结果后,Promise.all执行成功,
* 返回执行结果数组
* 这实现,请收下我的膝盖。。。
*/
function arrayToPromise(obj) {
return Promise.all(obj.map(toPromise, this));
}

接着是objectToPromise

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
/**
* 与数组略有不同,利用for循环来实现并行的异步调用,
* Promise.all()仅充当一个类计数器,并返回最终结果
*/
function objectToPromise(obj){
//results是将用于返回的对象,使用和obj相同的构造函数
var results = new obj.constructor();
//Object.keys方法用于返回对象的所有的属性名
var keys = Object.keys(obj);
//寄存所有对象属性的Promise的数组
var promises = [];

//利用for循环来实现异步
for (var i = 0; i < keys.length; i++) {
var key = keys[i];
//确保obj[key]为Promise对象,然后调用defer推入promises数组等待执行,否则直接将结果返回给result[key]
var promise = toPromise.call(this, obj[key]);
if (promise && isPromise(promise)) defer(promise, key);
else results[key] = obj[key];
}

//传入的是promise.then()返回的空Promise,所以此处Promise.all仅充当一个计数器,确保所有异步操作的resolve操作中对results对象的属性都赋值完毕后,返回最终的results对象
return Promise.all(promises).then(function () {
return results;
});

//执行异步操作,并在操作结果赋值给results[key]
function defer(promise, key) {
results[key] = undefined;
promises.push(promise.then(function (res) {
results[key] = res;
}));
}
}

好了,到这我们已经把co(4.X)的核心源码实现都搞定,它的庐山真面目就是这样啦~

如有任何不正确之处,欢迎指出^ ^

参考

co