Quantcast
Channel: CNode:Node.js专业中文社区
Viewing all articles
Browse latest Browse all 14821

使用mongodb的aggregate解决变态需求

$
0
0

背景现有1000条学生记录,结构如下:

{
	name:String,//名称
	clazz:{type:ObjectId,ref:"classes"},//班级id
	status:Number,//状态 1在校 2离校
	marks:[Number],//标记 1迟到 2早退 3请假
	age:Number//年龄
}

需求要求统计1000个学生的所在班级列表,列表内容如下(模拟数据,不要在意统计的合理不合理): 班级名称 | 在校人数 | 离校人数 | 没有迟到标记的在校人数 | 没有迟到标记的在校学生的平均年龄

实现代码:

db.getCollection('students').aggregate([{
    $group: {//班级分组拿到班级id和班级内的学员信息
        _id: "$clazz",
        stus: {
            $push: "$$ROOT"
        }
    }
}, {
    $project: {
        _id: 1,
        stus: 1,
        out_num: {//计算离校人员数量
            $size: {
                $filter: {
                    input: "$stus",
                    as: "stu",
                    cond: { $eq: ["$$stu.status", 2] }
                }
            }
        },
        in_num: {//计算在校人员数量
            $size: {
                $filter: {
                    input: "$stus",
                    as: "stu",
                    cond: { $eq: ["$$stu.status", 1] }
                }
            }
        },
        in_no_1: {//计算 在校并且没有迟到标记的学员列表
            $filter: {
                input: "$stus",
                as: "stu",
                cond: { $and: [{ $eq: ["$$stu.status", 1] }, { $eq: [{ $indexOfArray: ["$$stu.marks", 1] }, -1] }] }
            }
        },
    }
}, {
    $project: {
        _id: 1,
        out_num: 1,
        in_num: 1,
        in_no_1: 1,
        in_nu_1_num: { $size: "$in_no_1" },//计算在校并且没有迟到标记的学员数量
        avg: {//计算在校并且没有迟到标记的学员的平均年龄
            $cond: {
                if: { $eq: [{ $size: "$in_no_1" }, 0] },//如果班级内没有符合条件的学生则年龄记为0
                then: 0,
                else: {//班级内有符合条件的学生,计算学生年龄总和并除以学生数量
                    $divide: [{
                        $reduce: {
                            input: "$in_no_1",
                            initialValue: 0,
                            in : { $add: ['$$value', "$$this.age"] }
                        }
                    }, {
                        $size: "$in_no_1"
                    }]
                }
            }
        }
    }
}, {
    $lookup: {//关联查询班级信息,关联信息返回为数组
        from: "classes",
        localField: "_id",
        foreignField: "_id",
        as: "clazz"
    }
}, {
    $unwind: '$clazz'//拆分班级返回信息数组

}, {
    $project: {//整理最后的数据
        _id: 1,
        out_num: 1,
        in_num: 1,
        avg: 1,
        in_nu_1_num: 1,
        name: '$clazz.name'
    }
}])


代码就这些,需要看的是aggregate的各种操作命令(尤其是数组的)。好了笔记就记录这些了。有更好的解决方案,大家一起讨论。


Viewing all articles
Browse latest Browse all 14821

Trending Articles