理解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)
});

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

1. 首先是以下部分

 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