Mongodb 在MondoDB中：将对象数组从切片索引数组切片，并通过聚合推入数组

发布于01月19日

我想根据同一文档中可用的slicingIndex键对文档中的数组进行切片，并推送到数组中.

[
  {
    slicingIndex: [0, 4, 7, 10],
    array: [
      { name: 1 },
      { name: 2 },
      { name: 3 },
      { name: 4 },
      { name: 5 },
      { name: 6 },
      { name: 7 },
      { name: 8 },
      { name: 9 },
      { name: 10 }
    ]
  }
]

帖子主题:Re:Колибри

{
  "array": [
    [
      { "name": 1 },
      { "name": 2 },
      { "name": 3 },
      { "name": 4 }
    ],
    [
      { "name": 5 },
      { "name": 6 },
      { "name": 7 }
    ],
    [
      { "name": 8 },
      { "name": 9 },
      { "name": 10 }
    ]
  ]
}

这是我在下面try 的聚合，但没有达到预期的效果:

db.collection.aggregate([
  {
    "$project": {
      slicingIndex: 1,
      array: {
        "$map": {
          "input": "$slicingIndex",
          "as": "num",
          "in": {
            "$slice": [
              "$array",
              "$$num",
              {
                "$subtract": [
                  {
                    "$arrayElemAt": ["$slicingIndex", { "$indexOfArray": ["$slicingIndex", "$$num"] }]
                  },
                  "$$num"
                ]
              }
            ]
          }
        }
      }
    }
  }
])

推荐答案

MongoDb Playground Link:https://mongoplayground.net/p/oNzBVLRfIU0

$add & $indexOfArray used to take one step ahead element from current pointer in array like (n+1).
$let : I have used it as optional to make variable reusable if needed.
$map : To traverse array elements.
$arrayElemAt : To get from element from array.
$subtract : To perform operation like [ (n+1) - n ] where n is index in array.
$filter : To filter out elements (in our case: null).

db.collection.aggregate([
  {
    "$project": {
      array: {
        "$map": {
          "input": "$slicingIndex",
          "as": "elem",
          "in": {
            "$let": {
              "vars": {
                //create variable that holds element one index+1 to the current index
                "elemOneStepHeadToElem": {
                  "$arrayElemAt": [
                    "$slicingIndex",
                    {
                      $add: [
                        1,
                        {
                          "$indexOfArray": [
                            "$slicingIndex",
                            "$$elem"
                          ]
                        }
                      ]
                    }
                  ]
                }
              },
              "in": {
                "$slice": [
                  "$array",
                  "$$elem",
                  {
                    "$subtract": [
                      "$$elemOneStepHeadToElem",
                      "$$elem"
                    ]
                  }
                ]
              }
            }
          }
        }
      }
    }
  }//remove null from array from last index
  ,
  {
    "$project": {
      array: {
        "$filter": {
          "input": "$array",
          "cond": {
            $ne: [
              "$$this",
              null
            ]
          }
        }
      }
    }
  }
])