Node.js下MongoDB数据填充：mongoose-data-seed无法传递ObjectId引用问题

我之前也碰到过一模一样的问题——mongoose-data-seed默认把每个seed文件当成独立单元运行，确实没内置跨文件传递数据的机制，但我们可以通过两种实用的方式解决这个问题：

方法1：用全局共享模块存储生成的ObjectId #

这种方式适合你需要精准控制关联ID、不想依赖数据库查询的场景。

步骤1：创建共享数据模块 #

在项目根目录新建一个shared-seed-ids.js文件，专门用来暂存我们生成的关联ID：

// shared-seed-ids.js
const sharedIds = {
  users: [],
  posts: []
};

module.exports = sharedIds;

步骤2：编写主数据Seed文件（比如用户数据） #

生成用户数据时，把每个用户的_id保存到共享模块里：

// seeds/users.seed.js
const { Seeder } = require('mongoose-data-seed');
const User = require('../models/User');
const sharedIds = require('../shared-seed-ids');

const data = [
  { name: 'Alice', email: 'alice@example.com' },
  { name: 'Bob', email: 'bob@example.com' }
];

class UsersSeeder extends Seeder {
  async shouldRun() {
    return User.countDocuments().exec().then(count => count === 0);
  }

  async run() {
    // 创建用户并把生成的ObjectId存入共享模块
    const createdUsers = await User.create(data);
    sharedIds.users = createdUsers.map(user => user._id);
    return createdUsers;
  }
}

module.exports = UsersSeeder;

步骤3：编写关联数据Seed文件（比如帖子数据） #

生成帖子时，直接从共享模块里取出用户的ObjectId作为引用：

// seeds/posts.seed.js
const { Seeder } = require('mongoose-data-seed');
const Post = require('../models/Post');
const sharedIds = require('../shared-seed-ids');

class PostsSeeder extends Seeder {
  async shouldRun() {
    return Post.countDocuments().exec().then(count => count === 0);
  }

  async run() {
    const postData = [
      { title: 'First Post', content: 'Hello World', author: sharedIds.users[0] },
      { title: 'Second Post', content: 'Mongoose Seed', author: sharedIds.users[1] }
    ];
    return Post.create(postData);
  }
}

module.exports = PostsSeeder;

步骤4：配置Seed执行顺序 #

在seeders/index.js里确保主数据Seed先执行，这样关联数据Seed运行时共享模块里已经有数据了：

// seeders/index.js
const UsersSeeder = require('./users.seed');
const PostsSeeder = require('./posts.seed');

module.exports = [
  UsersSeeder,
  PostsSeeder
];

方法2：在关联Seed中直接查询数据库获取ObjectId #

如果你不想用全局模块，也可以在关联数据的Seed文件里直接查询已生成的主数据来获取ObjectId，这种方式更可靠，避免全局变量的潜在问题：

// seeds/posts.seed.js
const { Seeder } = require('mongoose-data-seed');
const Post = require('../models/Post');
const User = require('../models/User');

class PostsSeeder extends Seeder {
  async shouldRun() {
    return Post.countDocuments().exec().then(count => count === 0);
  }

  async run() {
    // 先查询已生成的用户文档
    const alice = await User.findOne({ email: 'alice@example.com' });
    const bob = await User.findOne({ email: 'bob@example.com' });

    const postData = [
      { title: 'First Post', content: 'Hello World', author: alice._id },
      { title: 'Second Post', content: 'Mongoose Seed', author: bob._id }
    ];
    return Post.create(postData);
  }
}

module.exports = PostsSeeder;

只要保证主数据Seed在seeders/index.js里排在前面，就能确保查询到对应的文档。

额外提醒：Mongoose Schema的引用配置 #

别忘了在Schema里正确配置引用关系，这样后续用populate()方法时才能正常关联数据：

// models/Post.js
const mongoose = require('mongoose');
const Schema = mongoose.Schema;

const postSchema = new Schema({
  title: String,
  content: String,
  author: {
    type: Schema.Types.ObjectId,
    ref: 'User' // 关联User模型
  }
});

module.exports = mongoose.model('Post', postSchema);

内容的提问来源于stack exchange，提问作者Zhong Huiwen