QuerySourceToCollections.cs

Go to the documentation of this file.

using System.Collections;
 
namespace Fauna.Linq;
 
public partial class QuerySource<T>
{
    public List<T> ToList() => ToEnumerable().ToList();
    public async Task<List<T>> ToListAsync(CancellationToken cancel = default)
    {
        var ret = new List<T>();
        await foreach (var e in ToAsyncEnumerable(cancel)) ret.Add(e);
        return ret;
    }
 
    public T[] ToArray() => ToEnumerable().ToArray();
    public async Task<T[]> ToArrayAsync(CancellationToken cancel = default) =>
        (await ToListAsync(cancel)).ToArray();
 
    public HashSet<T> ToHashSet() => ToHashSet(null);
    public Task<HashSet<T>> ToHashSetAsync(CancellationToken cancel = default) =>
        ToHashSetAsync(null, cancel);
 
    public HashSet<T> ToHashSet(IEqualityComparer<T>? comparer) => ToEnumerable().ToHashSet(comparer);
    public async Task<HashSet<T>> ToHashSetAsync(IEqualityComparer<T>? comparer, CancellationToken cancel = default)
    {
        var ret = new HashSet<T>(comparer);
        await foreach (var e in ToAsyncEnumerable(cancel)) ret.Add(e);
        return ret;
    }
 
    public Dictionary<K, V> ToDictionary<K, V>(Func<T, K> getKey, Func<T, V> getValue) where K : notnull =>
        ToDictionary(getKey, getValue, null);
    public Task<Dictionary<K, V>> ToDictionaryAsync<K, V>(Func<T, K> getKey, Func<T, V> getValue, CancellationToken cancel = default) where K : notnull =>
        ToDictionaryAsync(getKey, getValue, null, cancel);
 
    public Dictionary<K, V> ToDictionary<K, V>(Func<T, K> getKey, Func<T, V> getValue, IEqualityComparer<K>? comparer) where K : notnull =>
        ToEnumerable().ToDictionary(getKey, getValue, comparer);
    public async Task<Dictionary<K, V>> ToDictionaryAsync<K, V>(Func<T, K> getKey, Func<T, V> getValue, IEqualityComparer<K>? comparer, CancellationToken cancel = default) where K : notnull
    {
        var ret = new Dictionary<K, V>(comparer);
        await foreach (var e in ToAsyncEnumerable(cancel)) ret[getKey(e)] = getValue(e);
        return ret;
    }
 
    public record struct QuerySourceEnumerable(QuerySource<T> Source) : IEnumerable<T>
    {
        public IEnumerator<T> GetEnumerator()
        {
            var pe = Source.PaginateAsync().GetAsyncEnumerator();
            try
            {
                var mv = pe.MoveNextAsync().AsTask();
                mv.Wait();
                while (mv.Result)
                {
                    var page = pe.Current;
 
                    foreach (var e in page.Data)
                    {
                        yield return e;
                    }
 
                    mv = pe.MoveNextAsync().AsTask();
                    mv.Wait();
                }
            }
            finally { pe.DisposeAsync(); }
        }
 
        IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
    }
}