Mongoose 中的关联查询详解：populate 的用法与实现原理-JavaScript中文网-JavaScript教程资源分享门户

在 Node.js 的 Web 开发中，Mongoose 是一个广泛使用的 MongoDB ODM（Object-Document Mapping）库。它提供了丰富的 API，使得开发者可以更方便地操作 MongoDB 数据库。其中，关联查询是 Mongoose 中的一个重要功能，它可以让我们在查询数据时，同时查询关联的数据，从而减少数据库的访问次数，提高查询效率。

本文将详细介绍 Mongoose 中的关联查询功能，包括 populate 的用法、实现原理和示例代码，帮助读者更好地理解和使用这一功能。

populate 的用法

在 Mongoose 中，我们可以使用 populate 方法进行关联查询。它的基本用法如下：

Model.find({}).populate('path').exec(callback);

其中，Model 是 Mongoose 中的模型对象，path 是需要关联查询的字段，callback 是查询完成后的回调函数。具体来说，path 可以是一个字符串，也可以是一个对象，它指定了要关联查询的字段和相关的选项。下面是一些常见的 path 用法：

字符串形式：'path'，表示查询结果中，将 path 字段关联的文档一并查询出来。
对象形式：{ path: 'path', select: 'field -_id', options: { sort: { field: -1 } }, match: { field: { $gt: 100 } } }，其中：
- path：要关联查询的字段。
- select：指定要查询的字段，以空格分隔。可以使用 - 表示不查询某个字段。
- options：指定查询选项，比如排序、限制查询数量等。
- match：指定查询条件，使用 MongoDB 的查询语法。

下面是一个简单的示例，演示了如何使用 populate 方法进行关联查询：

// www.javascriptcn.com code example
// 定义两个模型
const Author = mongoose.model('Author', new mongoose.Schema({
  name: String,
  age: Number
}));

const Book = mongoose.model('Book', new mongoose.Schema({
  title: String,
  author: {
    type: mongoose.Schema.Types.ObjectId,
    ref: 'Author'
  }
}));

// 创建一个作者和一本书
const author = new Author({
  name: 'John',
  age: 30
});

const book = new Book({
  title: 'The Book',
  author: author._id
});

author.save();
book.save();

// 查询书籍，并关联查询作者信息
Book.find().populate('author').exec((err, books) => {
  console.log(books);
  // 输出：
  // [{
  //   title: 'The Book',
  //   author: { name: 'John', age: 30, _id: 5c8e97c6fc9f1c22f8d8d8c7 }
  // }]
});

在上面的示例中，我们定义了两个模型：Author 和 Book。Book 模型中有一个 author 字段，它是一个 ObjectId 类型，并通过 ref 属性指向 Author 模型。这样，我们就可以通过 populate 方法，查询书籍时同时查询作者信息，并将查询结果中的 author 字段替换成完整的作者文档。

实现原理

populate 方法的实现原理其实很简单：它会将查询条件中的所有 path 字段，转换成 MongoDB 的聚合管道操作，最终生成一个聚合查询。这个聚合查询会先查询出主模型的文档，然后在关联模型中查询符合条件的文档，并将其合并到主模型文档中。

具体来说，populate 方法会根据 path 字段生成一个 $lookup 聚合操作，用来在关联模型中查询符合条件的文档。$lookup 操作的语法如下：

{
  $lookup: {
    from: <collection>,
    localField: <field>,
    foreignField: <field>,
    as: <output>
  }
}

其中，from 表示关联的集合，localField 表示主模型的字段，foreignField 表示关联模型的字段，as 表示输出的字段名。$lookup 操作会返回一个数组，包含了查询结果中每个主模型文档关联的所有关联模型文档。

除了 $lookup 操作外，populate 方法还会根据 path 字段的其他选项，生成一些其他的聚合操作，比如 $project、$match、$sort 等，用来对查询结果进行进一步的处理。

示例代码

下面是一个更复杂的示例，演示了如何使用 populate 方法进行多层级的关联查询，并使用 $match 和 $sort 操作对查询结果进行处理：

// www.javascriptcn.com code example
// 定义三个模型
const User = mongoose.model('User', new mongoose.Schema({
  name: String,
  age: Number
}));

const Post = mongoose.model('Post', new mongoose.Schema({
  title: String,
  content: String,
  author: {
    type: mongoose.Schema.Types.ObjectId,
    ref: 'User'
  },
  comments: [{
    type: mongoose.Schema.Types.ObjectId,
    ref: 'Comment'
  }]
}));

const Comment = mongoose.model('Comment', new mongoose.Schema({
  content: String,
  author: {
    type: mongoose.Schema.Types.ObjectId,
    ref: 'User'
  },
  post: {
    type: mongoose.Schema.Types.ObjectId,
    ref: 'Post'
  }
}));

// 创建一个用户、一篇文章和两个评论
const user = new User({
  name: 'John',
  age: 30
});

const post = new Post({
  title: 'The Post',
  content: 'Hello world!',
  author: user._id
});

const comment1 = new Comment({
  content: 'Nice post!',
  author: user._id,
  post: post._id
});

const comment2 = new Comment({
  content: 'Great post!',
  author: user._id,
  post: post._id
});

user.save();
post.save();
comment1.save();
comment2.save();

// 查询文章，并关联查询作者信息、评论信息和评论作者信息
Post.find().populate({
  path: 'author comments',
  select: 'name age content',
  populate: {
    path: 'author',
    select: 'name age'
  }
}).match({
  'comments.content': /great/i
}).sort({
  'comments.content': -1
}).exec((err, posts) => {
  console.log(posts);
  // 输出：
  // [{
  //   title: 'The Post',
  //   content: 'Hello world!',
  //   author: { name: 'John', age: 30, _id: 5c8e9a1cfc9f1c22f8d8d8c9 },
  //   comments: [{
  //     content: 'Great post!',
  //     author: { name: 'John', age: 30 },
  //     post: 5c8e9a1cfc9f1c22f8d8d8ca,
  //     _id: 5c8e9a1cfc9f1c22f8d8d8cb
  //   }]
  // }]
});

在上面的示例中，我们定义了三个模型：User、Post 和 Comment。Post 模型中有一个 author 字段和一个 comments 字段，它们分别指向 User 和 Comment 模型。Comment 模型中有一个 author 字段和一个 post 字段，它们分别指向 User 和 Post 模型。

我们创建了一个用户、一篇文章和两个评论，然后使用 populate 方法进行关联查询。具体来说，我们关联查询了文章的作者信息和评论信息，并在评论信息中再次关联查询了评论作者信息。我们还使用 $match 操作筛选了评论内容中包含 great 的评论，并使用 $sort 操作按评论内容降序排列。

最终，我们得到了一篇文章和一个评论，它们分别包含了作者信息、评论信息和评论作者信息。这就是 populate 方法的强大之处，它可以让我们在查询数据时，轻松地获取所有相关的信息，从而简化代码，提高效率。

来源：JavaScript中文网，转载请注明来源 https://www.javascriptcn.com/post/67d423eba941bf71347d2a34

Mongoose 中的关联查询详解：populate 的用法与实现原理

populate 的用法

实现原理

示例代码

接着读