使用node.js和ES6 / ES7功能

时间:2016-07-11 18:35:44

标签: javascript node.js csv ecmascript-6 ecmascript-7

在Python中逐行读取CSV文件(即不将整个文件加载到内存中)很简单:

import csv
for row in csv.reader(open("file.csv")):
    print(row[0])

对node.js执行相同操作涉及使用node-csv,流和回调等内容。

是否可以使用新的ES6 / ES7功能(如迭代器,生成器,承诺和异步函数)以更像Python代码的方式迭代CSV文件的行?

理想情况下,我希望能够写出这样的内容:

for (const row of csvOpen('file.csv')) {
  console.log(row[0]);
}

(同样,不会立即将整个文件加载到内存中。)

1 个答案:

答案 0 :(得分:1)

我不熟悉node-csv,但听起来像使用生成器的迭代器应该这样做。只需将其包装在任何异步回调API周围:

let dummyReader = {
  testFile: ["row1", "row2", "row3"],
  read: function(cb) {
    return Promise.resolve().then(() => cb(this.testFile.shift())).catch(failed);
  },
  end: function() {
    return !this.testFile.length;
  }
}

let csvOpen = url => {
  let iter = {};
  iter[Symbol.iterator] = function* () {
    while (!dummyReader.end()) {
      yield new Promise(resolve => dummyReader.read(resolve));
    }
  }
  return iter;
};

async function test() {
  // The line you wanted:
  for (let row of csvOpen('file.csv')) {
    console.log(await row);
  }
}

test(); // row1, row2, row3

var failed = e => console.log(e.name +": "+ e.message);

请注意row这是一个承诺,但足够接近。

将其粘贴到babel