Mongodb 过滤数组匹配到的元素

阿阿阿阿阿阿杰2023-11-27 19:20

好久没写后端代码了，聚合查询已然生疏，遇到个问题竟然翻了好久文档才解决，以前遇到过，就是没有记录下来，这次留个底 🙃

场景

有类似于这样一个场景：

假设有一张表：tasks，有以下数据：

javascript 复制代码

[
  { 
  	_id: 1,
  	participants: [
    	{ name: '小张', age: 30 },
    	{ name: '小李', age: 32 }, 
    	{ name: '小张', age: 20 }
    ]
  },
  ...
]

预期结果是 :

javascript 复制代码

[
  { 
    _id: 1, 
    participants: [
      { name: '小张', age: 30 },
      { name: '小张', age: 20 }
    ]
  }
]

有同学可能会想到， db.tasks.find({_id: 1, "participants.name": '小张'}); 这种是不是就解决了呢？有这样的猜测没有什么毛病，不过老道的玩家是知道这样不行的，因为这个只能匹配到数组中符合条件的这一整条数据，但是不能过滤数组中仅满足条件的元素，结果是这样的：

yaml 复制代码

  [{ 
  	_id: 1,
  	participants: [
    	{ name: '小张', age: 30 },
    	{ name: '小李', age: 32 }, 
    	{ name: '小张', age: 20 }
    ]
  }]

解决办法

使用聚合查询的 $project 加数组的 $filter 操作符过滤，示例如下：

javascript 复制代码

db.tasks.aggregate([
  { $match: { _id: 1 } },
  { 
    $project: {
      	_id: 1,
  		participants: {
        	$filter: {
            	input: "$participants",
              	as: "result",
              	cond: {
                    $eq: ["$$result.name", '小张']
                }
            }
        }
  	} 
 }
]);

对你有用的话，来波素质三连哟 🌹