Nodejs中mongodb的arggregate如何做sum，求mongodb高手

要在Node.js中使用MongoDB的聚合框架来计算特定时间段内的amount总和，可以利用聚合管道（aggregation pipeline）中的多个阶段来实现。以下是一个示例代码，展示如何根据不同的时间范围（如今天、本周、本月、本年）来计算amount的总和。

首先确保你已经安装了mongoose或者直接使用mongodb驱动包来连接MongoDB数据库。

示例代码

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

async function calculateAmounts() {
    const uri = "your_mongodb_connection_string";
    const client = new MongoClient(uri, { useNewUrlParser: true, useUnifiedTopology: true });

    try {
        await client.connect();
        const database = client.db('your_database_name');
        const collection = database.collection('your_collection_name');

        // 获取当前时间和日期
        const now = new Date();
        const yearStart = new Date(now.getFullYear(), 0, 1);
        const monthStart = new Date(now.getFullYear(), now.getMonth(), 1);
        const weekStart = new Date(now.getFullYear(), now.getMonth(), now.getDate() - now.getDay());
        const todayStart = new Date(now.getFullYear(), now.getMonth(), now.getDate());

        const result = await collection.aggregate([
            {
                $match: {
                    _id: ObjectId("your_object_id"),
                    items: {
                        $elemMatch: {
                            date: {
                                $gte: todayStart,
                                $lt: new Date(todayStart.getTime() + 24 * 60 * 60 * 1000)
                            }
                        }
                    }
                }
            },
            {
                $unwind: "$items"
            },
            {
                $match: {
                    "items.date": {
                        $gte: todayStart,
                        $lt: new Date(todayStart.getTime() + 24 * 60 * 60 * 1000)
                    }
                }
            },
            {
                $group: {
                    _id: null,
                    todayTotal: { $sum: "$items.amount" }
                }
            },
            {
                $facet: {
                    today: [
                        {
                            $match: {
                                "items.date": {
                                    $gte: todayStart,
                                    $lt: new Date(todayStart.getTime() + 24 * 60 * 60 * 1000)
                                }
                            }
                        },
                        {
                            $group: {
                                _id: null,
                                todayTotal: { $sum: "$items.amount" }
                            }
                        }
                    ],
                    thisWeek: [
                        {
                            $match: {
                                "items.date": {
                                    $gte: weekStart,
                                    $lt: new Date(weekStart.getTime() + 7 * 24 * 60 * 60 * 1000)
                                }
                            }
                        },
                        {
                            $group: {
                                _id: null,
                                thisWeekTotal: { $sum: "$items.amount" }
                            }
                        }
                    ],
                    thisMonth: [
                        {
                            $match: {
                                "items.date": {
                                    $gte: monthStart,
                                    $lt: new Date(monthStart.getTime() + 31 * 24 * 60 * 60 * 1000)
                                }
                            }
                        },
                        {
                            $group: {
                                _id: null,
                                thisMonthTotal: { $sum: "$items.amount" }
                            }
                        }
                    ],
                    thisYear: [
                        {
                            $match: {
                                "items.date": {
                                    $gte: yearStart,
                                    $lt: new Date(yearStart.getTime() + 365 * 24 * 60 * 60 * 1000)
                                }
                            }
                        },
                        {
                            $group: {
                                _id: null,
                                thisYearTotal: { $sum: "$items.amount" }
                            }
                        }
                    ]
                }
            },
            {
                $project: {
                    todayTotal: { $arrayElemAt: ["$today.todayTotal", 0] },
                    thisWeekTotal: { $arrayElemAt: ["$thisWeek.thisWeekTotal", 0] },
                    thisMonthTotal: { $arrayElemAt: ["$thisMonth.thisMonthTotal", 0] },
                    thisYearTotal: { $arrayElemAt: ["$thisYear.thisYearTotal", 0] }
                }
            }
        ]).toArray();

        console.log(result[0]);
    } finally {
        await client.close();
    }
}

calculateAmounts().catch(console.error);

解释

1. 连接到MongoDB：首先连接到MongoDB数据库。
2. 定义时间范围：定义今天的开始时间、本周的开始时间、本月的开始时间和本年的开始时间。
3. 聚合管道：使用多个聚合阶段来过滤和分组数据，计算每个时间段的amount总和。
4. 输出结果：将结果打印出来。

通过这种方式，你可以一次性计算出不同时间段的总和。