前端人不了解的迭代器和生成器

2023年 9月 3日 36.7k 0

迭代器和生成器是 ES6 中引入的特性。迭代器通过一次消费一个项目列表来提高效率,类似于数据流。生成器是一种能够暂停执行的特殊函数。调用生成器允许以块的形式(一次一个)生成数据,而无需先将其存储在列表中。下面就来深入理解 JavaScript 中的迭代器和生成器,看看它们是如何使用的,又有何妙用!

迭代器

JavaScript 中的迭代器可以分别两种:同步迭代器和异步迭代器。

1. 同步迭代器

(1)迭代器和可迭代对象

在 JavaScript 中有很多方法可以遍历数据结构。例如,使用 for 循环或使用 while 循环。迭代器具有类似的功能,但有显着差异。

迭代器只需要知道集合中的当前位置,而其他循环则需要预先加载整个集合才能循环遍历它。迭代器使用 next() 方法访问集合中的下一个元素。但是,为了使用迭代器,值或数据结构应该是可迭代的。数组、字符串、映射、集合是 JavaScript 中的可迭代对象。普通对象是不可迭代的。

(2)定义迭代器

下面来看看集合不可迭代的场景:

const favouriteMovies = {
  a: '哈利波特',
  b: '指环王',
  c: '尖峰时刻',
  d: '星际穿越',
  e: '速度与激情',
}

这个对象是不可迭代的。如果使用普通的 for 循环遍历它,就会抛出错误。随着 ES6 中迭代器的引入,可以将其转换为可迭代对象以便遍历它。这些称为自定义迭代器。下面看看如何实现对象的遍历并打印出来:

favouriteMovies[Symbol.iterator] = function() {
  const ordered = Object.values(this).sort((a, b) => a - b);
  let i = 0;
  return {
    next: () => ({
      done: i >= ordered.length,
      value: ordered[i++]
    })
  }
}

for (const v of favouriteMovies) {
  console.log(v);
}

输出结果如下:

哈利波特
指环王
尖峰时刻
星际穿越
速度与激情

这里使用 Symbol.iterator() 来定义迭代器。任何具有 Symbol.iterator 键的结构都是可迭代的。

