通常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
这样,就自己完全手动实现了迭代器,成就感爆棚哈哈哈