MongoDB聚合查询：如何获取包含重复字段的完整文档信息

import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.aggregation.Aggregation;
import org.springframework.data.mongodb.core.aggregation.TypedAggregation;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.domain.Sort.Direction;
import org.bson.Document;
import java.util.List;
import java.util.stream.Collectors;

public class MongoDuplicateDataFetcher {

    private final MongoTemplate mongoTemplate;
    private final String collectionName;

    public MongoDuplicateDataFetcher(MongoTemplate mongoTemplate, String collectionName) {
        this.mongoTemplate = mongoTemplate;
        this.collectionName = collectionName;
    }

    /**
     * 从MongoDB集合中获取所有匹配的numBerId字段，包括重复值。
     * 此方法会返回一个可能包含重复numBerId的字符串列表。
     *
     * @param numBerId 匹配的numBerId前缀
     * @return 包含所有匹配numBerId的列表，可能包含重复值
     */
    public List<String> getAllNumBerIdsWithDuplicates(String numBerId) {
        TypedAggregation<Document> agg = TypedAggregation.newAggregation(Document.class,
            // 阶段1: 匹配符合条件的文档
            TypedAggregation.match(Criteria.where("numBerId").regex("^" + numBerId, "i")
                    .andOperator(Criteria.where("numBerId").ne(""))),
            // 阶段2: 限制返回文档数量
            TypedAggregation.limit(20000),
            // 阶段3: 对结果进行排序，这里选择按numBerId字段升序
            // 如果需要对整个文档进行排序，可以根据其他字段进行
            TypedAggregation.sort(Direction.ASC, "numBerId")
            // 注意: 移除了TypedAggregation.group("numBerId")，以保留所有文档和重复值
        );

        // 执行聚合，获取映射后的结果列表。
        // 此时，每个Document都是一个完整的匹配文档。
        List<Document> results = mongoTemplate.aggregate(agg, collectionName, Document.class).getMappedResults();

        // 从每个文档中提取numBerId字段并收集成列表
        return results.stream()
                .map(d -> (String) d.get("numBerId"))
                .collect(Collectors.toList());
    }

    /**
     * 从MongoDB集合中获取所有匹配的完整文档，包括重复值。
     * 此方法会返回一个包含完整文档对象的列表。
     *
     * @param numBerId 匹配的numBerId前缀
     * @return 包含所有匹配文档的列表，可能包含重复值
     */
    public List<Document> getAllDocumentsWithDuplicates(String numBerId) {
        TypedAggregation<Document> agg = TypedAggregation.newAggregation(Document.class,
            // 阶段1: 匹配符合条件的文档
            TypedAggregation.match(Criteria.where("numBerId").regex("^" + numBerId, "i")
                    .andOperator(Criteria.where("numBerId").ne(""))),
            // 阶段2: 限制返回文档数量
            TypedAggregation.limit(20000),
            // 阶段3: 对结果进行排序
            TypedAggregation.sort(Direction.ASC, "numBerId")
        );
        // 直接返回映射后的完整文档列表
        return mongoTemplate.aggregate(agg, collectionName, Document.class).getMappedResults();
    }
}