Menu
CoddyTech

使用 $group 分组

本课属于 Coddy Node.js 学习旅程的“使用 MongoDB 的数据库”单元。第 21 课,共 29 课。

统计一个类型只需要一次查询。要一次统计所有类型,数据库需要把书籍分类。

aggregate 会运行一个管道:它是由多个阶段组成的数组,每个阶段都会把输出传给下一个阶段。$group 阶段会为其 _id 的每个不同值生成一个输出文档。这里的 _id 不是文档 ID。它表示分组依据,每个输出文档都会在 _id 中保存该值。字段名前的 $,例如 '$team',表示“该文档中这个字段的值”。如果没有它,'team' 对每个文档来说都是相同的文本,因此它们都会进入同一组。$sum: 1 会为每个文档加 1。

await players
    .aggregate([{ $group: { _id: '$team', players: { $sum: 1 } } }])
    .toArray();
// [{ _id: 'red', players: 4 }, { _id: 'blue', players: 3 }]

分组结果只包含该 _id 和累加器计算出的字段;输入文档中的其他字段不会保留。

challenge icon

挑战

简单

这一步会打开课程中预先准备好的书架 API 项目,而不是你在 Counting Documents 中的项目文件(包括 server.js)。你之前的作品仍保留在它自己的课中。编辑 server.js 前,先阅读提供的文件。这份副本包含书架的 CRUD 路由、唯一 ISBN 索引、GET /books 筛选条件和 GET /stats 计数。

一个按类型分类的页面会显示每种类型有多少本书,所有数据都来自一次查询,其中也包括没人预料到的类型。

  1. 在 server.js 中添加一个 GET /stats/genres 路由。
  2. 对 books 执行聚合,其中包含一个 $group 阶段。它会为每种不同的 genre 生成一条结果,并统计具有该类型的书籍数量。
  3. 返回状态码 200 和聚合操作返回的数组。每个条目都在 _id 中保存类型,在 count 中保存书籍数量。在 $group 中,_id 不是 ObjectId:它是书籍分组依据的值。条目可以按任意顺序排列。使用初始提供的书籍数据时,响应如下:
    [ { "_id": "scifi", "count": 2 }, { "_id": "classic", "count": 1 } ]
    试试看:发送 GET /stats/genres,你会得到上面的两个条目。使用 { "title": "Leaves of Grass", "author": "Walt Whitman", "year": 1855, "isbn": "9780140421996", "genre": "poetry" } 发送 POST /books,你会得到状态码 201。再次发送 GET /stats/genres,会出现第三个条目:{ "_id": "poetry", "count": 1 }。

亲手试试

import express from 'express';
import { MongoClient, ObjectId } from 'mongodb';

const client = new MongoClient('mongodb://localhost:27017');
const db = client.db('shelf');
const books = db.collection('books');
const ratings = db.collection('ratings');

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' },
]);

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.listen(3000, () => {
    console.log('Bookshelf is running on port 3000');
});

使用 MongoDB 的数据库中的所有课

自己练习: Express 在线编辑器