在 JavaScript 中优雅的提取循环内的数据

翻译：疯狂的技术宅

原文：2ality.com/2018/04/ext…

在本文中，咱们将介绍两种提取循环内数据的方法：内部迭代和外部迭代。

循环

举个例子，假设有一个函数 logFiles()：

const fs = require('fs');
const path = require('path');

function logFiles(dir) {
  for (const fileName of fs.readdirSync(dir)) { // (A)
    const filePath = path.resolve(dir, fileName);
    console.log(filePath);
    const stats = fs.statSync(filePath);
    if (stats.isDirectory()) {
      logFiles(filePath); // (B)
    }
  }
}
logFiles(process.argv[2]);
复制代码

从 A 行开始的循环用来记录文件路径。它是 for-of 循环和递归的组合（递归调用在 B 行）。

若是你发现循环内的某些数据（迭代文件）有用，但又不想记录它，那应该怎么办？

内部迭代

提取循环内数据的第一个方法是内部迭代：

const fs = require('fs');
const path = require('path');

function logFiles(dir, callback) {
  for (const fileName of fs.readdirSync(dir)) {
    const filePath = path.resolve(dir, fileName);
    callback(filePath); // (A)
    const stats = fs.statSync(filePath);
    if (stats.isDirectory()) {
      logFiles(filePath, callback);
    }
  }
}
logFiles(process.argv[2], p => console.log(p));
复制代码

这种迭代方式与Array的 .forEach()相似：logFiles() 内实现循环并对每一个迭代值（行A）调用 callback。

外部迭代

内部迭代的替代方案是外部迭代：咱们实现了一个iterable，能够用生成器帮助咱们实现：

const fs = require('fs');
const path = require('path');

function* logFiles(dir) {
  for (const fileName of fs.readdirSync(dir)) {
    const filePath = path.resolve(dir, fileName);
    yield filePath;
    const stats = fs.statSync(filePath);
    if (stats.isDirectory()) {
      yield* logFiles(filePath); // (A)
    }
  }
}
for (const p of logFiles(process.argv[2])) {
  console.log(p);
}
复制代码

若是是内部迭代，logFiles() 会调用咱们（“推”给咱们）。而这一次，换咱们来调用它了（“拉”过来）。

请注意，在生成器中，必须经过 yield* 进行递归调用（第A行）：若是只调用 logFiles() 那么它会返回一个iterable。但咱们想要的是在该 iterable 中 yield 每一个项目。这就是 yield* 的做用。

生成器有一个很是好的特性，就是处理过程可以与内部迭代同样互锁：每当 logFiles() 建立另外一个 filePath 时，咱们可以当即查看它，而后 logFiles() 继续。这是一种简单的协做式多任务处理，其中 yield 暂停当前任务并切换到另外一个任务。

扩展阅读

• Chapter “Iterables and iterators” in “Exploring ES6”.
• Chapter “Generators” in “Exploring ES6”.

欢迎关注京程一灯公众号：jingchengyideng，获取更多前端干货内容。