可迭代对象具有以下行为:

  • 当 for..of 循环开始时,它首先查找错误。如果未找到,则它会访问方法和定义该方法的对象。
  • 以 for..of 循环方式迭代该对象。
  • 使用该输出对象的 next() 方法来获取要返回的下一个值。
  • 返回的值的格式为 done:boolean, value: any。返回 done:true 时循环结束。
  • 下面来创建一个 LeapYear 对象,该对象返回范围为 (start, end) 的闰年列表,并在后续闰年之间设置间隔。

    class LeapYear {
      constructor(start = 2020, end = 2040, interval = 4) {
        this.start = start;
        this.end = end;
        this.interval = interval;
      }
      [Symbol.iterator]() {
        let nextLeapYear = this.start;
        return {
          next: () => {
            if (nextLeapYear  console.log(iterator.value));

    输出结果如下:

    1
    2

    也可以使用 for await...of 来对异步迭代对象进行迭代:

    async function consumer() {
      for await (const asyncIterableElement of asyncIterable) {
        console.log(asyncIterableElement);
      }
    }
    
    consumer();

    异步迭代器和迭代器是异步生成器的基础,后面会介绍异步生成器。

    生成器

    JavaScript 中的生成器可以分别两种:同步生成器和异步生成器。

    1. 同步生成器

    (1)基本概念

    生成器是一个可以暂停和恢复并可以产生多个值的过程。JavaScript 中的生成器由一个生成器函数组成,它返回一个可迭代 Generator 对象。

    生成器是对 JavaScript 的强大补充。它们可以维护状态,提供一种制作迭代器的有效方法,并且能够处理无限数据流,可用于在前端实现无限滚动等。此外,当与 Promises 一起使用时,生成器可以模拟 async/await 功能,这使我们能够以更直接和可读的方式处理异步代码。尽管 async/await 是处理常见、简单的异步用例(例如从 API 获取数据)的一种更普遍的方式,但生成器具有更高级的功能。

    生成器函数是返回生成器对象的函数,由 function 关键字后面跟星号 (*) 定义,如下所示:

    function* generatorFunction() {}

    有时,我们可能会在函数名称旁边看到星号,而不是 function 关键字,例如 function *generatorFunction(),它的工作原理是相同的,但 function* 是一种更广泛接受的语法。

    生成器函数也可以在表达式中定义,就像常规函数一样:

    const generatorFunction = function* () {}

    生成器甚至可以是对象或类的方法:

    // 生成器作为对象的方法
    const generatorObj = {
      *generatorMethod() {},
    }
    
    // 生成器作为类的方法
    class GeneratorClass {
      *generatorMethod() {}
    }

    下面的例子都将使用生成器函数声明得语法。

    注意:与常规函数不同,生成器不能使用 new 关键字构造,也不能与箭头函数结合使用。

    现在我们知道了如何声明生成器函数,下面来看看生成器返回的可迭代生成器对象。

    (2)生成器对象

    传统的 JavaScript 函数会在遇到return 关键字时返回一个值。如果省略 return 关键字,函数将隐式返回 undefined。

    例如,在下面的代码中,我们声明了一个 sum() 函数,它返回一个值,该值是两个整数参数的和:

    function sum(a, b) {
      return a + b
    }

    调用该函数会返回一个值,该值是参数的总和:

    const value = sum(5, 6) // 11

    而生成器函数不会立即返回值,而是返回一个可迭代的生成器对象。在下面的例子中,我们声明了一个函数并给它一个单一的返回值,就像一个标准的函数:

    function* generatorFunction() {
      return 'Hello, Generator!'
    }

    当调用生成器函数时,它将返回生成器对象,我们可以将其分配给一个变量:

    const generator = generatorFunction()

    如果这是一个常规函数,我们希望生成器为我们提供函数中返回的字符串。然而,我们实际得到的是一个处于挂起状态的对象。因此,调用生成器将提供类似于以下内容的输出:

    generatorFunction {}
      [[GeneratorLocation]]: VM335:1
      [[Prototype]]: Generator
      [[GeneratorState]]: "suspended"
      [[GeneratorFunction]]: ƒ* generatorFunction()
      [[GeneratorReceiver]]: Window

    函数返回的生成器对象是一个迭代器。迭代器是一个具有可用的 next() 方法的对象,该方法用于迭代一系列值。next() 方法返回一个对象,其包含两个属性:

    • value:当前步骤的值;
    • done:布尔值,指示生成器中是否有更多值。

    next() 方法必须遵循以下规则:

    • 返回一个带有 done: false 的对象来继续迭代;
    • 返回一个带有 done: true 的对象来停止迭代。

    下面就来在生成器上调用 next() 并获取迭代器的当前值和状态:

    generator.next()

    这将得到以下输出结果:

    {value: "Hello, Generator!", done: true}

    调用 next() 时的返回值为 Hello, Generator!,并且 done 的状态为 true,因为该值来自关闭迭代器的返回值。由于迭代器完成,生成器函数的状态将从挂起变为关闭。这时再次调用生成器将输出以下内容:

    generatorFunction {}

    除此之外,生成器函数也有区别于普通函数的独特特征。下面我们就来了解一下 yield 运算符并看看生成器如何暂停和恢复执行。

    (3)yield 运算符

    生成器为 JavaScript 引入了一个新的关键字:yield。**yield**** 可以暂停生成器函数并返回 **yield** 之后的值,从而提供一种轻量级的方法来遍历值。**

    在下面的例子中,我们将使用不同的值暂停生成器函数三次,并在最后返回一个值。然后将生成器对象分配给 generator 变量。

    function* generatorFunction() {
      yield 'One'
      yield 'Two'
      yield 'Three'
    
      return 'Hello, Generator!'
    }
    
    const generator = generatorFunction()

    现在,当我们在生成器函数上调用 next() 时,它会在每次遇到 yield 时暂停。done 会在每次 yield 后设置为 false,表示生成器还没有结束。一旦遇到 return,或者函数中没有更多的 yield 时,done 就会变为 true,生成器函数就结束了。

    连续四次调用 next() 方法:

    generator.next()
    generator.next()
    generator.next()
    generator.next()

    这些将按顺序得到以下结果:

    {value: "One", done: false}
    {value: "Two", done: false}
    {value: "Three", done: false}
    {value: "Hello, Generator!", done: true}

    next() 非常适合从迭代器对象中提取有限数据。

    注意,生成器不需要 return;如果省略,最后一次迭代将返回 {value: undefined, done: true},生成器完成后对 next() 的任何后续调用也是如此。

    (4)遍历生成器

    使用 next() 方法可以遍历生成器对象,接收完整对象的所有 value 和 done 属性。不过,就像 Array、Map 和 Set 一样,Generator 遵循迭代协议,并且可以使用 for...of 进行迭代:

    function* generatorFunction() {
      yield 'One'
      yield 'Two'
      yield 'Three'
    
      return 'Hello, Generator!'
    }
    
    const generator = generatorFunction()
    
    for (const value of generator) {
      console.log(value)
    }

    输出结果如下:

    One
    Two
    Three

    扩展运算符也可用于将生成器的值分配给数组:

    const values = [...generator]
    
    console.log(values)

    输出结果如下:

    ['One', 'Two', 'Three']

    可以看到,扩展运算符和 for...of 都不会将 return 的值计入 value。

    注意:虽然这两种方法对于有限生成器都是有效的,但如果生成器正在处理无限数据流,则无法在不创建无限循环的情况下直接使用扩展运算符或 for...of。

    我们还可以从迭代结果中解构值:

    const [a, b, c]= generator;
    console.log(a);
    console.log(b);
    console.log(c);

    输出结果如下:

    One
    Two
    Three

    (5)关闭生成器

    如我们所见,生成器可以通过遍历其所有值将其 done 属性设置为 true 并将其状态设置为 closed 。除此之外,还有两种方法可以立即关闭生成器:使用 return() 方法和使用 throw() 方法。

    使用 return(),生成器可以在任何时候终止,就像在函数体中的 return 语句一样。可以将参数传递给 return(),或将其留空以表示未定义的值。

    下面来创建一个具有 yield 值但在函数定义中没有 return 的生成器:

    function* generatorFunction() {
      yield 'One'
      yield 'Two'
      yield 'Three'
    }
    
    const generator = generatorFunction()

    第一个 next() 将返回“One”,并将 done 设置为 false。如果在那之后立即在生成器对象上调用 return() 方法,将获得传递的值并将 done 设置为 true。对 next() 的任何额外调用都会给出默认的已完成生成器响应,其中包含一个 undefined 值。

    generator.next()
    generator.return('Return!')
    generator.next()

    输出结果如下:

    {value: "Neo", done: false}
    {value: "Return!", done: true}
    {value: undefined, done: true}

    return() 方法会强制生成器对象完成并忽略任何其他 yield 关键字。当需要使函数可取消时,这在异步编程中特别有用,例如当用户想要执行不同的操作时中断数据请求,因为无法直接取消 Promise。

    如果生成器函数的主体有捕获和处理错误的方法,则可以使用 throw() 方法将错误抛出到生成器中。这将启动生成器,抛出错误并终止生成器。

    下面来在生成器函数体内放一个 try...catch 并在发现错误时记录错误:

    function* generatorFunction() {
      try {
        yield 'One'
       yield 'Two'
      } catch (error) {
        console.log(error)
      }
    }
    
    const generator = generatorFunction()

    现在来运行 next() 方法,然后运行 throw() 方法:

    generator.next()
    generator.throw(new Error('Error!'))

    输出结果如下:

    {value: "One", done: false}
    Error: Error!
    {value: undefined, done: true}

    使用 throw() 可以将错误注入到生成器中,该错误被 try...catch 捕获并记录到控制台。

    (6)生成器对象方法和状态

    下面是生成器对象的方法:

    • next():返回生成器中的后面的值;
    • return():在生成器中返回一个值并结束生成器;
    • throw():抛出错误并结束生成器。

    下面是生成器对象的状态:

    • suspended:生成器已停止执行但尚未终止。
    • closed:生成器因遇到错误、返回或遍历所有值而终止。

    (7)yield 委托

    除了常规的 yield 运算符之外,生成器还可以使用 yield* 表达式将更多值委托给另一个生成器。当在生成器中遇到 yield* 时,它将进入委托生成器并开始遍历所有 yield 直到该生成器关闭。这可以用于分离不同的生成器函数以在语义上组织代码,同时仍然让它们的所有 **yield** 都可以按正确的顺序迭代。

    下面来创建两个生成器函数,其中一个将对另一个进行 yield* 操作:

    function* delegate() {
      yield 3
      yield 4
    }
    
    function* begin() {
      yield 1
      yield 2
      yield* delegate()
    }

    接下来,遍历 begin() 生成器函数:

    const generator = begin()
    
    for (const value of generator) {
      console.log(value)
    }

    输出结果如下:

    1
    2
    3
    4

    外部的生成器(begin)生成值 1 和 2,然后使用 yield* 委托给另一个生成器(delegate),返回 3 和 4。

    yield* 还可以委托给任何可迭代的对象,例如 Array 或 Map。yield 委托有助于组织代码,因为生成器中任何想要使用 yield 的函数也必须是一个生成器。

    (8)在生成器中传递值

    上面的例子中,我们使用生成器作为迭代器,并且在每次迭代中产生值。除了产生值之外,生成器还可以使用 next() 中的值。在这种情况下,yield 将包含一个值。

    需要注意,调用的第一个 next() 不会传递值,而只会启动生成器。为了证明这一点,可以记录 yield 的值并使用一些值调用 next() 几次。

    function* generatorFunction() {
      console.log(yield)
      console.log(yield)
    
      return 'End'
    }
    
    const generator = generatorFunction()
    
    generator.next()
    generator.next(100)
    generator.next(200)

    输出结果如下:

    100
    200
    {value: "End", done: true}

    除此之外,也可以为生成器提供初始值。下面来创建一个 for 循环并将每个值传递给 next() 方法,同时将一个参数传递给 inital 函数:

    function* generatorFunction(value) {
      while (true) {
        value = yield value * 10
      }
    }
    
    const generator = generatorFunction(0)
    
    for (let i = 0; i < 5; i++) {
      console.log(generator.next(i).value)
    }

    这将从 next() 中检索值并为下一次迭代生成一个新值,该值是前一个值乘以 10。输出结果如下:

    0
    10
    20
    30
    40

    处理启动生成器的另一种方法是将生成器包装在一个函数中,该函数将会在执行任何其他操作之前调用 next() 一次。

    (9)async/await

    async/await 使处理异步数据更简单、更容易理解。生成器具有比异步函数更广泛的功能,但能够复制类似的行为。以这种方式实现异步编程可以增加代码的灵活性。

    下面来构建一个异步函数,它使用 Fetch API 获取数据并将响应记录到控制台。

    首先定义一个名为 getUsers 的异步函数,该函数从 API 获取数据并返回一个对象数组,然后调用 getUsers:

    const getUsers = async function () {
      const response = await fetch('https://jsonplaceholder.typicode.com/users')
      const json = await response.json()
    
      return json
    }
    
    getUsers().then((response) => console.log(response))

    输出结果如下:

    图片图片

    使用生成器可以创建几乎相同但不使用 async/await 关键字的效果。相反,它将使用我们创建的新函数,并产生值而不是等待 Promise。

    const getUsers = asyncAlt(function* () {
      const response = yield fetch('https://jsonplaceholder.typicode.com/users')
      const json = yield response.json()
    
      return json
    })
    
    getUsers().then((response) => console.log(response))

    如我们所见,它看起来与 async/await 实现几乎相同,除了有一个生成器函数被传入以产生值。

    现在可以创建一个类似于异步函数的 asyncAlt 函数。asyncAlt 有一个 generatorFunction 参数,它是产生 fetch 返回的 Promise 的函数。asyncAlt 返回函数本身,并 resolve 它得到的每个 Promise,直到最后一个:

    function asyncAlt(generatorFunction) {
      return function () {
        // 创建并分配生成器对象
        const generator = generatorFunction()
    
        // 定义一个接受生成器下一次迭代的函数
        function resolve(next) {
          // 如果生成器关闭并且没有更多的值可以生成,则解析最后一个值
          if (next.done) {
            return Promise.resolve(next.value)
          }
    
          // 如果仍有值可以产生,那么它们就是Promise,必须 resolved。
          return Promise.resolve(next.value).then((response) => {
            return resolve(generator.next(response))
          })
        }
    
        // 开始 resolve Promise
        return resolve(generator.next())
      }
    }

    这样就会得到和async/await一样的结果:

    图片图片

    尽管这个方法可以为代码增加灵活性,但通常 async/await 是更好的选择,因为它抽象了实现细节并让开发者专注于编写高效代码。

    (10)使用场景

    很多开发人员认为生成器函数视为一种奇特的 JavaScript 功能,在现实中几乎没有应用。在大多数情况下,确实用不到生成器。

    生成器的优点:

    • 惰性求值:除非需要,否则不计算值。它提供按需计算。只有需要它时,value 才会存在。
    • 内存效率高:由于惰性求值,生成器的内存效率非常高,因为它不会为预先生成的未使用值分配不必要的内存位置。
    • 更简洁的代码:生成器提供更简洁的代码,尤其是在异步行为中。

    生成器在对性能要求高的场景中有很大的用处。特别是,它们适用于以下场景:

    • 处理大文件和数据集。
    • 生成无限的数据序列。
    • 按需计算昂贵的逻辑。

    Redux sagas 就是实践中使用的生成器的一个很好的例子。它是一个用于管理redux应用异步操作的中间件,redux-saga 通过创建 sagas 将所有异步操作逻辑收集在一个地方集中处理,可以用来代替 redux-thunk 中间件。

    2. 异步生成器

    ECMAScript 2018 中引入了异步生成器的概念,它是一种特殊类型的异步函数,可以随意停止和恢复其执行。

    同步生成器函数和异步生成器函数的区别在于,后者从迭代器对象返回一个异步的、基于 Promise 的结果。

    要想创建异步生成器函数,需要声明一个带有星号 * 的生成器函数,前缀为 async:

    async function* asyncGenerator() {
    
    }

    一旦进入函数,就可以使用 yield 来暂停执行:

    async function* asyncGenerator() {
      yield 'One'
      yield 'Two'
    }

    这里 yield 会暂停执行并返回一个迭代器对象给调用者。这个对象既是可迭代对象,又是迭代器。

    异步生成器函数不会像常规函数那样在一步中计算出所有结果。相反,它会逐步提取值。我们可以使用两种方法从异步生成器解析 Promise:

    • 在迭代器对象上调用 next();
    • 使用 for await...of 异步迭代。

    对于上面的例子,可以这样做:

    async function* asyncGenerator() {
      yield 'One';
      yield 'Two';
    }
    
    const go = asyncGenerator();
    
    go.next().then(iterator => console.log(iterator.value));
    go.next().then(iterator => console.log(iterator.value));

    输出结果如下:

    'One';
    'Two'

    另一种方法使用异步迭代 for await...of。要使用异步迭代,需要用 async 函数包装它:

    async function* asyncGenerator() {
      yield 'One';
      yield 'Two';
    }
    
    async function consumer() {
      for await (const value of asyncGenerator()) {
        console.log(value);
      }
    }
    
    consumer();

    for await...of 非常适合提取非有限数据流。

    相关文章

    JavaScript2024新功能:Object.groupBy、正则表达式v标志
    PHP trim 函数对多字节字符的使用和限制
    新函数 json_validate() 、randomizer 类扩展…20 个PHP 8.3 新特性全面解析
    使用HTMX为WordPress增效:如何在不使用复杂框架的情况下增强平台功能
    为React 19做准备:WordPress 6.6用户指南
    如何删除WordPress中的所有评论

    发布评论