生成器

生成器

生成器使用function*进行定义，调用生成器，并不会执行函数体，而是会返回一个生成器对象，这个生成器对象是一个迭代器，调用next()方法会开始执行函数体，直到遇到yield语句。yield语句的值会成为next()方法的返回值

function* getText(){
    yield "t"
    yield "e"
    yield "x"
    yield "t"
}

let get = getText()
get.next().value    // t
get.next().value    // e
get.next().value    // x
get.next().value    // t
get.next().done     // true

生成器也是可迭代对象，所以也可使用扩展操作符和 for/of

let text = [...getText()]   // [ 't', 'e', 'x', 't' ]
for(let i of getText()){
    console.log(i)
}

在对象字面量中定义，注意不能使用箭头函数

let o = {
    *g(){
        yield 1
    }
}

在自定义可迭代类中，[Symbol.iterator]方法可直接用生成器简化定义

class Range{
    constructor(from, to){
        this.from = from
        this.to = to
    }
    *[Symbol.iterator](){
        for(let x = Math.ceil(this.from); x <= this.to; x++){
            yield x
        }
    }
}

yield*

yield* 关键字与 yield类似，他不止送回一个值，而是迭代可迭代对象并返回得到的每一个值，当需要逐个迭代多个生成器时，通常会用到以下函数

function* sequence(...iterables){
    for(let iterable of iterables){
        for(let item of iterable){
            yield item
        }
    }
}
[...sequence("abc", [1, 2, 3])]    // [ 'a', 'b', 'c', 1, 2, 3 ]

此时可以使用yield*简化，注意，yield 和 yield*只能用在生成器函数中，所以for/of不能简化为forEach

function* sequence(...iterables){
    for(let iterable of iterables){
        yield* iterable
    }
}
[...sequence("abc", [1, 2, 3])]    // [ 'a', 'b', 'c', 1, 2, 3 ]

return

生成器函数也可以有返回值

const one = function*(){
    yield 1
    return "one"
}
// 正常迭代中，不会出现返回值的值
let a1 = [...one()]  // [1]
// 但显示调用next()时，可以得到
let a2 = one()
a2.next()   // { value: 1, done: false }
a2.next()   // { value: 'one', done: true }
// 如果生成器已经完成，则不会再有值
a2.next()   // { value: undefined, done: true }