MongoDB Java驱动是否支持用$indexOfArray添加字段并实现自定义排序？

需求说明 #

需要对学生集合按照自定义等级顺序（Gold、Silver、Bronze）排序，而非常规的字母升序/降序。

示例数据 #

• 学生1：Name: A，grade: Silver
• 学生2：Name: B，grade: Gold
• 学生3：Name: C，grade: Bronze

预期排序结果 #

B（Gold）→ A（Silver）→ C（Bronze）

已实现的MongoDB Shell聚合查询 #

db.collection.aggregate([
  { $match: { filterQuery } },
  {
    $addFields: { 
      "customGradeOrder": { $indexOfArray: [ ["Gold", "Silver", "Bronze"], "$grade" ] } 
    }
  },
  { $sort: { customGradeOrder: 1 } }
]);

Java驱动实现方案 #

MongoDB Java驱动的Aggregates类完全支持通过$indexOfArray运算符添加字段并赋值，具体实现如下：

核心逻辑

1. 用Aggregates.addFields()对应Shell中的$addFields，新增自定义排序字段
2. 用Expressions.indexOfArray()封装$indexOfArray运算符，传入自定义等级数组和grade字段引用
3. 用Aggregates.sort()按新增的自定义字段升序排序

完整代码示例

import com.mongodb.client.MongoCollection;
import com.mongodb.client.model.Aggregates;
import com.mongodb.client.model.Sorts;
import com.mongodb.client.model.Filters;
import org.bson.Document;
import java.util.Arrays;
import static com.mongodb.client.model.Expressions.*;

// 获取学生集合
MongoCollection<Document> studentsCollection = mongoDatabase.getCollection("students");

// 构建聚合管道
List<Document> aggregationPipeline = Arrays.asList(
    // 可选：添加过滤条件，替换为你的实际查询
    Aggregates.match(Filters.exists("grade")),
    // 新增customGradeOrder字段，计算grade在自定义序列中的索引
    Aggregates.addFields(
        field("customGradeOrder", indexOfArray(
            Arrays.asList("Gold", "Silver", "Bronze"),
            field("grade")
        ))
    ),
    // 按自定义索引升序排序
    Aggregates.sort(Sorts.ascending("customGradeOrder"))
);

// 执行聚合并输出结果
studentsCollection.aggregate(aggregationPipeline).forEach(doc -> {
    System.out.printf("Name: %s, Grade: %s%n", doc.getString("Name"), doc.getString("grade"));
});

代码说明

• field("grade")对应Shell中的$grade，用于引用文档中的grade字段
• indexOfArray()方法会返回当前grade值在自定义数组中的位置索引，Gold对应0，Silver对应1，Bronze对应2，升序排序后就符合预期的自定义顺序
• 如果不需要过滤数据，可以去掉Aggregates.match()阶段

内容的提问来源于stack exchange，提问作者Madhav Kumar Jha