Json 到 node.js 中的 csv

Question

我正在尝试在 node.js 中将一个非常大的 json 转换为 csv，但它花费了太多时间并且在转换时还导致 100% cpu。

  jsonToCsv: function (data) {
    var keys = Object.keys(data[0]);
    var csv = [keys.join(",")];
    console.time("CSVGeneration");
    data.forEach(function (row) {
      var line = ''; 
      keys.forEach(function (key) {
        if (typeof row[key] === 'string') {
          row[key] = "" + file_utils.escapeCsv(row[key]) + "";
        } 
        line += row[key] + ",";
      });
      csv.push(line);  
    });
    console.timeEnd("CSVGeneration");
    csv = csv.join("\n");
    return csv;
  },
  escapeCsv: function (x) {
    if (x)
      return ('' + x.replace(/"/g, '').replace(/,/g, ' ').replace(/\n/g, " ").replace(/\r/g, " ") + '');
    else
      return ('');
  },

平均运行 1Lac 行时，它从未恢复到甚至记录时间。我不得不手动终止该进程。

有人可以提出更好的替代方案吗？

Answer 1

在回答这个问题之前：假设您的代码正在运行，这个问题属于https://codereview.stackexchange.com/。

至于你的问题：

• 像forEach() 这样的新数组访问函数虽然在编码时相当舒适，但通常性能不佳。一个简单的for 循环是性能关键情况下的更好选择。
• 在escapeCsv() 中，您应用了4 个不同的正则表达式替换，每个替换为一个字符。将它们合二为一。
• 假设您的数据已经以某种方式结构化，允许 Csv 转换（data 是一个对象数组，每个对象都具有相同的属性），没有必要为每个对象单独检索键。李>

应用这个，产生以下代码：

function escapeCsv(x) {
    if (x) {
        return ('' + x).replace( /[",\n\r]/gi, '' );
    } else {
        return ('');
    }
}

function jsonToCsv(data) {
    var keys = Object.keys(data[0]),
        csv = [keys.join(",")];

    var row = new Array( keys.length );
    for (var i = 0; i < data.length; i++) {
        for (var j = 0; j < keys.length; j++) {
            if (typeof data[i][keys[j]] === 'string') {
                row[j] = '"' + escapeCsv(data[i][keys[j]]) + '"';
            } else {
                row[j] = data[i][keys[j]] || '';
            }
        }
        csv.push(row.join(','));
    }

    return csv.join("\n");
}

根据 jsPerf，仅此一项就可以提高大约 3-5 的性能。

如果您生成的 CSV 可以流式传输到文件或直接传输到客户端，则可以进一步改进并减少内存负载，因为 CSV 不必存储在内存中。

Fiddle to play around with the functions原名和你一样，新的名字后缀2。

jsPerf.com comparison