目前我需要将一个大型 CSV 文件推送到 mongo 数据库中，值的顺序需要确定数据库条目的键:

示例 CSV 文件:

9,1557,358,286,Mutantville,4368,2358026,,M,0,0,0,1,0
9,1557,359,147,Wroogny,4853,2356061,,D,0,0,0,1,0

将其解析为数组的代码:

var fs = require("fs");

var csv = require("fast-csv");

fs.createReadStream("rank.txt")
    .pipe(csv())
    .on("data", function(data){
        console.log(data);
    })
    .on("end", function(data){
        console.log("Read Finished");
    });

代码输出:

[ '9',
  '1557',
  '358',
  '286',
  'Mutantville',
  '4368',
  '2358026',
  '',
  'M',
  '0',
  '0',
  '0',
  '1',
  '0' ]
[ '9',
  '1557',
  '359',
  '147',
  'Wroogny',
  '4853',
  '2356061',
  '',
  'D',
  '0',
  '0',
  '0',
  '1',
  '0' ]

如何将数组插入我的 mongoose 模式以进入 mongo 数据库？

架构:

var mongoose = require("mongoose");


var rankSchema = new mongoose.Schema({
   serverid: Number,
   resetid: Number,
   rank: Number,
   number: Number,
   name: String,
   land: Number,
   networth: Number,
   tag: String,
   gov: String,
   gdi: Number,
   protection: Number,
   vacation: Number,
   alive: Number,
   deleted: Number
});

module.exports = mongoose.model("Rank", rankSchema);

数组的顺序需要与模式的顺序相匹配，例如在数组中第一个数字 9 需要始终保存为它们的键“serverid”等等。我正在使用 Node.JS

最佳答案

您可以通过获取 headers 使用 fast-csv 来完成此操作来自模式定义，它将解析的行作为“对象”返回。你实际上有一些不匹配，所以我用更正标记了它们:

const fs = require('mz/fs');
const csv = require('fast-csv');

const { Schema } = mongoose = require('mongoose');

const uri = 'mongodb://localhost/test';

mongoose.Promise = global.Promise;
mongoose.set('debug', true);

const rankSchema = new Schema({
  serverid: Number,
  resetid: Number,
  rank: Number,
  name: String,
  land: String,         // <-- You have this as Number but it's a string
  networth: Number,
  tag: String,
  stuff: String,        // the empty field in the csv
  gov: String,
  gdi: Number,
  protection: Number,
  vacation: Number,
  alive: Number,
  deleted: Number
});

const Rank = mongoose.model('Rank', rankSchema);

const log = data => console.log(JSON.stringify(data, undefined, 2));

(async function() {

  try {
    const conn = await mongoose.connect(uri);

    await Promise.all(Object.entries(conn.models).map(([k,m]) => m.remove()));

    let headers = Object.keys(Rank.schema.paths)
      .filter(k => ['_id','__v'].indexOf(k) === -1);

    console.log(headers);

    await new Promise((resolve,reject) => {

      let buffer = [],
          counter = 0;

      let stream = fs.createReadStream('input.csv')
        .pipe(csv({ headers }))
        .on("error", reject)
        .on("data", async doc => {
          stream.pause();
          buffer.push(doc);
          counter++;
          log(doc);
          try {
            if ( counter > 10000 ) {
              await Rank.insertMany(buffer);
              buffer = [];
              counter = 0;
            }
          } catch(e) {
            stream.destroy(e);
          }

          stream.resume();

        })
        .on("end", async () => {
          try {
            if ( counter > 0 ) {
              await Rank.insertMany(buffer);
              buffer = [];
              counter = 0;
              resolve();
            }
          } catch(e) {
            stream.destroy(e);
          }
        });

    });


  } catch(e) {
    console.error(e)
  } finally {
    process.exit()
  }


})()

只要架构实际上与提供的 CSV 一致，就可以了。这些是我可以看到的更正，但如果您需要以不同方式对齐的实际字段名称，则需要进行调整。但基本上有一个 Number在有String的位置本质上是一个额外的字段，我假设它是 CSV 中的空白字段。

一般的事情是从模式中获取字段名称数组，并在生成 csv 解析器实例时将其传递到选项中:

let headers = Object.keys(Rank.schema.paths)
  .filter(k => ['_id','__v'].indexOf(k) === -1);

let stream = fs.createReadStream('input.csv')
  .pipe(csv({ headers }))

一旦你真的这样做了，你就会得到一个“对象”而不是一个数组:

