koa源碼分析-co模塊以及thunk

caikeal 發(fā)布于2019-08-19 17:50 / 2067人閱讀

摘要：以及模塊之前都是返回的函數(shù)之后的都是返回在語言中，函數(shù)替換的是將多參數(shù)函數(shù)，替換成單參數(shù)的版本，且只接受回調(diào)函數(shù)作為參數(shù)。

Thunk以及CO模塊

co4.0之前都是返回的thunk函數(shù)
之后的都是返回promise

thunk

thunk：在 JavaScript 語言中，Thunk 函數(shù)替換的是將多參數(shù)函數(shù)，替換成單參數(shù)的版本，且只接受回調(diào)函數(shù)作為參數(shù)。

// 正常版本的readFile（多參數(shù)版本）
fs.readFile(fileName, callback);

// Thunk版本的readFile（單參數(shù)版本）
var readFileThunk = Thunk(fileName);
readFileThunk(callback);

var Thunk = function (fileName){
     return function (callback){
    return fs.readFile(fileName, callback); 
  };
};

生產(chǎn)環(huán)境中，可以使用thunkify將函數(shù)轉(zhuǎn)換為thunk 函數(shù)

問題：

為什么node 里面大部分的callback都是第一個(gè)參數(shù)是err呢？

為什么要做thunk 轉(zhuǎn)換呢？

在redux里面也有thunk middleware，這個(gè)thunk是什么意思呢？

CO 模塊

co的原理很簡(jiǎn)單，就是將傳入的generator function 轉(zhuǎn)換為一個(gè)thunk，并且轉(zhuǎn)換后thunk 的generator的每個(gè)value值作為下一個(gè)狀態(tài)的輸入

function co(fn) {
   var isGenFun = isGeneratorFunction(fn);

   return function (done) {//返回的thunk函數(shù)，done 作為回調(diào)函數(shù)
        var ctx = this;
    
        // in toThunk() below we invoke co()
        // with a generator, so optimize for
        // this case
        var gen = fn;
    
        //gen function 轉(zhuǎn)換為generator
        if (isGenFun) {
          var args = slice.call(arguments), len = args.length;
          var hasCallback = len && "function" == typeof args[len - 1];
          done = hasCallback ? args.pop() : error;
          gen = fn.apply(this, args);
        } else {
          done = done || error;
        }
        //函數(shù)執(zhí)行的時(shí)候就會(huì)執(zhí)行next函數(shù)，進(jìn)入函數(shù)體里面
        next();

        // #92
        // wrap the callback in a setImmediate
        // so that any of its errors aren"t caught by `co`
        function exit(err, res) {
          setImmediate(done.bind(ctx, err, res));
        }

        function next(err, res) {
          var ret;
    
          // multiple args
          if (arguments.length > 2) res = slice.call(arguments, 1);
    
          // error
          if (err) {
            try {
              ret = gen.throw(err);
            } catch (e) {
              return exit(e);
            }
          }
    
          // ok
          if (!err) {
            try {
              ret = gen.next(res);
            } catch (e) {
              return exit(e);
            }
          }
    
          // done
          if (ret.done) return exit(null, ret.value);
    
          // normalize
          ret.value = toThunk(ret.value, ctx);
    
          // run
          if ("function" == typeof ret.value) {
            var called = false;
            try {
            
              //比如執(zhí)行yield readFile("test.json"), ret.value就是readFile函數(shù)，函數(shù)接受一個(gè)callback，callback調(diào)用next方法，講readFile的結(jié)果傳入了next函數(shù)
              ret.value.call(ctx, function(){
                //這里可以防止next函數(shù)被多次執(zhí)行
                if (called) return;
                called = true;
                next.apply(ctx, arguments);
              });
            } catch (e) {
              setImmediate(function(){
                if (called) return;
                called = true;
                next(e);
              });
            }
            return;
          }
    
          // invalid
          next(new Error("yield a function, promise, generator, array, or object"));
        }
      }
}

通過上面的co源碼分析，可以看下面的例子

co(function *() {
    var file = yield readFile("test.json");
    
    ?//這里的file是通過gen.next()?賦值的
    console.log(file);
    
    var ret = yield writeFile(file, "dest.json");
    
    console.log(ret);    
})

了解了這些基本概念后就可以進(jìn)入koa的源碼閱讀了，具體的可以參考下一篇。