MongoDB將結束時間與下一個檔案的開始時間進行比較

我有一個類似的集合，我按照它們的開始時間對它們進行了排序：

{"name": 'A', "startTime": '1634626355', "endTime": '1634631405'}
{"name": 'A', "startTime": '1634631406', "endTime": '1634631864'}
{"name": 'A', "startTime": '1634631865', "endTime": '1634656048'}
{"name": 'A', "startTime": '1634712642', "endTime": '1634718856'}

如何比較檔案，如果檔案 endTime 和下一個檔案 startTime 持續時間小于 5 分鐘，請將其合并。

這是我試圖實作的結果（第 1 個 3 個檔案合并為 1 個，其中使用第一個檔案的 startTime 和第三個檔案的 endTime）：

{"name": 'A', "startTime": '1634626355', "endTime": '1634656048'}
{"name": 'A', "startTime": '1634712642', "endTime": '1634718856'}

謝謝

uj5u.com熱心網友回復：

首先，您永遠不應該將日期/時間值存盤為字串，這是一個設計缺陷。始終存盤適當的Date物件。

此解決方案無需自查找即可作業，因此性能可能更好：

db.collection.aggregate([
   {
      $set: {
         startDateTime: { $toDate: { $multiply: ["$startTime", 1000] } },
         endDateTime: { $toDate: { $multiply: ["$endTime", 1000] } }
      },
   },
   { $sort: { startDateTime: 1 } },
   { $group: { _id: null, data: { $push: "$$ROOT" } } },
   {
      $set: {
         data: {
            $reduce: {
               input: "$data",
               initialValue: [],
               in: {
                  $cond: {
                     if: {
                        $or: [
                           { $eq: [{ $size: "$$value" }, 0] }, // for the initail element
                           {
                              $gt: [
                                 {
                                    $dateDiff: { // calculate difference
                                       endDate: "$$this.startDateTime",
                                       startDate: { $last: "$$value.endDateTime" },
                                       unit: "minute"
                                    }
                                 },
                                 5 // more than 5 Minutes
                              ]
                           }
                        ]
                     },
                     then: { $concatArrays: ["$$value", ["$$this"]] }, // append new element
                     else: {
                        $map: {
                           input: "$$value",
                           as: "data",
                           in: {
                              $cond: {
                                 if: { $eq: ["$$data._id", { $last: "$$value._id" }] }, // find last element
                                 then: { // update last element
                                    $mergeObjects: [
                                       "$$data",
                                       { endDateTime: "$$this.endDateTime" },
                                       { endTime: "$$this.endTime" }
                                    ]
                                 }, 
                                 else: "$$data"
                              }
                           }
                        }
                     }
                  }
               }
            }
         }
      }
   },
   // some cosmetic
   { $unwind: "$data" },
   { $replaceRoot: { newRoot: "$data" } }
])

蒙戈游樂場

uj5u.com熱心網友回復：

您可以$lookup在聚合管道中使用以找出需要洗掉的檔案。然后，執行 forEach 以洗掉它們。

db.collection.aggregate([
  {
    $addFields: {
      endDateTime: {
        "$toDate": {
          "$multiply": [
            {
              $toLong: "$endTime"
            },
            1000
          ]
        }
      }
    },
    
  },
  {
    "$lookup": {
      "from": "collection",
      let: {
        end: "$endDateTime"
      },
      pipeline: [
        {
          "$addFields": {
            startDateTime: {
              "$toDate": {
                "$multiply": [
                  {
                    $toLong: "$startTime"
                  },
                  1000
                ]
              }
            }
          }
        },
        {
          $match: {
            $expr: {
              $and: [
                {
                  $lte: [
                    {
                      $subtract: [
                        "$startDateTime",
                        "$$end"
                      ]
                    },
                    300000
                  ]
                },
                {
                  $lte: [
                    "$$end",
                    "$startDateTime"
                  ]
                }
              ]
            }
          }
        }
      ],
      "as": "lessThan5min"
    }
  },
  {
    "$unwind": "$lessThan5min"
  },
  {
    "$replaceRoot": {
      "newRoot": "$lessThan5min"
    }
  }
]).forEach(function(doc){
  db.collection.remove({ "_id": doc._id });
});

這里是Mongo playground，可以找出需要洗掉的檔案以供參考。

標籤：MongoDB 聚合框架

上一篇：更新mongodb中的Key

下一篇：pymongo-bson.errors.InvalidDocument僅有時無明顯原因引發