30 KiB
迭代器
概念
迭代器(Iterator)是 JavaScript 中一种特殊的对象,它提供了一种统一的、通用的方式遍历个各种不同类型的数据结构。可以遍历的数据结构包括:数组、字符串、Set、Map 等可迭代对象。我们也可以自定义实现迭代器,以支持遍历自定义的数据结构。
通过迭代器,我们可以按顺序逐个获取数据中的元素,不需要手动跟踪索引(索引也可称之为指针、游标)。迭代器的行为很像数据库中的游标(cursor)。
我们也不需要关心可迭代对象内部的实现细节,即不需要关心目标对象是数组还是字符串,还是其他的数据结构。对于迭代器来说,这些数据结构都是一样的处理方式。
迭代器是一种常见的编程模式,最早出现在1974年设计的CLU编程语言中。不仅仅在JS中,其他许多编程语言(比如 Java、Python 等)都提供了迭代器的概念和实现。技术实现各有不同,但目的都是帮助我们用通用的方式遍历对象的数据结构,提高代码的简洁性、可读性和维护性。
迭代协议
迭代协议并不是编程语言的内置实现或语法,而是协议。迭代协议具体分为两个协议:可迭代协议、迭代器协议。
迭代器协议规定了产生一系列值(无论是有限个还是无限个)的标准方式。
迭代器是一个具体的对象,这个对象要符合迭代器协议。在JS中,某个对象只有实现了符合特定要求的 next() 方法,这个对象才能成为迭代器。
实现原理:next() 方法
在JS中,迭代器的实现原理是通过定义一个特定的next()
方法,该方法在每次迭代中返回一个包含两个属性的对象:done 和 value。
具体来说,next() 方法有如下要求:
(1)参数:无参数或者有一个参数。
(2)返回值:返回一个应当有以下两个属性的对象。属性值如下:
-
done 属性(Boolean 类型):表示迭代是否已经完成。当迭代器遍历完所有元素时,done 为 true,否则为 false。具体解释如下:
-
如果迭代器可以产生序列中的下一个值,则为 false,这等价于没有指定 done 属性。
-
如果迭代器已将序列迭代完毕,则为 true。这种情况下,value 可以省略,如果 value 依然存在,即为迭代结束之后的默认返回值。
-
-
value 属性:包含当前迭代步骤的值,可能是具体的值,也可能是 undefined。每次调用 next() 方法时,迭代器返回下一个值。done 为true时,可以省略。
举例:为数组创建迭代器
按照上面讲的迭代器协议,我们可以给一个数组手动创建一个用于遍历的迭代器。代码举例如下:
const strArr = ['qian', 'gu', 'yi', 'hao'];
// 为数组封装迭代器
function createArrayIterator(arr) {
let index = 0;
return {
next: () => {
if (index < arr.length) {
return { done: false, value: arr[index++] };
} else {
return { done: true };
}
},
};
}
const strArrIterator = createArrayIterator(strArr);
console.log(JSON.stringify(strArrIterator.next()));
console.log(JSON.stringify(strArrIterator.next()));
console.log(JSON.stringify(strArrIterator.next()));
console.log(JSON.stringify(strArrIterator.next()));
console.log(JSON.stringify(strArrIterator.next()));
打印结果:
{"done":false,"value":"qian"}
{"done":false,"value":"gu"}
{"done":false,"value":"yi"}
{"done":false,"value":"hao"}
{"done":true}
你可能会有疑问:实际开发中,我们真的需要大费周章地为一个简单的数组写一个迭代器函数吗?数组直接拿来遍历不就完事了吗?
是的,这大可不必。初衷是为了了解迭代器的原理。
可迭代对象
我们要注意区分一些概念:迭代器、可迭代对象、容器。迭代器是提供迭代功能的对象。可迭代对象是被迭代的目标对象,也称之为容器。
概念
当一个对象实现了 iterable protocol 协议时,它就是一个可迭代对象。这个对象要求必须实现了 @@iterator
方法,在内部封装了迭代器。我们可以通过 Symbol.iterator
函数调用该迭代器。
当我们使用迭代器的方式去遍历数组、字符串、Set、Map 等数据结构时,这些数据对象就属于可迭代对象。这些数据对象本身,内部就自带了迭代器。
可是,有些数据对象,并不具备可迭代的能力,那要怎么封装成可迭代对象呢?以及,可迭代对象需要具备什么特征?可迭代对象有什么用处?这就是本段要讲的内容。
可迭代对象的特征
凡是可迭代对象,都具备如下特征:
1、可迭代对象都有一个 [Symbol.iterator] 函数。通过这个函数,我们可以进行数据遍历操作。以一个简单的数组进行举例:
const myArr = ['qian', 'gu', 'yi', 'hao'];
console.log(typeof myArr[Symbol.iterator]);
console.log(myArr[Symbol.iterator]);
console.log(typeof myArr[Symbol.iterator]());
console.log(myArr[Symbol.iterator]());
// 获取数组自带的迭代器对象
const myIterator = myArr[Symbol.iterator]();
// 通过迭代器的 next() 方法遍历数组
console.log(myIterator.next());
console.log(myIterator.next());
console.log(myIterator.next());
console.log(myIterator.next());
console.log(myIterator.next());
打印结果:
2、可迭对象可以进行 for ... of 操作。其实 for ... of 底层就是调用了 @@iterator
方法。代码举例:
const myArr = ['qian', 'gu', 'yi', 'hao'];
// 可迭代对象可以进行 for ... of 操作。for ... of 也是一种遍历操作。
for (const item of myArr) {
// 这里的 item,其实就是迭代器里的 value 属性的值。
console.log(item);
}
打印结果:
qian
gu
yi
hao
原生可迭代对象
以下这些对象,都是原生可迭代对象,请务必记住:
- String 字符串
- Array 数组
- Map
- Set
- arguments 对象
- NodeList 对象(DOM节点的集合)
原生可迭代对象的内部已经实现了可迭代协议,它们都符合可迭代对象的特征。比如,它们内部都有一个迭代器;他们可以用 for ... of 进行遍历。
为何要记住上面这些可迭代对象,因为可迭代对象的应用场景非常多,且非常好用。我们继续往下学习。
可迭代对象的应用场景(重要)
可迭代对象有许多应用场景,包括但不仅限于:
1、JavaScript的语法:
- for ... of
- 展开语法 ...
- yield*
- 解构赋值
2、创建一些对象:
- new Map([Iterable]):参数是可选的,可不传参数,也可以传一个可迭代对象作为参数
- new WeakMap([iterable])
- new Set([iterable])
- new WeakSet([iterable])
3、方法调用
- Array.from(iterable):将一个可迭代对象转为数组
- Promise.all(iterable)
- Promise.race(iterable)
今后在遇到这些应用场景时,这些原生可迭代对象可以直接拿来用。
比如说,通过阅读官方文档后我们得知,new Set()
的写法中,括号里的参数可以不写,也可以传入一个可迭代对象 iterable
。那么,字符串、数组、Set、Map等可迭代对象,在你需要的时候都可以传进去使用。而且,const a = new Set()
写法中,变量 a 也是一个可迭代对象。
Promise.all(iterable)
只能传数组吗?非也。准确来说,Promise.all()的参数中,传入的不是数组,而是一个可迭代对象。代码举例:
const promise1 = Promise.resolve('promise1 resolve');
const promise2 = Promise.resolve('promise2 resolve');
const promise3 = Promise.resolve('promise3 resolve');
const promiseSet = new Set();
promiseSet.add(promise1);
promiseSet.add(promise2);
promiseSet.add(promise3);
// 准确来说,Promise.all()的参数中,传入的不是数组,而是一个可迭代对象
Promise.all(promiseSet).then(res => {
console.log('res:', res);
});
代码举例:
res: ['promise1 resolve', 'promise2 resolve', 'promise3 resolve']
arguments 同样是一个可迭代对象,但不是数组。我们可以通过Array.from(iterable)
方法将 arguments 转为数组,进而让其享受数组的待遇,调用数组的各种方法。代码举例:
foo('a', 'b', 'c');
// 定义函数
function foo() {
// Array.from() 中的参数可以传入可迭代对象,将参数转为数组。arguments 是 foo() 函数的参数
const arr = Array.from(arguments);
console.log(arr);
}
打印结果:
['a', 'b', 'c']
学完了迭代器、可迭代对象的知识之后,很多关于函数传参、数据遍历、数据结构等方面的JS知识,就能融会贯通了。
手写迭代器
很多数据对象由于不是可迭代对象,我们可以为其手动创建一个迭代器,这个数据对象就成了可迭代对象。
为普通对象创建外部迭代器
代码举例:
const myObj1 = {
strArr: ['qian', 'gu', 'yi', 'hao'],
};
// 为 myObj.strArr 封装迭代器
let index = 0;
const strArrIterator = {
next: () => {
if (index < myObj1.strArr.length) {
return { done: false, value: myObj1.strArr[index++] };
} else {
return { done: true };
}
},
};
console.log(strArrIterator.next());
console.log(strArrIterator.next());
console.log(strArrIterator.next());
console.log(strArrIterator.next());
console.log(strArrIterator.next());
打印结果:
{done: false, value: 'qian'}
{done: false, value: 'gu'}
{done: false, value: 'yi'}
{done: false, value: 'hao'}
{done: true}
将普通对象封装为可迭代对象
上面的数据 myObj1,不属于可迭代对象,因此我们单独写了一个迭代器对象 strArrIterator。但是这两个对象是分开的。
还有一种更高级的做法是,把迭代器封装到数据对象的内部。完事之后,这个数据对象就是妥妥的可迭代对象。
将普通的数据对象封装为可迭代对象时,具体做法是:在数据对象内部,创建一个名为[Symbol.iterator]
的迭代器函数,这个函数名是固定的(这种写法属于计算属性名);然后这个函数内需要返回一个迭代器,用于迭代当前的数据对象。
我们以下面这两个对象为例:
const myObj1 = {
strArr: ['qian', 'gu', 'yi', 'hao'],
};
const myObj2 = {
name: 'qianguyihao',
skill: 'web',
};
如果尝试用 for of 去遍历它们,会报错:
const myObj2 = {
name: 'qianguyihao',
skill: 'web',
};
for (const item of myObj2) {
// 打印报错:Uncaught TypeError: myObj2 is not iterable。意思是:myObj2 不是可迭代对象
console.log(item);
}
所以,我们可以将这两个普通对象封装为可迭代对象。
1、将 myObj1 封装为可迭代对象,遍历 myObj1.strArr。代码举例如下:
const myObj1 = {
strArr: ['qian', 'gu', 'yi', 'hao'],
// 在 myObj1 的内部创建一个迭代器
[Symbol.iterator]: function () {
let index = 0;
const strArrIterator = {
next: function () {
if (index < myObj1.strArr.length) {
return { done: false, value: myObj1.strArr[index++] };
} else {
return { done: true };
}
},
};
return strArrIterator;
},
};
// 获取 myObj1 的迭代器对象
const strArrIterator = myObj2[Symbol.iterator]();
// 通过迭代器遍历 myObj1.strArr 的数据
console.log(strArrIterator.next());
console.log(strArrIterator.next());
console.log(strArrIterator.next());
console.log(strArrIterator.next());
console.log(strArrIterator.next());
打印结果:
{done: false, value: 'qian'}
{done: false, value: 'gu'}
{done: false, value: 'yi'}
{done: false, value: 'hao'}
{done: true}
上方代码有一个改进之处,如果把迭代器函数改成箭头函数,就可以通过 this.strArr
代表 myObj2.strArr
了,写法更简洁。代码改进如下:
const myObj1 = {
strArr: ['qian', 'gu', 'yi', 'hao'],
// 在 myObj1 的内部创建一个迭代器
[Symbol.iterator]: function () {
let index = 0;
const strArrIterator = {
next: () => {
if (index < this.strArr.length) {
return { done: false, value: this.strArr[index++] };
} else {
return { done: true };
}
},
};
return strArrIterator;
},
};
// 获取 myObj1 的迭代器对象
const strArrIterator = myObj2[Symbol.iterator]();
// 通过迭代器遍历 myObj1.strArr 的数据
console.log(strArrIterator.next());
console.log(strArrIterator.next());
console.log(strArrIterator.next());
console.log(strArrIterator.next());
console.log(strArrIterator.next());
打印结果不变。
2、将 myObj2 封装为可迭代对象,遍历里面的键值对。代码举例如下:
const myObj2 = {
name: 'qianguyihao',
skill: 'web',
// 将普通对象 myObj2 封装为可迭代对象,目的是遍历 myObj2 的键值对
[Symbol.iterator]: function () {
// const keys = Object.keys(this); // 获取对象的 key
// const values = Object.values(this); // 获取对象的 value
const entries = Object.entries(this); // 获取对象的键值对
let index = 0;
const iterator = {
next: () => {
if (index < entries.length) {
return { done: false, value: entries[index++] };
} else {
return { done: true };
}
},
};
return iterator;
},
};
// 可迭对象可以进行for of操作,遍历对象的键值对
for (const item of myObj2) {
const [key, value] = item;
console.log(key, value);
}
打印结果:
name qianguyihao
skill web
将自定义类封装为可迭代对象
在面向对象开发时,如果你希望自己创建的类也具备可迭代的能力,那么,你可以在定义类的时候手动添加一个 @@iterator
方法,让其成为可迭代对象。
代码举例:
// 定义类
class Person {
constructor(name, arr) {
this.name = name;
this.arr = arr;
}
// 定义一个名为 [Symbol.iterator] 的实例方法,封装迭代器
[Symbol.iterator]() {
let index = 0;
const iterator = {
next: () => {
if (index < this.arr.length) {
return { done: false, value: this.arr[index++] };
} else {
return { done: true };
}
},
};
return iterator;
}
}
const person1 = new Person('千古壹号', ['前端', '工程师']);
const person2 = new Person('许嵩', ['想象之中', '有何不可']);
// Person的实例已经封装为可迭代对象了,可以通过 for ... of 进行遍历
for (const item of person2) {
console.log(item);
}
打印结果:
想象之中
有何不可
如何中断迭代器,停止继续遍历
迭代器在遍历数据对象的过程中,如果我们希望在符合指定条件下停止继续遍历,那么,我们可以使用 break、return、throw 等关键字中断迭代器。其中, break 关键字用得最多。
此外,我们还可在迭代器函数中添加一个名为return()
的方法,这个方法的作用是监听迭代器的中断,书写代码的位置与 next()
方法并列。
代码举例如下:
const myObj2 = {
name: 'qianguyihao',
skill: 'web',
// 将普通对象 myObj2 封装为可迭代对象,目的是遍历 myObj2 的键值对
[Symbol.iterator]: function () {
const entries = Object.entries(this); // 获取对象的键值对
let index = 0;
const iterator = {
next: () => {
if (index < entries.length) {
return { done: false, value: entries[index++] };
} else {
return { done: true };
}
},
// 【关键代码】监听迭代器的中断
return: () => {
console.log('迭代器被中断了');
return { done: true };
},
};
return iterator;
},
};
// 可迭对象可以进行 for of 操作,遍历对象的键值对
for (const item of myObj2) {
const [key, value] = item;
console.log(key, value);
if (value == 'qianguyihao') {
// 【关键代码】如果发现 value 为 qianguyihao,则中断迭代器,停止继续遍历
break;
}
}
打印结果:
name qianguyihao
迭代器被中断了
根据打印结果可以看出,迭代器只遍历了 myObj2 对象的第一个元素,符合指定条件后,通过 break 语句中断了迭代器,停止了继续遍历;与此同时,迭代器中的 return() 函数监听到了迭代器的中断。对了,return() 函数中,还需要写 return { done: true }
表示迭代器的使命已结束;如果不写这行则会报错:Uncaught TypeError: Iterator result undefined is not an object
。
生成器
概念
我们平时写的函数,基本是通过 return 返回值,或者发生异常,函数才会终止执行。这还不够灵活。
生成器是 ES6 中新增的一种特殊的函数,所以也称为“生成器函数”。它可以更灵活地控制函数什么时候执行, 什么时候暂停等等。
生成器函数使用 function*
语法编写。最初调用时,生成器函数不执行任何代码,而是返回一个称为 Generator 的迭代器。通过调用生成器的 next() 方法时,Generator 函数将执行,直到遇到 yield 关键字时暂停执行。
可以根据需要多次调用该函数,并且每次都返回一个新的 Generator,但每个 Generator 只能迭代一次。
生成器函数和普通函数的区别
- 生成器函数需要在
function
关键字后面加一个符号*
。 - 生成器函数可以通过
yield
关键字控制函数的执行流程。 - 生成器函数的返回值是一个生成器(Generator)。生成器是一种特殊的迭代器。
生成器函数拆解
定义一个生成器函数,以及生成器函数的执行时机
如果要定义一个生成器函数,我们需要在function
单词的后面追加一个*
符号。代码举例:
function* foo() {
console.log('1');
console.log('2');
console.log('3');
}
foo();
上方代码中,*
符号既可以写在 function
单词的后面,即function* foo()
;也可以写在foo
单词的前面,即function *foo()
。这两种写法都可以,第一种写法更常见。
但是上面的代码写完后,并不会有打印结果,因为我们还需要调用生成器的 next()方法,生成器函数才会执行,直到遇到 yield 关键字后暂停执行;最后遇到 return关键字,或者遇到函数末尾时,结束执行。代码举例:
// 通过 * 符号,定义一个生成器函数
function* foo() {
console.log('1');
yield;
console.log('2');
// 下面这行 console.log('a') 会跟 yield 一起执行
yield console.log('a');
console.log('3');
}
const generator = foo(); // 返回一个生成器对象
// 调用生成器的 next()方法,生成器才会执行,直到遇到 yield 后暂停执行
generator.next(); // 这行代码执行后,打印结果是:1
generator.next(); // 这行代码执行后,打印结果是:1 2 a
generator.next(); // 这行代码执行后,打印结果是:1 2 a 3
仔细看注释,生成器 generator 每调用一次 next() ,foo()函数里的代码就往下执行一次,直到遇到 yield 后暂停。
next() 方法的返回值
生成器既然是一种特殊的迭代器,那么它也有 next()方法,而且 next()方法里同样有 done 和 value 这两个属性。我们来看看这两个属性的默认属性值是什么:
// 通过 * 符号,定义一个生成器函数
function* foo() {
console.log('阶段1');
yield;
console.log('阶段2');
yield;
console.log('阶段3');
return; // 执行 return 之后,函数不再继续往下走了,生成器的 next()方法的 done 属性值为 true。
console.log('阶段4');
}
// 执行生成器函数,返回一个生成器对象
const generator = foo();
// 调用生成器的 next()方法,生成器才会执行,直到遇到 yield 后暂停执行;遇到 return关键字,或者遇到函数末尾时,结束执行。
console.log(generator.next());
console.log(generator.next());
console.log(generator.next());
打印结果:
阶段1
{value: undefined, done: false}
阶段2
{value: undefined, done: false}
阶段3
{value: undefined, done: true}
上方代码的打印结果可以看出,生成器函数在遇到函数的末尾,或者遇到 return 之后,函数就不再继续往下走了,next()方法的 done 属性值为 true。
还可以看到,next()方法的 value 属性值默认为 undefined,如果某些情况下我们希望 value属性有值的话,可以通过 yield 关键字进行传递。代码举例:
// 通过 * 符号,定义一个生成器函数
function* foo() {
console.log('阶段1');
yield 'a'; // 【关键代码】yield 后面写的内容,就是传递给 next() 方法的 value 属性值
console.log('阶段2');
yield 'b';
console.log('阶段3');
return; // 这里的 return,相当于 return undefined
console.log('阶段4');
}
// 执行生成器函数,返回一个生成器对象
const generator = foo();
// 调用生成器的 next()方法,生成器才会执行,直到遇到 yield 后暂停执行;遇到 return关键字,或者遇到函数末尾时,结束执行。
console.log(generator.next()); // 打印生成器对象的 next()方法
console.log(generator.next());
console.log(generator.next());
打印结果:
阶段1
{value: 'a', done: false}
阶段2
{value: 'b', done: false}
阶段3
{value: undefined, done: true}
next() 方法的参数
根据前面的代码实例得知,生成器函数是分多个阶段执行的。此时有一个诉求:如何给当前阶段传递参数呢?
答案是:可以通过当前阶段 next() 方法的参数,给当前阶段传值。这个参数值会成为上一个阶段 yield 语句的返回值。
代码举例:
// 通过 * 符号,定义一个生成器函数
function* foo() {
console.log('阶段1');
// 【关键代码】第二次调用 next()方法时,通过 res2 接收 next()方法的参数值
const res2 = yield 'a';
console.log('阶段2:', res2);
// 第三次调用 next()方法时,通过 res3 接收 next()方法的参数值
const res3 = yield 'b';
console.log('阶段3:', res3);
return;
}
// 执行生成器函数,返回一个生成器对象
const generator = foo();
// 调用生成器的 next()方法,生成器才会执行,直到遇到 yield 后暂停执行;遇到 return关键字,或者遇到函数末尾时,结束执行。
console.log(generator.next()); // 执行第一阶段
console.log(generator.next('next2')); // 执行第二阶段,并传参
console.log(generator.next('next3')); // 指定第三阶段
打印结果:
阶段1
{value: 'a', done: false}
阶段2: next2
{value: 'b', done: false}
阶段3: ntext3
{value: undefined, done: true}
在理解上方代码时需要注意的是,将 next2 这个属性值赋值给 res2,这个操作的执行时机是在第二阶段的最开始做的,不是在第一阶段的末尾做的。并且,这个属性值是通过第一阶段的 yield 返回值接收的。
如何中途结束生成器的执行
如果想在中途结束生成器的执行,有三种方式:
- 方式1:return 语句。这个在前面已经讲过。
- 方式2:通过生成器的 return() 函数。
- 方式3:通过生成器的 throw() 函数抛出异常。
方式2的代码举例:
// 通过 * 符号,定义一个生成器函数
function* foo() {
console.log('阶段1');
const res2 = yield 'a';
console.log('阶段2:', res2);
const res3 = yield 'b';
console.log('阶段3:', res3);
return;
}
// 执行生成器函数,返回一个生成器对象
const generator = foo();
console.log(generator.next());
// 【关键代码】通过生成器的 return()方法, 立即结束 foo 函数的执行
console.log(generator.return('next2'));
// 这行写了也没用,阶段2、阶段3都不会执行的
console.log(generator.next('next3'));
打印结果:
阶段1
{value: 'a', done: false}
{value: 'next2', done: true}
{value: undefined, done: true}
上方代码可以看出,阶段2、阶段3都不会执行;return()方法里的参数传给了 value 属性。
方式3的代码举例:
// 通过 * 符号,定义一个生成器函数
function* foo() {
console.log('阶段1');
const res2 = yield 'a';
console.log('阶段2:', res2);
const res3 = yield 'b';
console.log('阶段3:', res3);
return;
}
// 执行生成器函数,返回一个生成器对象
const generator = foo();
console.log(generator.next());
// 【关键代码】通过生成器的 throw()方法抛出异常, 立即结束 foo 函数的执行
console.log(generator.throw(new Error('next2 error')));
// 这行写了也没用,阶段2、阶段3都不会执行的
console.log(generator.next('next3'));
打印结果:
阶段1
{value: 'a', done: false}
Uncaught Error: next2 error
生成器的应用
用生成器代替迭代器
在前面的迭代器内容中,我们学习过“将普通对象封装为可迭代对象”。那段代码改用生成器的写法也可以实现。代码举例:
const myObj2 = {
name: 'qianguyihao',
skill: 'web',
// 将普通对象 myObj2 封装为可迭代对象,目的是遍历 myObj2 的键值对。通过生成器 function* 的的方式实现
[Symbol.iterator]: function* () {
const entries = Object.entries(this); // 获取对象的键值对
for (let index = 0; index < entries.length; index++) {
// 【关键代码】通过 yield 控制迭代器分阶段执行;并将每个阶段的值存放到迭代器的 next() 方法的 value 属性中
yield entries[index];
}
},
};
// 写法1:通过 for ... of 遍历可迭代对象
for (const item of myObj2) {
const [key, value] = item;
console.log(key, value);
}
console.log('---');
// 写法2:通过 next() 方法遍历可迭代对象。与写法1等价。
const iterator = myObj2[Symbol.iterator]();
console.log(iterator.next());
console.log(iterator.next());
console.log(iterator.next());
打印结果:
name qianguyihao
demo.html:30 skill web
---
done: false, value:['name', 'qianguyihao']
done: false, value:['skill', 'web']
done: true, value:undefined
在指定数字范围内生成一个值
代码举例:
function* createValueGenerator(start, end) {
// 前闭后开
for (let i = start; i < end; i++) {
yield i;
}
}
const valueGenerator = createValueGenerator(1, 3);
console.log(valueGenerator.next());
console.log(valueGenerator.next());
console.log(valueGenerator.next());
打印结果:
{value: 1, done: false}
{value: 2, done: false}
{value: undefined, done: true}
使用 yield* 迭代可迭代对象
yield* 的介绍
语法格式:
yield* 某个可迭代对象
yield*
是 yield 的一种语法糖,也就是一种简写形式。它会依次迭代一个可迭对对象,每次迭代一个值,并产生一个新的可迭代对象。
我们在前面讲过可迭代对象的应用场景,其中就包括 yield*
。
先来看下面这段代码,用生成器的方式迭代数组:
function* createArrayGenerator(arr) {
for (const item of arr) {
yield item;
}
}
const myArr = ['a', 'b', 'c'];
const arrGenerator = createArrayGenerator(myArr);
console.log(arrGenerator.next());
console.log(arrGenerator.next());
console.log(arrGenerator.next());
console.log(arrGenerator.next());
打印结果:
{value: 'a', done: false}
{value: 'b', done: false}
{value: 'c', done: false}
{value: undefined, done: true}
上面这段代码,换成 yield*
的写法会非常简洁,如下:
function* createArrayGenerator(arr) {
// 【关键代码】yield* 的后面必须是一个可迭代对象
yield* arr;
}
const myArr = ['a', 'b', 'c'];
const arrGenerator = createArrayGenerator(myArr);
console.log(arrGenerator.next());
console.log(arrGenerator.next());
console.log(arrGenerator.next());
console.log(arrGenerator.next());
打印结果不变。代码解释:yield*
的后面必须是一个可迭代对象,而且 createArrayGenerator()函数会返回一个可迭代对象 arrGenerator。
将自定义类封装为可迭代对象
我们在前面学习迭代器时,曾通过迭代器“将自定义类封装为可迭代对象”。那段代码,改用生成器 yield*
的方式也可以实现。代码举例:
// 定义类
class Person {
constructor(name, arr) {
this.name = name;
this.arr = arr;
}
//【关键代码】在定义生成器函数时,如果没有 function 这个单词的话,也可以直接在函数名的前面添加 * 符号。
*[Symbol.iterator]() {
// 【关键代码】一行代码,直接遍历 this.arr 这个可迭代对象,同时返回一个新的可迭代对象
yield* this.arr;
}
}
const person1 = new Person('千古壹号', ['前端', '工程师']);
const person2 = new Person('许嵩', ['想象之中', '有何不可']);
// Person的实例已经封装为可迭代对象了,可以通过 for ... of 进行遍历
for (const item of person1) {
console.log(item);
}
console.log('---');
// 也可以通过 Symbol.iterator() 方法进行遍历
const personIterator = person2[Symbol.iterator]();
console.log(personIterator.next());
console.log(personIterator.next());
console.log(personIterator.next());
打印结果:
前端
工程师
---
{value: '想象之中', done: false}
{value: '有何不可', done: false}
{value: undefined, done: true}