神奇的co

co是什么?

https://github.com/tj/co

官方说:The ultimate generator based flow-control goodness for nodejs (supports thunks, promises, etc)

我的解释

  • 是generator的执行器
  • 可是使用generator里的Yieldable支持的所有形式(supports thunks, promises, etc)
  • 返回的结果是promise

那么它的用途就很明显了

  • 所有出现generator的地方它都可以出现
  • 在以后的async/await里await后面可以接promise

源码解析

co@4.6版本不到240行代码,整体来说,还算比较简单。但并不容易阅读

  1. // 核心代码
  2. function co(gen) {
  3. // 缓存this
  4. var ctx = this;
  5. var args = slice.call(arguments, 1)
  6. // we wrap everything in a promise to avoid promise chaining,
  7. // which leads to memory leak errors.
  8. // see https://github.com/tj/co/issues/180
  9. // 重点,co的返回值是Promise对象。为什么可以then和catch的根源
  10. return new Promise(function(resolve, reject) {
  11. // 如果你懂Promise规范,就知道这是解决状态回调,这是首次调用
  12. onFulfilled();
  13. /**
  14. * @param {Mixed} res
  15. * @return {Promise}
  16. * @api private
  17. */
  18. function onFulfilled(res) {
  19. var ret;
  20. try {
  21. ret = gen.next(res);
  22. } catch (e) {
  23. return reject(e);
  24. }
  25. next(ret);
  26. }
  27. /**
  28. * @param {Error} err
  29. * @return {Promise}
  30. * @api private
  31. */
  32. // 如果你懂Promise规范,就知道这是拒绝状态回调
  33. function onRejected(err) {
  34. var ret;
  35. try {
  36. ret = gen.throw(err);
  37. } catch (e) {
  38. return reject(e);
  39. }
  40. next(ret);
  41. }
  42. // generator执行器
  43. // 如果ret.done,返回ret.value
  44. // 否则,
  45. function next(ret) {
  46. // 如果执行完成,直接调用resolve把promise置为成功状态
  47. if (ret.done) return resolve(ret.value);
  48. // 把yield的值转换成promise
  49. // 支持 promise,generator,generatorFunction,array,object
  50. // toPromise的实现可以先不管,只要知道是转换成promise就行了
  51. var value = toPromise.call(ctx, ret.value);
  52. // 成功转换就可以直接给新的promise添加onFulfilled, onRejected。当新的promise状态变成结束态(成功或失败)。就会调用对应的回调。整个next链路就执行下去了。
  53. // 为什么generator可以无限的next下去呢?
  54. // return value.then(onFulfilled, onRejected);意味着,又要执行onFulfilled了
  55. // onFulfilled里调用next(ret);
  56. if (value && isPromise(value)) return value.then(onFulfilled, onRejected);
  57. // 如果以上情况都没发生,报错
  58. return onRejected(new TypeError('You may only yield a function, promise, generator, array, or object, '
  59. + 'but the following object was passed: "' + String(ret.value) + '"'));
  60. }
  61. });
  62. }

读此源码要点

  • 必须深刻理解Promise实现,知道构造函数里的onFulfilled和onRejected是什么意思
  • 必须了解generator的执行机制,理解迭代器里的next以及next的返回对象{value:’’,done: true}

核心代码入口是onFulfilled,无论如何第一次的next(ret)是一定要执行的,因为generator必须要next()一下的。

所以next(ret)一定是重点,而且我们看onFulfilled和onRejected里都调用它,也就是所有的逻辑都会丢在这个next(ret)方法里。它实际上是一个状态机的简单实现。

  1. // generator执行器
  2. // 如果ret.done,返回ret.value
  3. // 否则,
  4. function next(ret) {
  5. // 如果执行完成,直接调用resolve把promise置为成功状态
  6. if (ret.done) return resolve(ret.value);
  7. // 把yield的值转换成promise
  8. // 支持 promise,generator,generatorFunction,array,object
  9. // toPromise的实现可以先不管,只要知道是转换成promise就行了
  10. var value = toPromise.call(ctx, ret.value);
  11. // 成功转换就可以直接给新的promise添加onFulfilled, onRejected。当新的promise状态变成结束态(成功或失败)。就会调用对应的回调。整个next链路就执行下去了。
  12. // 为什么generator可以无限的next下去呢?
  13. // return value.then(onFulfilled, onRejected);意味着,又要执行onFulfilled了
  14. // onFulfilled里调用next(ret);
  15. if (value && isPromise(value)) return value.then(onFulfilled, onRejected);
  16. // 如果以上情况都没发生,报错
  17. return onRejected(new TypeError('You may only yield a function, promise, generator, array, or object, '
  18. + 'but the following object was passed: "' + String(ret.value) + '"'));
  19. }

情景1: 状态完成

  1. // 如果执行完成,直接调用resolve把promise置为成功状态
  2. if (ret.done) return resolve(ret.value);

情景2: next,跳回onFulfilled,递归

  1. // 成功转换就可以直接给新的promise添加onFulfilled, onRejected。当新的promise状态变成结束态(成功或失败)。就会调用对应的回调。整个next链路就执行下去了。
  2. // 为什么generator可以无限的next下去呢?
  3. // return value.then(onFulfilled, onRejected);意味着,又要执行onFulfilled了
  4. // onFulfilled里调用next(ret);
  5. if (value && isPromise(value)) return value.then(onFulfilled, onRejected);

情景3: 捕获异常

  1. // 如果以上情况都没发生,报错
  2. return onRejected(new TypeError('You may only yield a function, promise, generator, array, or object, '
  3. + 'but the following object was passed: "' + String(ret.value) + '"'));

以上是核心代码说明。之前我们讲了co实际有2种api,有参数和无参数的,很明显以上是无参数的generator执行器,那么有参数的wrap呢?

  1. // 为有参数的generator调用,提供简单包装
  2. co.wrap = function (fn) {
  3. createPromise.__generatorFunction__ = fn;
  4. return createPromise;
  5. function createPromise() {
  6. // 重点,把arguments给fn当参数。
  7. // call和apply是常规js api
  8. return co.call(this, fn.apply(this, arguments));
  9. }
  10. };

通过call和apply组合使用,知识点比较简单,但这样用还是挺巧妙的。

其他的就基本是工具类了,其实也挺有意思的,自己看吧

co引出的“血案”

ES6的Generator本意是为了计算而设计的迭代器,但tj觉得它可以用于流程控制,于是就有了co,co的历史可以说经历了目前所有的流程控制方案,而且由于支持Generator和yield就导致yieldable。

实际上co和Generator是把双刃剑,给了我们强大便利的同时,也增加了非常多的概念,可能是过渡性的,也可能是过时的。

可是,你真的需要了解这么多么?从学习的角度,当然是多多意义,如果从实用的角度看,你可能不需要。

存在即合理,那么我们就看看这“血案”吧:

  • 学习ES6的Generator
  • 了解ES6的迭代器和迭代器相关的2种协议,了解for-of
  • 了解co和co的2种用法,源码
  • 了解yieldable 5种(包括不常用Thunk)
  • 如果是koa,还需要了解convert和compose