Fill missing dates in records

后端 未结 4 1569
佛祖请我去吃肉
佛祖请我去吃肉 2020-11-30 07:03

I have a collection of ProductViews.

ProductView

{
    productId: \'5b8c0f3204a10228b00a1745,
    createdAt: \'2018-09-         


        
4条回答
  •  隐瞒了意图╮
    2020-11-30 07:48

    You need few additional stages to return default values. First of all you need to use $group with _id set to null to collect all results in one document. Then you can use $map with an array of days as an input. Inside that $map you can use $indexOfArray to find if that date exists in your current result set. If yes (index != -1) then you can return that value, otherwise you need to return default subdocument with views set to 0. Then you can use $unwind to get back a list of documents and $replaceRoot to promote nested stats to a top level.

    ProductView.aggregate([
        { $match: { productId: '5b8c0f3204a10228b00a1745' } },
        { $project: { day: { $substr: ["$createdAt", 0, 10] } } },
        {
            $group: {
                _id: "$day",
                count: { $sum: 1 },
                time: { $avg: "$createdAt" },
            }
        },
        { $sort: { _id: 1 } },
        {
            $project: {
                date: '$_id',
                views: '$count',
            },
        },
        {
            $group: {
                _id: null,
                stats: { $push: "$$ROOT" }
            }
        },
        {
            $project: {
                stats: {
                    $map: {
                        input: [ "2018-09-01", "2018-09-02", "2018-09-03", "2018-09-04", "2018-09-05" ],
                        as: "date",
                        in: {
                            $let: {
                                vars: { dateIndex: { "$indexOfArray": [ "$stats._id", "$$date" ] } },
                                in: { 
                                    $cond: {
                                        if: { $ne: [ "$$dateIndex", -1 ] },
                                        then: { $arrayElemAt: [ "$stats", "$$dateIndex" ] },
                                        else: { _id: "$$date", date: "$$date", views: 0 }
                                    } 
                                }
                            }
                        }
                    }
                }
            }
        },
        {
            $unwind: "$stats"
        },
        {
            $replaceRoot: {
                newRoot: "$stats"
            }
        }
    ]).exec((err, result) => ...)
    

    You can generate a static list of dates in your application logic using simple loop. I believe that's possible in MongoDB as well (using $range) but it might complicate this aggregation pipeline. Let me know if you're fine with that or you want to try to generate that array of dates in MongoDB.

提交回复
热议问题