SqlSugar/Src/Asp.NetCore2/MongoDb.Ado.data/ExecuteDbDataReaderItemsAsync/QueryAggregateHandlerAsync.cs

using MongoDB.Bson;
using MongoDB.Bson.Serialization;
using MongoDB.Driver;
using System;
using System.Collections.Generic;
using System.Data.Common;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;

namespace MongoDb.Ado.data
{
    public class QueryAggregateHandlerAsync : IQueryHandlerAsync
    {
        public HandlerContext Context { get; set; }
        public CancellationToken token { get; set; }
        public async Task<DbDataReader> HandlerAsync(IMongoCollection<BsonDocument> collection, BsonValue doc)
        {
            // 解析 JSON 字符串为 BsonArray
            var pipeline =  doc.AsBsonArray; ;

            // 构建聚合管道
            var aggregateFluent = Context?.IsAnyServerSession == true?
                collection.Aggregate<BsonDocument>(Context.ServerSession,pipeline.Select(stage => new BsonDocument(stage.AsBsonDocument)).ToArray()):
                collection.Aggregate<BsonDocument>(pipeline.Select(stage => new BsonDocument(stage.AsBsonDocument)).ToArray());

            // 执行聚合查询并返回 DbDataReader
            var cursor =await aggregateFluent.ToListAsync(token);
            var result = MongoDbDataReaderHelper.ToDataReader(cursor);
            return result;
        }
    }
}