Enhance MongoDbRepository

framework/src/Volo.Abp.MongoDB/Volo/Abp/Domain/Repositories/MongoDB/IMongoDbRepository.cs

@@ -23,9 +23,9 @@ public interface IMongoDbRepository<TEntity> : IRepository<TEntity>
     [Obsolete("Use GetMongoQueryableAsync method.")]
     IMongoQueryable<TEntity> GetMongoQueryable();
 
-    Task<IMongoQueryable<TEntity>> GetMongoQueryableAsync(CancellationToken cancellationToken = default);
+    Task<IMongoQueryable<TEntity>> GetMongoQueryableAsync(CancellationToken cancellationToken = default, AggregateOptions options = null);
 
-    Task<IAggregateFluent<TEntity>> GetAggregateAsync(CancellationToken cancellationToken = default);
+    Task<IAggregateFluent<TEntity>> GetAggregateAsync(CancellationToken cancellationToken = default, AggregateOptions options = null);
 }
 
 public interface IMongoDbRepository<TEntity, TKey> : IMongoDbRepository<TEntity>, IRepository<TEntity, TKey>

framework/src/Volo.Abp.MongoDB/Volo/Abp/Domain/Repositories/MongoDB/MongoDbRepository.cs

@@ -518,12 +518,12 @@ public class MongoDbRepository<TMongoDbContext, TEntity>
         );
     }
 
-    public virtual Task<IMongoQueryable<TEntity>> GetMongoQueryableAsync(CancellationToken cancellationToken = default)
+    public virtual Task<IMongoQueryable<TEntity>> GetMongoQueryableAsync(CancellationToken cancellationToken = default, AggregateOptions aggregateOptions = null)
     {
-        return GetMongoQueryableAsync<TEntity>(cancellationToken);
+        return GetMongoQueryableAsync<TEntity>(cancellationToken, aggregateOptions);
     }
 
-    protected virtual async Task<IMongoQueryable<TOtherEntity>> GetMongoQueryableAsync<TOtherEntity>(CancellationToken cancellationToken = default)
+    protected virtual async Task<IMongoQueryable<TOtherEntity>> GetMongoQueryableAsync<TOtherEntity>(CancellationToken cancellationToken = default, AggregateOptions aggregateOptions = null)
     {
         cancellationToken = GetCancellationToken(cancellationToken);
 
@@ -532,12 +532,12 @@ public class MongoDbRepository<TMongoDbContext, TEntity>
 
         return ApplyDataFilters<IMongoQueryable<TOtherEntity>, TOtherEntity>(
             dbContext.SessionHandle != null
-                ? collection.AsQueryable(dbContext.SessionHandle)
-                : collection.AsQueryable()
+                ? collection.AsQueryable(dbContext.SessionHandle, aggregateOptions)
+                : collection.AsQueryable(aggregateOptions)
         );
     }
 
-    public virtual async Task<IAggregateFluent<TEntity>> GetAggregateAsync(CancellationToken cancellationToken = default)
+    public virtual async Task<IAggregateFluent<TEntity>> GetAggregateAsync(CancellationToken cancellationToken = default, AggregateOptions aggregateOptions = null)
     {
         cancellationToken = GetCancellationToken(cancellationToken);
 
@@ -545,8 +545,8 @@ public class MongoDbRepository<TMongoDbContext, TEntity>
         var collection = await GetCollectionAsync(cancellationToken);
 
         var aggregate = dbContext.SessionHandle != null
-                ? collection.Aggregate(dbContext.SessionHandle)
-                : collection.Aggregate();
+                ? collection.Aggregate(dbContext.SessionHandle, aggregateOptions)
+                : collection.Aggregate(aggregateOptions);
 
         if (typeof(ISoftDelete).IsAssignableFrom(typeof(TEntity)) && DataFilter.IsEnabled<ISoftDelete>())
         {

framework/src/Volo.Abp.MongoDB/Volo/Abp/Domain/Repositories/MongoDbCoreRepositoryExtensions.cs

@@ -43,16 +43,16 @@ public static class MongoDbCoreRepositoryExtensions
         return repository.ToMongoDbRepository().GetMongoQueryable();
     }
 
-    public static Task<IMongoQueryable<TEntity>> GetMongoQueryableAsync<TEntity>(this IReadOnlyBasicRepository<TEntity> repository, CancellationToken cancellationToken = default)
+    public static Task<IMongoQueryable<TEntity>> GetMongoQueryableAsync<TEntity>(this IReadOnlyBasicRepository<TEntity> repository, CancellationToken cancellationToken = default, AggregateOptions aggregateOptions = null)
         where TEntity : class, IEntity
     {
-        return repository.ToMongoDbRepository().GetMongoQueryableAsync(cancellationToken);
+        return repository.ToMongoDbRepository().GetMongoQueryableAsync(cancellationToken, aggregateOptions);
     }
 
-    public static Task<IAggregateFluent<TEntity>> GetAggregateAsync<TEntity>(this IReadOnlyBasicRepository<TEntity> repository, CancellationToken cancellationToken = default)
+    public static Task<IAggregateFluent<TEntity>> GetAggregateAsync<TEntity>(this IReadOnlyBasicRepository<TEntity> repository, CancellationToken cancellationToken = default, AggregateOptions aggregateOptions = null)
         where TEntity : class, IEntity
     {
-        return repository.ToMongoDbRepository().GetAggregateAsync(cancellationToken);
+        return repository.ToMongoDbRepository().GetAggregateAsync(cancellationToken, aggregateOptions);
     }
 
     public static IMongoDbRepository<TEntity> ToMongoDbRepository<TEntity>(this IReadOnlyBasicRepository<TEntity> repository)