使用 $lookup 联接
本课属于 Coddy Node.js 学习旅程的“使用 MongoDB 的数据库”单元。第 25 课,共 29 课。
书中存储了作者的姓名,而作者的详细信息保存在另一个集合中。客户端希望在一次响应中获取两者。
$lookup 阶段用于联接。对于每个文档,它会在 from 集合中查找 foreignField 等于当前文档的 localField 的文档,并将它们放入一个由 as 命名的新数组字段中。
await orders.aggregate([
{ $match: { _id: orderId } },
{ $lookup: {
from: 'customers',
localField: 'customerEmail',
foreignField: 'email',
as: 'customer',
} },
]).toArray();
// [{ _id: ..., customerEmail: 'ada@x.io', customer: [{ email: 'ada@x.io', ... }] }]toArray() 会返回一个订单数组,这里数组中只有一个订单。取它的第一项即可获得订单本身。
localField 属于流水线中的文档,foreignField 属于另一个集合。即使只有一个匹配项,结果也会是一个数组。
from 集合会将所有匹配的文档放入由 as 命名的数组中。提供的路由会取流水线的第一个结果,以返回请求的那本书。
挑战
简单此步骤会打开课程预先准备好的书架 API 项目,而不是你在 Aggregation Pipelines 中的项目文件(包括 server.js)。你之前的作品仍保留在原来的课中。编辑 server.js 前,请先阅读提供的文件。这份副本包含书架路由、流派计数和评分最高项的聚合管道,以及预置的作者数据和响应处理。
在 server.js 中预先准备好的 GET /books/:id/full 路由里,在现有的匹配阶段之后添加一个查找阶段。通过 book.author 与 author.name 相等来关联作者。将匹配结果放入 authorInfo。保留提供的代码,它会获取第一个结果,并在图书不存在时返回 404。
试试看:使用 GET /books 并复制 Emma 的 _id。GET /books/<Emma’s _id>/full 会返回 Emma,其中的 authorInfo 数组包含 Jane Austen 和 born 1775。作者未知的图书会返回 authorInfo: []。有效但不存在的图书 ID 仍会得到 404。
亲手试试
import express from 'express';
import { MongoClient, ObjectId } from 'mongodb';
import { authorList } from './authors.js';
const client = new MongoClient('mongodb://localhost:27017');
const db = client.db('shelf');
const books = db.collection('books');
const ratings = db.collection('ratings');
const authors = db.collection('authors');
await books.createIndex({ isbn: 1 }, { unique: true });
await books.insertMany([
{ title: 'Dune', author: 'Frank Herbert', year: 1965, isbn: '9780441013593', genre: 'scifi' },
{ title: 'Kindred', author: 'Octavia E. Butler', year: 1979, isbn: '9780807083697', genre: 'scifi' },
{ title: 'Emma', author: 'Jane Austen', year: 1815, isbn: '9780141439587', genre: 'classic' },
]);
await authors.insertMany(authorList);
const app = express();
app.use(express.json());
// 所有错误都返回 { error: message }。
function sendError(res, status, message) {
res.status(status).json({ error: message });
}
// 将 :id 转为 ObjectId;如果 id 无效,则返回 400 响应和 null。
function parseId(req, res) {
if (ObjectId.isValid(req.params.id)) {
return new ObjectId(req.params.id);
}
sendError(res, 400, 'Not a valid id');
return null;
}
app.get('/', (req, res) => {
res.send('Bookshelf API');
});
app.get('/books', async (req, res) => {
const filter = {};
if (req.query.author) {
filter.author = req.query.author;
}
if (req.query.from || req.query.to) {
filter.year = {};
if (req.query.from) {
filter.year.$gte = Number(req.query.from);
}
if (req.query.to) {
filter.year.$lte = Number(req.query.to);
}
}
const list = await books.find(filter).toArray();
res.json(list);
});
app.post('/books', async (req, res) => {
const book = req.body;
try {
await books.insertOne(book);
} catch (err) {
if (err.code === 11000) {
return sendError(res, 409, 'That ISBN is already on the shelf');
}
throw err;
}
res.status(201).json(book);
});
app.get('/books/newest', async (req, res) => {
const n = Number(req.query.n) || 3;
const list = await books
.find({}, { projection: { title: 1, year: 1, _id: 0 } })
.sort({ year: -1 })
.limit(n)
.toArray();
res.json(list);
});
app.get('/books/:id', async (req, res) => {
const id = parseId(req, res);
if (!id) return;
const book = await books.findOne({ _id: id });
if (!book) {
return sendError(res, 404, 'No such book');
}
res.json(book);
});
app.patch('/books/:id', async (req, res) => {
const id = parseId(req, res);
if (!id) return;
const result = await books.updateOne({ _id: id }, { $set: req.body });
if (result.matchedCount === 0) {
return sendError(res, 404, 'No such book');
}
res.json(await books.findOne({ _id: id }));
});
app.post('/books/:id/copies', async (req, res) => {
const id = parseId(req, res);
if (!id) return;
const result = await books.updateOne({ _id: id }, { $inc: { copies: 1 } });
if (result.matchedCount === 0) {
return sendError(res, 404, 'No such book');
}
res.json(await books.findOne({ _id: id }));
});
app.post('/books/:id/tags', async (req, res) => {
const id = parseId(req, res);
if (!id) return;
const result = await books.updateOne({ _id: id }, { $push: { tags: req.body.tag } });
if (result.matchedCount === 0) {
return sendError(res, 404, 'No such book');
}
res.json(await books.findOne({ _id: id }));
});
app.delete('/books/:id', async (req, res) => {
const id = parseId(req, res);
if (!id) return;
const result = await books.deleteOne({ _id: id });
if (result.deletedCount === 0) {
return sendError(res, 404, 'No such book');
}
res.status(204).end();
});
app.put('/books/:id/ratings', async (req, res) => {
const id = parseId(req, res);
if (!id) return;
const { user, stars } = req.body;
await ratings.updateOne({ bookId: id, user }, { $set: { stars } }, { upsert: true });
res.json({ user, stars });
});
app.get('/books/:id/ratings', async (req, res) => {
const id = parseId(req, res);
if (!id) return;
res.json(await ratings.find({ bookId: id }).toArray());
});
app.get('/stats', async (req, res) => {
const filter = req.query.genre ? { genre: req.query.genre } : {};
res.json({
books: await books.countDocuments(filter),
ratings: await ratings.countDocuments(),
});
});
app.get('/stats/genres', async (req, res) => {
const genres = await books
.aggregate([{ $group: { _id: '$genre', count: { $sum: 1 } } }])
.toArray();
res.json(genres);
});
app.get('/stats/top', async (req, res) => {
const top = await ratings
.aggregate([
{ $group: { _id: '$bookId', avg: { $avg: '$stars' }, count: { $sum: 1 } } },
{ $match: { count: { $gte: 2 } } },
{ $sort: { avg: -1 } },
{ $limit: 3 },
])
.toArray();
res.json(top);
});
app.get('/books/:id/full', async (req, res) => {
const id = parseId(req, res);
if (!id) return;
const [book] = await books
.aggregate([
{ $match: { _id: id } },
])
.toArray();
if (!book) {
return sendError(res, 404, 'No such book');
}
res.json(book);
});
app.listen(3000, () => {
console.log('Bookshelf is running on port 3000');
});
使用 MongoDB 的数据库中的所有课
自己练习: Express 在线编辑器