如何使用Java MongoDB驱动按数组长度对MongoDB文档排序？

嗨，这个需求实现起来并不复杂，咱们可以借助MongoDB的聚合框架或者投影排序来搞定，下面给你两种实用的方案，适配最新版的MongoDB Java驱动（4.x及以上版本）：

方案一：使用聚合框架（推荐） #

聚合框架能帮你动态计算数组长度，再基于这个长度完成排序，逻辑清晰且灵活，适合多数场景。核心思路是：

1. 用$size操作符计算目标数组的长度，新增一个临时字段存储这个值
2. 按照临时字段的值进行升序/降序排序
3. （可选）移除临时字段，返回原始文档结构

Java代码示例 #

假设你的集合名为yourCollection，数组字段名为targetArray，代码如下：

import com.mongodb.client.MongoClients;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.model.Aggregates;
import com.mongodb.client.model.Projections;
import com.mongodb.client.model.Sorts;
import org.bson.Document;

import java.util.Arrays;

public class SortByArrayLength {
    public static void main(String[] args) {
        // 连接MongoDB
        try (var mongoClient = MongoClients.create("mongodb://localhost:27017")) {
            MongoDatabase database = mongoClient.getDatabase("yourDatabase");
            MongoCollection<Document> collection = database.getCollection("yourCollection");

            // 构建聚合管道
            var pipeline = Arrays.asList(
                // 新增临时字段arrayLength，存储数组长度
                Aggregates.addFields(new Document("arrayLength", new Document("$size", "$targetArray"))),
                // 按arrayLength升序排序，降序用Sorts.descending("arrayLength")
                Aggregates.sort(Sorts.ascending("arrayLength")),
                // 移除临时字段，只保留原始文档字段
                Aggregates.project(Projections.exclude("arrayLength"))
            );

            // 执行聚合并遍历结果
            collection.aggregate(pipeline).forEach(doc -> System.out.println(doc.toJson()));
        }
    }
}

方案二：投影+排序（简化版） #

如果只是在查询时需要按数组长度排序，不需要保留临时字段，也可以直接在投影中计算数组长度，再基于投影字段排序：

Java代码示例 #

import com.mongodb.client.MongoClients;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.model.Projections;
import com.mongodb.client.model.Sorts;
import org.bson.Document;

public class SortByArrayLengthSimple {
    public static void main(String[] args) {
        try (var mongoClient = MongoClients.create("mongodb://localhost:27017")) {
            MongoDatabase database = mongoClient.getDatabase("yourDatabase");
            MongoCollection<Document> collection = database.getCollection("yourCollection");

            // 投影时计算数组长度，同时排序
            collection.find()
                .projection(Projections.fields(
                    Projections.include("_id", "targetArray"), // 保留需要的字段
                    Projections.computed("arrayLength", new Document("$size", "$targetArray"))
                ))
                .sort(Sorts.ascending("arrayLength")) // 升序排序
                .forEach(doc -> System.out.println(doc.toJson()));
        }
    }
}

注意事项 #

• 记得替换代码中的yourDatabase、yourCollection、targetArray为你实际的数据库名、集合名和数组字段名
• 如果使用MongoDB 3.x版本的驱动，API写法会略有不同（比如MongoClient的创建方式），但核心的聚合逻辑是一致的
• 如果需要频繁按数组长度排序，建议给文档新增一个字段存储数组长度，每次更新数组时同步更新这个字段——因为$size无法使用普通索引，同步字段后可以直接基于它排序并利用索引提升性能

内容的提问来源于stack exchange，提问作者Marcolia