在Spring.Mongodb中按日期间隔聚合的Mongo模板

import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.aggregation.Aggregation;
import org.springframework.data.mongodb.core.aggregation.AggregationOperation;
import org.springframework.data.mongodb.core.aggregation.GroupOperation;
import org.springframework.data.mongodb.core.aggregation.MatchOperation;
import org.springframework.data.mongodb.core.query.Criteria;

import java.util.Date;
import java.util.List;

public class AggregationExample {

    private MongoTemplate mongoTemplate;

    public List<AggregationResult> aggregateByDateInterval(Date startDate, Date endDate, int interval) {
        // 创建日期范围的筛选条件
        Criteria dateCriteria = Criteria.where("date").gte(startDate).lte(endDate);

        // 创建日期间隔的分组操作
        GroupOperation groupByInterval = Aggregation.group()
                .addToSet("date").as("dates")
                .push("$$ROOT").as("documents");

        // 创建聚合操作，按日期范围筛选并按日期间隔分组
        AggregationOperation aggregation = Aggregation.newAggregation(
                Aggregation.match(dateCriteria),
                groupByInterval
        );

        // 执行聚合操作
        AggregationResults<AggregationResult> results = mongoTemplate.aggregate(
                aggregation, "collectionName", AggregationResult.class);

        return results.getMappedResults();
    }

    // 定义聚合结果的类
    private static class AggregationResult {
        private List<Date> dates;
        private List<Document> documents;

        // 省略getter和setter
    }
}