Generator函數(shù)

迭代器 vs 生成器

迭代器

• 有next方法，執(zhí)行返回結(jié)果對象
• 結(jié)果對象：value，done

function createIterator(item){
  var i = 0;
  return {
    next: function(){
      var done = i >= items.length;
      var value = !done ? items(i++) : undefined;
      return {
        done: done,
        value: value
      }
    }
  };
}

var iterator = createIterator([1,2,3]);

iterator.next();
iterator.next();
iterator.next();
iterator.next();

可迭代協(xié)議

• [Symbol.iterator]屬性
• 內(nèi)置可迭代對象：String Array Map Set等

迭代器協(xié)議

• next方法：done,value

Generator函數(shù)(生成器)

• ES6異步編程解決方案
• 聲明：通過function *聲明
• 返回值：符合可迭代協(xié)議和迭代器協(xié)議的生成器對象
• 在執(zhí)行時能暫停，又能從暫停處繼續(xù)執(zhí)行

生成器對象

• next(param)
• return(param)
• throw(param)

yield

• 只能出現(xiàn)在Generator函數(shù)
• 用來暫停和恢復(fù)生成器函數(shù)
  ** next執(zhí)行 **
• 遇yield暫停，將緊跟yield表達式的值作為返回的對象的value
• 沒有yield,一直執(zhí)行到return,將return的值作為返回的對象的value
• 沒有return,將undefined作為返回的對象的value
  ** next參數(shù) **
• next方法可以帶一個參數(shù)，該參數(shù)會被當(dāng)作一個yield表達式的返回值

function* createIterator(){
  let first = yield 1;
  let second = yield first + 2;
  yield second + 3;
}

let iterator = createIterator();

iterator.next(); // {value:1,done:false}
iterator.next(4); // {value:6,done:false}
iterator.next(5); // {value:8,done:false}
iterator.next(); // {value:undefined,done:true}

yield* 生成器函數(shù)/可迭代對象

• 委托給其他可迭代對象
• 作用：復(fù)用生成器

function* generator1(){
  yield 1;
  yield 2;
}

function* generator2(){
  yield 100;
  yield* generator1();
  yield 200;
}

let g2 = generator2();
g2.next(); // {value:100,done:false}
g2.next(); // {value:1,done:false}
g2.next(); // {value:2,done:false}
g2.next(); // {value:200,done:false}
g2.next(); // {value:undefined,done:true}

return(param)

• 給定param值終結(jié)遍歷器，param可缺省

function* createIterator(){
  yield 1;
  yield 2;
  yield 3;
}

let iterator = createIterator();

iterator.next(); // {value:1,done:false}
iterator.return(); // {value:undefined,done:true}
iterator.next(); // {value:undefined,done:true}

throw(param)

• 讓生成器對象內(nèi)部拋出錯誤

function* createIterator(){
  let first = yield 1;
  let second;
  try{
    second = yield first + 2;
  }catch(e){
    second = 6;
  }
  yield second +3;
}
let iterator = createIterator();

iterator.next(); // {value:1,done:false}
iterator.next(10); // {value:12,done:false}
iterator.throw(new Error('error')); // {value:9,done:false}
iterator.next(); // {value:undefined,done:true}

Generator函數(shù)的實現(xiàn)原理

協(xié)程

• 一個線程存在多個協(xié)程，但同時只能執(zhí)行一個
• Generator函數(shù)是協(xié)程在ES6的實現(xiàn)
• Yield掛起X協(xié)程（交給其他協(xié)程），next喚醒X協(xié)程

Generator函數(shù)應(yīng)用

//需求：按順序讀取文件
function readFilesByCallback(){
  const fs = require('fs');
  const files = [
    "Users/kitty/testgenerator/1.json",
    "Users/kitty/testgenerator/2.json",
    "Users/kitty/testgenerator/3.json",
  ];
  fs.readFile(files[0],function(err, data){
    console.log(data.toString());
    fs.readFile(files[1],function(err, data){
      console.log(data.toString());
      fs.readFile(files[2],function(err, data){
        console.log(data.toString());
      });
    });
  });
}

//調(diào)用
readFilesByCallback();

//上面的代碼回調(diào)太多，用generator優(yōu)化代碼
function* readFilesByGenerator(){
  const fs = require("fs");
  const files = [
    "Users/kitty/testgenerator/1.json",
    "Users/kitty/testgenerator/2.json",
    "Users/kitty/testgenerator/3.json",
  ];
  let fileStr = "";
  function readFile(filename){
    fs.readFile(filename,function(err, data){
      console.log(data.toString());
      f.next(data,toString())
    });
  }
  yield readFile(files[0]);
  yield readFile(files[1]);
  yield readFile(files[2]);
}

//調(diào)用
const f = readFilesByGenerator();
f.next();

//缺點：耦合性太高，readFile里面的方法與外面調(diào)用的f.next()是同一個，代碼不優(yōu)雅
//修改方法：用Thunk函數(shù)

Thunk函數(shù)

• 求值策略 傳值調(diào)用，傳名調(diào)用sum(x+1,x+2)
  - 傳值調(diào)用：算好了x+1和x+2在調(diào)用sum
  - 傳名調(diào)用：調(diào)用sum函數(shù)時在計算x+1,x+2的值
• thunk函數(shù)時傳名調(diào)用的實現(xiàn)方式之一
• 可以實現(xiàn)自動執(zhí)行Generator函數(shù)

const fs = require("fs");
const Thunk = function(fn){//fn:fs.readFile
  return function(...args){
    return function(callback){
      return fn.call(this,...args,callback);//fn.readFile(...args,next())
    };
  };
};
const readFileThunk = Thunk(fs.readFile);//第一層return

function run(fn){
  var gen = fn();//g()
  function next(err, data){
    var result = gen.next(data);//{value:function,done:false}
    if(result.done) return;
    result.value(next);//重點 next傳入callback
  }
  next();
}

const g = function*(){
  const s1 = yield readFileThunk('Users/kitty/testgenerator/1.json');
  console.log(s1.toString());
  const s2 = yield readFileThunk('Users/kitty/testgenerator/2.json');
  console.log(s2.toString());
  const s3 = yield readFileThunk('Users/kitty/testgenerator/3.json');
  console.log(s3.toString());
}

run(g)

作業(yè)

• co模塊的源碼分析-koa引用了co模塊