JavaScript中的迭代器(interator)和生成器(generator)

通常js原生的可迭代对象,比如Array, Map, Set等都可以使用for...of来进行迭代,也可使用...obj的方式来拆解开,是因为他们对应的对象在底层都实现了可迭代方法@@iterator ,其实就是Symbol.interator方法。这个方法在调用后会返回一个可迭代对象,该对象中只需要包含next方法,每次迭代都需要调用next方法,next方法返回类型是{value: any, done: boolean}。value表示当前迭代到的值,done表示迭代是否完成。是不是感觉这种方法有点似曾相识,没错,就是generator函数本尊了。generator函数完全符合这种模式,就像下面:

function * gen() {
    yield 1;
    yield 2;
    yield 3;
}
const myIterator = gen();
console.log(myIterator.next());    //{value: 1, done: false}
console.log(myIterator.next());    //{value: 2, done: false}
console.log(myIterator.next());    //{value: 3, done: false}
console.log(myIterator.next());    //{value: undefined, done: true}

执行后返回迭代对象,迭代对象每次调用next方法返回{value: any, done: boolean}类型的对象。那么如何实现类似于Array,Map,Set这样的可以使用for...of迭代的对象呢,就是上面说的Symbol.iterator方法,只要在对象中或者在它的原型链的某个对象中实现Symbol.interator属性(key)对应的方法,就可以了,比如下面:

function * gen() {
    yield 1;
    yield 2;
    yield 3;
}
const myIterator = gen();
console.log(myIterator.next());    //{value: 1, done: false}
console.log(myIterator.next());    //{value: 2, done: false}
console.log(myIterator.next());    //{value: 3, done: false}
console.log(myIterator.next());    //{value: undefined, done: true}

const iter = {
    [Symbol.iterator]: gen
}

for (let v of iter) {
    console.log(v);
}
// 输出结果1,2,3

这样就相当于自己实现了一个像原生迭代器一样的迭代器。原生迭代器没有链表,那么就可以自行实现一个链表,加入Symbol.iterator方法,照样可以使用for...of来迭代。

那么问题来了,这里使用generator方法主要就是像使用它返回对象的next方法,那么自己实现一个这样功能的函数不就可以替代generator函数了?答案是可以的,虽然没有必要,但是可以装逼呀,说干就干

function gen(start = 0, end = Infinity, step = 1) {
    let nextIndex = start;
    let iterationCount = 0;

    const rangeIterator = {
       next: function() {
           let result;
           if (nextIndex < end) {
               result = { value: nextIndex, done: false }
               nextIndex += step;
               iterationCount++;
               return result;
           }
           return { value: iterationCount, done: true }
       }
    };
    return rangeIterator;
}

const myIterator = gen(0, 3);
console.log(myIterator.next());    //{value: 1, done: false}
console.log(myIterator.next());    //{value: 2, done: false}
console.log(myIterator.next());    //{value: 3, done: false}
console.log(myIterator.next());    //{value: undefined, done: true}

const iter = {
    [Symbol.iterator]: gen
}

for (let v of iter) {
    console.log(v);
}
// 输出结果1,2,3

这样,就自己完全手动实现了迭代器,成就感爆棚哈哈哈


版权声明:本文为m0_37907835原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。