{
  "serverid": "9",
  "resetid": "1557",
  "rank": "358",
  "name": "286",
  "land": "Mutantville",
  "networth": "4368",
  "tag": "2358026",
  "stuff": "",
  "gov": "M",
  "gdi": "0",
  "protection": "0",
  "vacation": "0",
  "alive": "1",
  "deleted": "0"
}

不要担心“类型”，因为 Mongoose 会根据模式转换值。

其余的发生在 data 的处理程序中事件。为了获得最大效率，我们使用 insertMany() 每 10,000 行只写入一次数据库。实际如何到达服务器和进程取决于 MongoDB 版本，但根据内存使用的“权衡”和编写合理的网络请求。如有必要，请减小数字。

重要的部分是将这些调用标记为 async函数和 await insertMany() 的结果在继续之前。我们还需要 pause() 流和 resume() 在每个项目上，否则我们冒着覆盖 buffer 的风险在实际发送之前要插入的文件数量。 pause() 和 resume() 有必要在管道上施加“背压”，否则元素只会不断“出来”并发射 data事件。

自然地，控制 10,000 个条目需要我们在每次迭代和流完成时检查，以便清空缓冲区并将任何剩余文档发送到服务器。

这确实是您想要做的，因为您肯定不想在 data 的“每次”迭代中都向服务器发出异步请求。事件或基本上无需等待每个请求完成。对于“非常小的文件”，您可以不用检查它，但对于任何现实世界的负载，由于尚未完成的“飞行中”异步调用，您肯定会超出调用堆栈。

仅供引用 - package.json用过的。 mz 是可选的，因为它只是一个现代化的 Promise我习惯使用的标准 Node “内置”库的启用库。该代码当然可以与 fs 完全互换。模块。

{
  "description": "",
  "main": "index.js",
  "dependencies": {
    "fast-csv": "^2.4.1",
    "mongoose": "^5.1.1",
    "mz": "^2.7.0"
  },
  "keywords": [],
  "author": "",
  "license": "ISC"
}

实际上，对于 Node v8.9.x 及更高版本，我们甚至可以通过实现 AsyncIterator 来简化这一切。通过 stream-to-iterator 模块。还在Iterator<Promise<T>>模式，但它应该在 Node v10.x 成为稳定的 LTS 之前执行:

const fs = require('mz/fs');
const csv = require('fast-csv');
const streamToIterator = require('stream-to-iterator');

const { Schema } = mongoose = require('mongoose');

const uri = 'mongodb://localhost/test';

mongoose.Promise = global.Promise;
mongoose.set('debug', true);

const rankSchema = new Schema({
  serverid: Number,
  resetid: Number,
  rank: Number,
  name: String,
  land: String,
  networth: Number,
  tag: String,
  stuff: String,        // the empty field
  gov: String,
  gdi: Number,
  protection: Number,
  vacation: Number,
  alive: Number,
  deleted: Number
});

const Rank = mongoose.model('Rank', rankSchema);

const log = data => console.log(JSON.stringify(data, undefined, 2));

(async function() {

  try {
    const conn = await mongoose.connect(uri);

    await Promise.all(Object.entries(conn.models).map(([k,m]) => m.remove()));

    let headers = Object.keys(Rank.schema.paths)
      .filter(k => ['_id','__v'].indexOf(k) === -1);

    //console.log(headers);

    let stream = fs.createReadStream('input.csv')
      .pipe(csv({ headers }));

    const iterator = await streamToIterator(stream).init();

    let buffer = [],
        counter = 0;

    for ( let docPromise of iterator ) {
      let doc = await docPromise;
      buffer.push(doc);
      counter++;

      if ( counter > 10000 ) {
        await Rank.insertMany(buffer);
        buffer = [];
        counter = 0;
      }
    }

    if ( counter > 0 ) {
      await Rank.insertMany(buffer);
      buffer = [];
      counter = 0;
    }

  } catch(e) {
    console.error(e)
  } finally {
    process.exit()
  }

})()

基本上，所有流“事件”处理以及暂停和恢复都被简单的 for 取代了循环:

const iterator = await streamToIterator(stream).init();

for ( let docPromise of iterator ) {
  let doc = await docPromise;
  // ... The things in the loop
}

简单!这在以后的 Node 实现中使用 for..await..of 进行了清理当它变得更稳定时。但是以上在指定版本及更高版本上运行良好。

关于javascript - 使用 Mongoose 模式导入 CSV，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/50343116/