mongodb：使用字段名称的动态参数对嵌套数组进行排序

如何解决mongodb：使用字段名称的动态参数对嵌套数组进行排序

假设在聚合管道中，其中一个步骤将产生以下结果：

{
    customer: "WN",sort_category: "category_a",locations: [ 
        {
            city: "Elkana",category_a: 11904.0,category_b: 74.0,category_c: 657.0,},{
            city: "Haifa",category_a: 20.0,category_b: 841.0,category_c: 0,{
            city" : "Jerusalem",category_a: 451.0,category_b: 45.0,category_c: 712.0,}
    ]
}
{
...
}

下一步是对集合中每个文档的嵌套对象列表进行排序。

嵌套对象的列表应按包含字段名称的动态参数排序。

例如-locations的列表应按category_a的值排序。

category_a是sort_category字段中给出的参数。

解决方法

这是不使用$function的解决方案：

db.tests.aggregate([
    {$unwind:"$locations"},{$project: {
        _id: 1,customer: 1,sort_category: 1,locations: 1,locationsKV:{$objectToArray:"$locations"}
        }
    },{$unwind:"$locationsKV"},{$project:{
        _id: 1,locationsKV: 1,category: {
            $cond:[{$eq: ["$sort_category","$locationsKV.k"]},"$locationsKV.v",0]},agg: {
            $cond: [{$eq: ["$sort_category",true,false]
            }
        }
    },{$match: {agg: true}},{$sort: {category: 1}},{$group: {
        _id: "$_id",customer: {$first: "$customer"},sort_category: {$first: "$sort_category"},locations: {$push: "$locations"}
        }
    },{$project:{ _id: 0}}
])

,

您可以尝试自定义方式，

• $addFields将使用category和sort_category在locations数组中的每个对象内添加一个名为键$map的复制字段$reduce，{1} li>
• $unwind解构locations数组
我们在$sort数组中添加的
• category和locations字段
• $project删除category字段
• $group by _id并重建locations数组
• $project删除_id字段

db.collection.aggregate([
  {
    $addFields: {
      locations: {
        $map: {
          input: "$locations",as: "l",in: {
            $mergeObjects: [
              "$$l",{
                category: {
                  $reduce: {
                    input: { $objectToArray: "$$l" },initialValue: null,in: {
                      $cond: [{ $eq: ["$$this.k","$sort_category"] },"$$this.v","$$value"]
                    }
                  }
                }
              }
            ]
          }
        }
      }
    }
  },{ $unwind: "$locations" },{ $sort: { "locations.category": 1 } },{ $project: { "locations.category": 0 } },{
    $group: {
      _id: "$_id",customer: { $first: "$customer" },sort_category: { $first: "$sort_category" },locations: { $push: "$locations" }
    }
  },{ $project: { _id: 0 } }
])

Playground

---

如果您打算将 MongoDB升级到v4.4 ，或者这也将对其他人有所帮助，则$function是自定义操作和用户定义操作的选项。

共有3个属性：

• body函数定义。您可以将函数定义指定为BSON类型的Code或String，使用function(){定义我们自己的函数，我们传递了locations数组和sort_category这是动态字段，函数逻辑是排序的按降序
• args参数传递给函数主体
• lang身体使用的语言。您必须指定lang：“ js”

db.collection.aggregate([
  {
    $addFields: {
      locations: {
        $function: {

          body: function(locations,sort_category){
            return locations.sort(function(a,b){
              // DESCENDING ORDER
              return b[sort_category] - a[sort_category]
              // ASCENDING ORDER
              // return a[sort_category] - b[sort_category]
            })
          },args: ["$locations","$sort_category"],lang: "js"

        }
      }
    }
  }
])

有关限制和注意事项的更多准则Refer。