MongoDB（37）如何使用聚合进行排序？

在MongoDB中，使用聚合管道进行排序操作的关键阶段是 $sort。该阶段允许你根据一个或多个字段对文档进行升序或降序排序。

使用 $sort 阶段进行排序

$sort 阶段的基本结构如下：

{ 
  $sort: { 
    <field1>: <order1>, 
    <field2>: <order2>, 
    // 其他排序字段
  } 
}

其中，排序顺序可以是：

• 1：升序
• -1：降序

代码示例

以下是使用Node.js进行排序操作的详细示例。首先确保安装了MongoDB的Node.js驱动：

npm install mongodb

插入示例数据

const { MongoClient } = require('mongodb');

async function insertData() {
    const uri = "mongodb://localhost:27017";
    const client = new MongoClient(uri, { useUnifiedTopology: true });

    try {
        await client.connect();
        const db = client.db('myDatabase');
        const collection = db.collection('orders');

        await collection.deleteMany({}); // 清空集合

        await collection.insertMany([
            { customerId: 1, amount: 100, status: "shipped" },
            { customerId: 1, amount: 200, status: "pending" },
            { customerId: 2, amount: 150, status: "shipped" },
            { customerId: 2, amount: 50, status: "pending" },
            { customerId: 3, amount: 250, status: "shipped" }
        ]);

        console.log("Data inserted");
    } finally {
        await client.close();
    }
}

insertData().catch(console.error);

使用聚合管道进行排序

async function aggregateData() {
    const uri = "mongodb://localhost:27017";
    const client = new MongoClient(uri, { useUnifiedTopology: true });

    try {
        await client.connect();
        const db = client.db('myDatabase');
        const collection = db.collection('orders');

        // 使用 $sort 进行排序
        console.log("\n$sort stage:");
        let result = await collection.aggregate([
            { $sort: { amount: -1 } } // 按照 amount 字段进行降序排序
        ]).toArray();
        console.log(result);

        // 使用 $sort 进行多字段排序
        console.log("\n$sort stage with multiple fields:");
        result = await collection.aggregate([
            { $sort: { status: 1, amount: -1 } } // 先按 status 升序，再按 amount 降序排序
        ]).toArray();
        console.log(result);

    } finally {
        await client.close();
    }
}

aggregateData().catch(console.error);

在这个示例中，我们演示了如何使用 $sort 阶段进行排序操作：

1. 单字段排序 ：按 amount 字段降序排序。
2. 多字段排序 ：先按 status 字段升序排序，再按 amount 字段降序排序。

运行这个脚本后，你会得到如下结果（示例输出）：

// 单字段排序结果
$sort stage:
[
  { customerId: 3, amount: 250, status: 'shipped' },
  { customerId: 1, amount: 200, status: 'pending' },
  { customerId: 2, amount: 150, status: 'shipped' },
  { customerId: 1, amount: 100, status: 'shipped' },
  { customerId: 2, amount: 50, status: 'pending' }
]

// 多字段排序结果
$sort stage with multiple fields:
[
  { customerId: 1, amount: 200, status: 'pending' },
  { customerId: 2, amount: 50, status: 'pending' },
  { customerId: 3, amount: 250, status: 'shipped' },
  { customerId: 2, amount: 150, status: 'shipped' },
  { customerId: 1, amount: 100, status: 'shipped' }
]

其他语言示例

类似的排序操作也可以在其他编程语言中实现，如Python。以下是Python的示例代码：

安装PyMongo

在终端中运行以下命令来安装PyMongo：

pip install pymongo

使用Python进行排序

from pymongo import MongoClient

def main():
    client = MongoClient('mongodb://localhost:27017/')
    db = client['myDatabase']
    collection = db['orders']

    # 使用 $sort 进行单字段排序
    print("\n$sort stage:")
    pipeline = [
        { '$sort': { 'amount': -1 } } # 按照 amount 字段进行降序排序
    ]
    result = list(collection.aggregate(pipeline))
    for doc in result:
        print(doc)

    # 使用 $sort 进行多字段排序
    print("\n$sort stage with multiple fields:")
    pipeline = [
        { '$sort': { 'status': 1, 'amount': -1 } } # 先按 status 升序，再按 amount 降序排序
    ]
    result = list(collection.aggregate(pipeline))
    for doc in result:
        print(doc)

if __name__ == '__main__':
    main()

运行这个脚本后，你会得到类似的结果。通过这些示例，你可以了解到如何在不同编程语言中使用MongoDB的聚合管道进行排序操作，并在一个或多个字段上对数据进行升序或降序排序。