import com.mongodb.MongoClient;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import org.bson.Document;

import java.util.Arrays;

public class MongoDBAggregation {
    public static void main(String[] args) {
        // 连接到MongoDB数据库
        MongoClient mongoClient = new MongoClient("localhost", 27017);
        MongoDatabase database = mongoClient.getDatabase("mydb");
        MongoCollection<Document> collection = database.getCollection("mycollection");

        // 构建聚合管道
        Document matchStage = new Document("$match", new Document("status", "A"));
        Document groupStage = new Document("$group", new Document("_id", "$category").append("total", new Document("$sum", "$quantity")));
        Document sortStage = new Document("$sort", new Document("total", -1));
        Document limitStage = new Document("$limit", 5);

        // 执行聚合操作
        collection.aggregate(Arrays.asList(matchStage, groupStage, sortStage, limitStage))
                .forEach(document -> System.out.println(document.toJson()));

        // 关闭MongoDB连接
        mongoClient.close();
    }
}