如何使用 JS/node 正确使用 async/await

Question

我正在尝试创建一个脚本，该脚本将使用从几个 API 调用返回的数据填充我的 Mongo 数据库。理想情况下，这将每隔一段时间手动运行一次，或者在调度程序上运行以使用新数据更新数据库，因此我正在尝试在“MongoClient.connect”函数中执行所有操作。我想一次导入这个数据一个 API 调用，并试图避免链接承诺和 .then 的丑陋。我已经对使用 async/await 进行了一些研究，在我看来，使用 'await' 关键字可以有效地解开返回的承诺。如果是这种情况，我无法完全弄清楚为什么它没有正确等待。这是我目前所拥有的：

const MongoClient = require('mongodb').MongoClient;
const fs = require('fs');
const assert = require('assert');
const ProPublicaAPI = require('./ProPublicaAPI');
var constants = require('../common/constants');
var https = require('https');

const url = 'mongodb://localhost:27017';


dbName = 'government';

MongoClient.connect(url, function(err, client) {
  assert.equal(null, err);
  console.log('successfully connected to mongoDB server: ' + url);

  const db = client.db(dbName);
  try{
    var senateInsert = waitInsert(db, 'senate');
  }
  catch(err) {
    console.log(err);
  }
  console.log('closing database connection');
  client.close();
})

async function waitInsert(db, collection) {
  //I made this function just to call inside the MongoClient.connect() above
  return await insertSenateDocuments(db, collection);
}

function insertSenateDocuments(db, chamber) {
  return new Promise(function(resolve) {
    var collection = db.collection(chamber);
    //method that calls API, to be converted into async/await later
    senateMembers = ProPublicaAPI.getAllMembersByChamber(chamber);
    senateMembers.then(function(promisedMembers) {
      console.log("Recieved data from api");
      collection.insertMany(promisedMembers, function(err, result) {
        if(err) return console.log(err);
        console.log(`Imported ${chamber} data`);
        resolve(result);
      });
    })
  })
}

这是堆栈跟踪：

successfully connected to mongoDB server: mongodb://localhost:27017
about to close
Recieved collection Data
{ MongoError: server instance pool was destroyed
    at basicWriteValidations (D:\Dev\node\node_modules\mongodb-core\lib\topologies\server.js:636:41)
    at Server.insert (D:\Dev\node\node_modules\mongodb-core\lib\topologies\server.js:737:16)
    at Server.insert (D:\Dev\node\node_modules\mongodb\lib\topologies\topology_base.js:315:25)
    at executeCommands (D:\Dev\node\node_modules\mongodb\lib\bulk\ordered.js:525:23)
    at executeOperation (D:\Dev\node\node_modules\mongodb\lib\utils.js:408:22)
    at OrderedBulkOperation.execute (D:\Dev\node\node_modules\mongodb\lib\bulk\ordered.js:602:10)

    at bulkWrite (D:\Dev\node\node_modules\mongodb\lib\collection.js:627:8)
    at executeOperation (D:\Dev\node\node_modules\mongodb\lib\utils.js:408:22)
    at Collection.insertMany (D:\Dev\node\node_modules\mongodb\lib\collection.js:512:10)
    at D:\Dev\node\dataimporter\mongo.js:39:18
  name: 'MongoError',
  message: 'server instance pool was destroyed' }
PS D:\Dev\node\dataimporter> node mongo
successfully connected to mongoDB server: mongodb://localhost:27017
closing database connection
Recieved data from api
{ MongoError: server instance pool was destroyed
    at basicWriteValidations (D:\Dev\node\node_modules\mongodb-core\lib\topologies\server.js:636:41)
    at Server.insert (D:\Dev\node\node_modules\mongodb-core\lib\topologies\server.js:737:16)
    at Server.insert (D:\Dev\node\node_modules\mongodb\lib\topologies\topology_base.js:315:25)
    at executeCommands (D:\Dev\node\node_modules\mongodb\lib\bulk\ordered.js:525:23)
    at executeOperation (D:\Dev\node\node_modules\mongodb\lib\utils.js:408:22)
    at OrderedBulkOperation.execute (D:\Dev\node\node_modules\mongodb\lib\bulk\ordered.js:602:10)
    at bulkWrite (D:\Dev\node\node_modules\mongodb\lib\collection.js:627:8)
    at executeOperation (D:\Dev\node\node_modules\mongodb\lib\utils.js:408:22)
    at Collection.insertMany (D:\Dev\node\node_modules\mongodb\lib\collection.js:512:10)
    at D:\Dev\node\dataimporter\mongo.js:39:18
  name: 'MongoError',
  message: 'server instance pool was destroyed' }

如您所见，在插入文档之前调用了 client.close()。

Answer 1

你必须awaitwaitInsert

MongoClient.connect(url, async function(err, client) {
  assert.equal(null, err);
  console.log('successfully connected to mongoDB server: ' + url);

  const db = client.db(dbName);
  try{
    var senateInsert = await waitInsert(db, 'senate');
  }
  catch(err) {
    console.log(err);
  }
  console.log('closing database connection');
  client.close();
})

Answer 2

更简单：

bulkWriteItems(array, collection) {
    return collection.bulkWrite(array, {ordered: true, w: 1})
}

呼叫：

async saveData() {
    let bulk = [];
    for (let i = 0; i < 100; i++) {
        let name = `item${i}`
        bulk.push({insertOne: {name}});
    }
    return db.collection(COLLECTION_OBJECTS).then(collection => this.bulkWriteItems(bulk, collection))
}