根据Mongodb中两个日期之差对记录进行计数

时间:2019-02-07 12:17:18

标签: mongodb pymongo djongo

根据下面的文档,我需要计算记录inicial_date和end_date之间的差异大于和小于5分钟且名称= A。

[{
    "_id": 1,
    "name": "A",
    "inicial_date": "2019-01-01 12:00:00",
    "end_date": "2019-01-01 12:01:00"
},{
    "_id": 2,
    "name": "A",
    "inicial_date": "2019-01-01 12:05:00",
    "end_date": "2019-01-01 12:07:00"
},{
    "_id": 3,
    "name": "A",
    "inicial_date": "2019-01-01 12:00:00",
    "end_date": "2019-01-01 12:06:00"
},{
    "_id": 4,
    "name": "A",
    "inicial_date": "2019-01-01 12:04:00",
    "end_date": "2019-01-01 12:05:00"
},
    "_id": 5,
    "name": "A",
    "inicial_date": "2019-01-01 12:10:00",
    "end_date": "2019-01-01 12:20:00"
},{
    "_id": 6,
    "name": "A",
    "inicial_date": "2019-01-01 12:00:00",
    "end_date": "2019-01-01 12:08:00"
},{
    "_id": 7,
    "name": "A",
    "inicial_date": "2019-01-01 13:00:00",
    "end_date": "2019-01-01 13:01:00"
},{
    "_id": 8,
    "name": "B",
    "inicial_date": "2019-01-01 14:00:00",
    "end_date": "2019-01-01 14:09:00"
}]

预期结果:

{
    "less_than_5": 4,
    "greater_than_5": 3
}

1 个答案:

答案 0 :(得分:0)

好吧,您可以使用汇总查询

    db.collection.aggregate([
    {
        $match: {
            name: "A"
        }
    },
    {
        // find time difference, the result is in milliseconds
        $project: {
            timeDiffInMilliseconds: {
                $subtract: [
                    {
                        $toDate: "$end_date"
                    },
                    {
                        $toDate: "$inicial_date"
                    }
                ]
            }
        }
    },
    {
        // convert the time difference to minutes
        $project: {
            timeDiffInMinutes: {
                $divide: [
                    "$timeDiffInMilliseconds",
                    60000
                ]
            }
        }
    },
    {
        // check if the number of minutes is greater than 5 mins or not
        $project: {
            timeDiffGreaterThan5Mins: {
                $cond: [
                    {
                        $gt: [
                            "$timeDiffInMinutes",
                            5
                        ]
                    },
                    1,
                    0
                ]
            }
        }
    },
    {
        // group according to greater than 5 minutes or not
        $group: {
            _id: "null",
            greater_than_5: {
                $sum: {
                    $cond: [
                        {
                            $eq: [
                                "$timeDiffGreaterThan5Mins",
                                1
                            ]
                        },
                        1,
                        0
                    ]
                }
            },
            less_than_5: {
                $sum: {
                    $cond: [
                        {
                            $eq: [
                                "$timeDiffGreaterThan5Mins",
                                0
                            ]
                        },
                        1,
                        0
                    ]
                }
            }
        }
    }
])

提高效率

db.collection.aggregate([
  {
    $match: {
      name: "A"
    }
  },
  {
    $project: {
      more_than_5_mins: {
        $cond: [
          {
            $gt: [
              {
                $subtract: [
                  {
                    $toDate: "$end_date"
                  },
                  {
                    $toDate: "$inicial_date"
                  }
                ]
              },
              300000
            ]
          },
          1,
          0
        ]
      }
    }
  },
  {
    $group: {
      _id: "",
      less_than_5: {
        $sum: {
          $cond: [
            {
              $eq: [
                "$more_than_5_mins",
                0
              ]
            },
            1,
            0
          ]
        }
      },
      greater_than_5: {
        $sum: {
          $cond: [
            {
              $eq: [
                "$more_than_5_mins",
                1
              ]
            },
            1,
            0
          ]
        }
      }
    }
  }
])