Получить индекс массива, когда фильтр в mongodb - PullRequest
Новая
       5

Получить индекс массива, когда фильтр в mongodb

0 голосов
/ 12 декабря 2018

Я собираюсь сделать запрос для получения данных с индексом подмассива, чье число является четным числом.

Входное значение: 

{
    "_id" : ObjectId("5b3b206c4a25da19d05f41a2"),
    "numbers" : [ 
        {
            "name" : "Foobar",
            "lines" : [ 
                {
                    "number" : 6,
                }, 
                {
                    "number" : 11,

                },
                {
                    "number" : 15,

                },
                {
                    "number" : 8,
                }

            ]
        }
    ]
}

Желаемый вывод: 

{
    "_id" : ObjectId("5b3b206c4a25da19d05f41a2"),
    "numbers" : [ 
        {
            "name" : "Foobar",
            "lines" : [ 
                {
                    "index"  : 0,
                    "number" : 6,
                },
                {
                    "index"  : 3,
                    "number" : 8,
                }
            ]
        }
    ]
}

Самым важным является получение индекса предметов "строк".

1 Ответ

0 голосов
/ 12 декабря 2018

Вам нужно использовать $map для перебора первого массива, а затем $filter с вложенным массивом.

db.collection.aggregate([
  { "$project": {
    "numbers": {
      "$map": {
        "input": "$numbers",
        "as": "nn",
        "in": {
          "name": "$$nn.name",
          "lines": {
            "$filter": {
              "input": {
                "$map": {
                  "input": "$$nn.lines",
                  "as": "ll",
                  "in": {
                    "number": "$$ll.number",
                    "index": { "$indexOfArray": ["$$nn.lines.number", "$$ll.number"] }
                  }
                }
              },
              "as": "ll",
              "cond": { "$eq": [{ "$mod": ["$$ll.number", 2] }, 0] }
            }
          }
        }
      }
    }
  }}
])

Даже если вам нужен еще один трюк агрегации 

db.collection.aggregate([
  { "$project": {
    "numbers": {
      "$map": {
        "input": {
          "$map": {
            "input": "$numbers",
            "as": "nn",
            "in": {
              "name": "$$nn.name",
              "lines": {
                "$filter": {
                  "input": { "$zip": { "inputs": ["$$nn.lines", { "$range": [0, { "$size": "$$nn.lines" }] }] }},
                  "as": "ll",
                  "cond": {
                    "$let": {
                      "vars": { "num": { "$arrayElemAt": ["$$ll", 0] }},
                      "in": { "$eq": [{ "$mod": ["$$num.number", 2] }, 0] }
                    }
                  }
                }
              }
            }
          }
        },
        "as": "nn",
        "in": {
          "name": "$$nn.name",
          "lines": {
            "$map": {
              "input": "$$nn.lines",
              "as": "kk",
              "in": {
                "number": { "$arrayElemAt": ["$$kk.number", 0] },
                "index": { "$arrayElemAt": ["$$kk", 1] }
              }
            }
          }
        }
      }
    }
  }}
])

Наконец, оба вы получите похожий Выход 

[
  {
    "_id": ObjectId("5b3b206c4a25da19d05f41a2"),
    "numbers": [
      {
        "name": "Foobar",
        "lines": [
          {
            "index": 0,
            "number": 6
          },
          {
            "index": 3,
            "number": 8
          }
        ]
      }
    ]
  }
]
...