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

Question

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

{
    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 字段中给出的参数。

Answer 1

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

db.tests.aggregate([
    {$unwind:"$locations"}, 
    {$project: {
        _id: 1,
        customer: 1,
        sort_category: 1,
        locations: 1,
        locationsKV:{$objectToArray:"$locations"}
        }
    }, 
    {$unwind:"$locationsKV"}, 
    {$project:{
        _id: 1,
        customer: 1,
        sort_category: 1,
        locations: 1,
        locationsKV: 1,
        category: {
            $cond:[{$eq: ["$sort_category","$locationsKV.k"]}, 
            "$locationsKV.v", 0]},
        agg: {
            $cond: [{$eq: ["$sort_category","$locationsKV.k"]}, 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}}
])

Answer 2

您可以尝试自定义方式，

• $addFields 将使用 $map 和 $reduce 在 locations 数组中的每个对象中添加一个名为 category 的密钥 sort_category 的复制字段
• $unwind解构locations数组
• $sort by 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。