(MongoDB)在同一管道中结合联合和交集

发布于10月04日

我在最新版本的MongoDB和pymongo中运行了以下聚合管道:

[
  {
    "$project": {
      "union": {
        "$setUnion": [
          "$query_a",
          "$query_b"
        ]
      }
    }
  },
  {
    "$unwind": "$union"
  },
  {
    "$group": {
      "_id": "$union.ID",
      "date_a": {
        "$addToSet": "$union.date_a"
      },
      "date_b": {
        "$addToSet": "$union.date_b"
      }
    }
  },
  {
    "$unwind": "$date_a"
  },
  {
    "$unwind": "$date_b"
  },
  {
    "$project": {
      "_id": 1,
      "date_a": "$date_a",
      "date_b": "date_b",
      "diff": {
        "$subtract": [
          {
            "$toInt": "$date_b"
          },
          {
            "$toInt": "$date_a"
          }
        ]
      }
    }
  },
  {
    "$match": {
      "diff": {
        "$gt": 0,
        "$lte": 20
      }
    }
  },
  
]

这给出了两条管线Query_a和Query_b的并集.在此并集之后，我希望获得ID上的与管线Query_c的交集:(Query_A Union Query_B)交集Query_c.

对于这playground example，所需的输出将是:

    [
      {
        "ID": "c80ea2cb-3272-77ae-8f46-d95de600c5bf",
        
      },
      {
        "ID": "cdbcc129-548a-9d51-895a-1538200664e6",
      }
    ]

推荐答案

您可以稍微更改和增强您的管道，以获得所需的输出.

db.collection.aggregate([
  {
    "$project": {
      "union": {
        // do the intersection here
        "$filter": {
          "input": {
            "$setUnion": [
              "$query_a",
              "$query_b"
            ]
          },
          "as": "elem",
          "cond": {
            // only take IDs in query_c
            "$in": ["$$elem.ID", "$query_c.ID"]
          }
        }
      }
    }
  },
  {
    "$unwind": "$union"
  },
  {
    "$group": {
      "_id": "$union.ID",
      "date_a": {
        "$addToSet": "$union.date_a"
      },
      "date_b": {
        "$addToSet": "$union.date_b"
      }
    }
  },
  {
    "$unwind": "$date_a"
  },
  {
    "$unwind": "$date_b"
  },
  {
    "$project": {
      "diff": {
        "$subtract": [
          {
            "$toInt": "$date_b"
          },
          {
            "$toInt": "$date_a"
          }
        ]
      }
    }
  },
  {
    "$match": {
      "diff": {
        "$gt": 0,
        "$lte": 20
      }
    }
  },
  { // get unique _id's
    "$group": {
      "_id": "$_id"
    }
  },
  { // rename _id to ID
    "$project": {
      "_id": 0,
      "ID": "$_id"
    }
  }
])

试穿一下mongoplayground.net号.