diff --git a/src/NHibernate/Async/Multi/QueryBatchExtensions.cs b/src/NHibernate/Async/Multi/QueryBatchExtensions.cs index 5786847c97e..24e5db5fa7c 100644 --- a/src/NHibernate/Async/Multi/QueryBatchExtensions.cs +++ b/src/NHibernate/Async/Multi/QueryBatchExtensions.cs @@ -61,6 +61,16 @@ partial class FutureList : IFutureList } } + partial class FutureEnumerable : IFutureEnumerable + { + + public async Task> GetEnumerableAsync(CancellationToken cancellationToken = default(CancellationToken)) + { + cancellationToken.ThrowIfCancellationRequested(); + return await (_result.GetValueAsync(cancellationToken)).ConfigureAwait(false); + } + } + #endregion Helper classes } } diff --git a/src/NHibernate/Multi/QueryBatchExtensions.cs b/src/NHibernate/Multi/QueryBatchExtensions.cs index c5405ff9a66..3b10d428265 100644 --- a/src/NHibernate/Multi/QueryBatchExtensions.cs +++ b/src/NHibernate/Multi/QueryBatchExtensions.cs @@ -567,7 +567,7 @@ public IList Value } } - class FutureEnumerable : IFutureEnumerable + partial class FutureEnumerable : IFutureEnumerable { private readonly IFutureList _result; @@ -576,11 +576,6 @@ public FutureEnumerable(IQueryBatch batch, IQueryBatchItem query) _result = new FutureList(batch, query); } - public async Task> GetEnumerableAsync(CancellationToken cancellationToken = default(CancellationToken)) - { - return await _result.GetValueAsync(cancellationToken); - } - public IEnumerable GetEnumerable() { return _result.Value;