理解co执行逻辑

对于co的一个最简版的实现，代码如下 （https://gist.github.com/iwillwen/8488050）

function co(generator) {
return function(fn) {
var gen = generator();
function next(err, result) {
if(err){
return fn(err);
}
var step = gen.next(result);
if (!step.done) {
step.value(next);
} else {
fn(null, step.value);
}
}
next();
}
}

为了理解这个代码，来看一个例子

var co = require('./co');
// wrap the function to thunk
function readFile(filename) {
return function(callback) {
require('fs').readFile(filename, 'utf8', callback);
};
}

co(function * () {
var file1 = yield readFile('./file/a.txt');
var file2 = yield readFile('./file/b.txt');

console.log(file1);
console.log(file2);
return 'done';
})(function(err, result) {
console.log(result)
});

现在就执行顺序来说明一下

首先是以下部分

co(function * () {
var file1 = yield readFile('./file/a.txt');
var file2 = yield readFile('./file/b.txt');

console.log(file1);
console.log(file2);
return 'done';
})

　　　这部分代码执行了co函数，并把function*()作为参数generator传入，也就是说，整个部分变成了

function(fn) {
var gen = generator();
function next(err, result) {
if(err){
return fn(err);
}
var step = gen.next(result);
if (!step.done) {
step.value(next);
} else {
fn(null, step.value);
}
}
next();
}(function(err, result) {
console.log(result)
})

　　这个部分执行这个function，并把

function(err, result) {
　　console.log(result)
}

　　作为fn的实参传入。

　　2. 执行过程中，`var gen`为 `function *()`部分，下面是函数`next()`的声明，之后执行了`next()`函数，此时的`err`和`result`均为空。

　　3. `err`为空，因此`if(err)`不执行，执行 `var step = gen.next(result);`，由于`result`这里为空，因此

var step = gen.next(result)
step.value
=readFile('./file/a.txt')
=function(callback) {
require('fs').readFile('./file/a.txt', 'utf8', callback);
};

　　4. `step.done `为`false`, 执行`step.value(next);`，即执行

function(callback) {
require('fs').readFile('./file/a.txt', 'utf8', callback);
};

　　函数，并`next`作为`callback`传入。`readFile`传入回调函数的参数为`err`和`data`，因此在读取完文件后，`err`和a文件的内容(作为result的实参)传入了　`next`函数并执行。

　　5. 与2，3类似，执行`var step = gen.next(result)`，这一步由于result的值为a文件的内容，因此`file1`被赋值为a文件的内容（next方法可以带一个参数，该参数就会被当作上一个yield语句的返回值。）。与4类似，err和b文件的内容(作为result的实参)传入`next`函数并执行。

　　6. 再次执行到`var step = gen.next(result)`的时候，`file2`被赋予文件b的内容，由于`generator`下面没有`yield`了，于是执行到`return`，`step.value='done'`,`step.done=true`，函数从上次yield语句停下的地方，一直执行到return语句（如果没有return语句，就执行到函数结束）。（next方法返回的对象的value属性，就是紧跟在return语句后面的表达式的值（如果没有return语句，则value属性的值为undefined），done属性的值true，表示遍历已经结束。）

　　7. 执行`fn(null, step.value);` 即

function(err, 'done') {
console.log('done');
}();

　　8. 因此代码的执行结果为

　　content in a.txt

　　content in b.txt

　　done