如何在NodeJS中使用db.command()执行原生MongoDB操作及自定义脚本

实现方案 #

你需要先引入文件系统相关模块读取txt脚本，再匹配执行对应的MongoDB操作，完整实现代码如下：

const { MongoClient } = require("mongodb");
const fs = require("fs/promises");
const path = require("path");

// 配置项
const uri = "mongodb:uri";
const SCRIPT_DIR = "./your-script-folder"; // 替换为你存放txt脚本的文件夹路径
const client = new MongoClient(uri);

// 执行单条MongoDB脚本的方法
async function executeMongoScript(db, scriptContent) {
  // 清除脚本前后空白和末尾分号，避免执行异常
  const cleanedScript = scriptContent.trim().replace(/;$/, "");
  // 匹配集合操作，格式为db.getCollection("集合名").操作方法(参数)
  const matchResult = cleanedScript.match(/db\.getCollection\("([^"]+)"\)\.(.+)\((.*)\)/);
  if (!matchResult) {
    throw new Error(`不支持的脚本格式: ${scriptContent}`);
  }
  const [, collectionName, operation, paramsStr] = matchResult;
  const collection = db.collection(collectionName);
  // 解析参数为JSON格式
  const params = paramsStr ? JSON.parse(paramsStr) : [];
  // 执行对应操作
  return await collection[operation](...(Array.isArray(params) ? params : [params]));
}

async function run() {
  try {
    await client.connect();
    const db = client.db("employee");

    // 读取脚本目录下所有文件
    const files = await fs.readdir(SCRIPT_DIR);
    // 过滤出txt文件
    const txtFiles = files.filter(file => path.extname(file).toLowerCase() === '.txt');

    // 遍历执行每个txt中的脚本
    for (const file of txtFiles) {
      const filePath = path.join(SCRIPT_DIR, file);
      const scriptContent = await fs.readFile(filePath, "utf8");
      // 支持单文件内多条脚本，按换行分割过滤空行
      const scripts = scriptContent.split(/\r?\n/).filter(line => line.trim());
      
      for (const script of scripts) {
        console.log(`正在执行脚本[${file}]: ${script}`);
        const result = await executeMongoScript(db, script);
        console.log(`执行成功，结果：`, result);
      }
    }
  } catch (err) {
    console.error("执行出错：", err);
  } finally {
    await client.close();
  }
}
run().catch(console.dir);

注意事项 #

• 以上实现默认你的txt脚本都是db.getCollection("xxx").xxx()格式的单条操作，如果你的脚本有更复杂的语法，可以改用Node.js的vm模块注入db上下文执行，可控场景下也可以直接使用eval，示例如下：

// 简单eval实现（仅推荐脚本都是你自己可控的场景使用）
async function executeMongoScriptByEval(db, scriptContent) {
  // 注入db变量到当前上下文
  global.db = db;
  return await eval(scriptContent);
}

• 建议给脚本加执行记录，避免重复执行同版本脚本，你可以在MongoDB里建一个专门的script_execution_log集合，每次执行完脚本就存储文件名、执行时间、执行结果，下次执行前先判断该脚本是否已经执行过，避免重复操作引发数据异常。

内容的提问来源于stack exchange，提问作者Prasad Kanaparthi