MongoDB（38）如何使用聚合进行投影？

在MongoDB中，使用聚合管道进行投影操作的关键阶段是 $project。该阶段允许你选择要包含或排除的字段，创建新的字段，重命名字段，甚至进行计算和嵌套文档的操作。

使用 $project 阶段进行投影

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

{ 
  $project: { 
    <field1>: <expression1>, 
    <field2>: <expression2>, 
    // 其他字段和表达式
  } 
}

在 $project 阶段中，表达式可以是：

• 1：包含该字段
• 0：排除该字段
• <表达式>：对字段进行计算或变换

代码示例

以下是使用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');

        // 使用 $project 进行投影，选择特定字段
        console.log("\n$project stage (select fields):");
        let result = await collection.aggregate([
            { $project: { customerId: 1, amount: 1, _id: 0 } } // 选择 customerId 和 amount 字段，排除 _id 字段
        ]).toArray();
        console.log(result);

        // 使用 $project 进行投影，创建新字段
        console.log("\n$project stage (create new fields):");
        result = await collection.aggregate([
            { $project: { 
                customerId: 1, 
                amount: 1, 
                amountWithTax: { $multiply: ["$amount", 1.1] }, // 创建新字段 amountWithTax，计算含税金额
                _id: 0 
            }}
        ]).toArray();
        console.log(result);

        // 使用 $project 进行投影，重命名字段
        console.log("\n$project stage (rename fields):");
        result = await collection.aggregate([
            { $project: { 
                clientID: "$customerId", // 重命名 customerId 为 clientID
                totalAmount: "$amount",  // 重命名 amount 为 totalAmount
                _id: 0 
            }}
        ]).toArray();
        console.log(result);

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

aggregateData().catch(console.error);

在这个示例中，我们演示了如何使用 $project 阶段进行投影操作：

1. 选择字段 ：只选择 customerId 和 amount 字段，同时排除 _id 字段。
2. 创建新字段 ：创建一个新字段 amountWithTax，计算含税金额。
3. 重命名字段 ：将 customerId 重命名为 clientID，将 amount 重命名为 totalAmount。

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

// 选择字段结果
$project stage (select fields):
[
  { customerId: 1, amount: 100 },
  { customerId: 1, amount: 200 },
  { customerId: 2, amount: 150 },
  { customerId: 2, amount: 50 },
  { customerId: 3, amount: 250 }
]

// 创建新字段结果
$project stage (create new fields):
[
  { customerId: 1, amount: 100, amountWithTax: 110 },
  { customerId: 1, amount: 200, amountWithTax: 220 },
  { customerId: 2, amount: 150, amountWithTax: 165 },
  { customerId: 2, amount: 50, amountWithTax: 55 },
  { customerId: 3, amount: 250, amountWithTax: 275 }
]

// 重命名字段结果
$project stage (rename fields):
[
  { clientID: 1, totalAmount: 100 },
  { clientID: 1, totalAmount: 200 },
  { clientID: 2, totalAmount: 150 },
  { clientID: 2, totalAmount: 50 },
  { clientID: 3, totalAmount: 250 }
]

其他语言示例

类似的投影操作也可以在其他编程语言中实现，如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']

    # 使用 $project 进行投影，选择特定字段
    print("\n$project stage (select fields):")
    pipeline = [
        { '$project': { 'customerId': 1, 'amount': 1, '_id': 0 } } # 选择 customerId 和 amount 字段，排除 _id 字段
    ]
    result = list(collection.aggregate(pipeline))
    for doc in result:
        print(doc)

    # 使用 $project 进行投影，创建新字段
    print("\n$project stage (create new fields):")
    pipeline = [
        { '$project': { 
            'customerId': 1, 
            'amount': 1, 
            'amountWithTax': { '$multiply': ['$amount', 1.1] }, # 创建新字段 amountWithTax，计算含税金额
            '_id': 0 
        }}
    ]
    result = list(collection.aggregate(pipeline))
    for doc in result:
        print(doc)

    # 使用 $project 进行投影，重命名字段
    print("\n$project stage (rename fields):")
    pipeline = [
        { '$project': { 
            'clientID': '$customerId', // 重命名 customerId 为 clientID
            'totalAmount': '$amount',  // 重命名 amount 为 totalAmount
            '_id': 0 
        }}
    ]
    result = list(collection.aggregate(pipeline))
    for doc in result:
        print(doc)

if __name__ == '__main__':
    main()

运行这个脚本后，你会得到类似的结果。通过这些示例，你可以了解到如何在不同编程语言中使用MongoDB的聚合管道进行投影操作，并选择、创建和重命名字段。