IClient.cs

Go to the documentation of this file.

using System.Runtime.CompilerServices;
using Fauna.Exceptions;
using Fauna.Serialization;
using Fauna.Types;
using Fauna.Mapping;
 
namespace Fauna;
 
interface IClient
{
    public Task<QuerySuccess<T>> QueryAsync<T>(
        Query query,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
        where T : notnull;
 
    public Task<QuerySuccess<object?>> QueryAsync(
        Query query,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
 
    public Task<QuerySuccess<T>> QueryAsync<T>(
        Query query,
        IDeserializer<T> deserializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
 
    public Task<QuerySuccess<object?>> QueryAsync(
        Query query,
        IDeserializer deserializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
 
    public IAsyncEnumerable<Page<T>> PaginateAsync<T>(
        Query query,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
        where T : notnull;
 
    public IAsyncEnumerable<Page<T>> PaginateAsync<T>(
        Page<T> page,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
        where T : notnull;
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Query query,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Page<object?> page,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
 
    public IAsyncEnumerable<Page<T>> PaginateAsync<T>(
        Query query,
        IDeserializer<T> elemDeserializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
 
    public IAsyncEnumerable<Page<T>> PaginateAsync<T>(
        Page<T> page,
        IDeserializer<T> elemDeserializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Query query,
        IDeserializer elemDeserializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Page<object?> page,
        IDeserializer elemDeserializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
}
 
 
public abstract class BaseClient : IClient
{
    internal BaseClient() { }
 
    internal abstract MappingContext MappingCtx { get; }
 
    internal abstract Task<QuerySuccess<T>> QueryAsyncInternal<T>(
        Query query,
        IDeserializer<T> deserializer,
        MappingContext ctx,
        QueryOptions? queryOptions,
        CancellationToken cancel
    );
 
    #region IClient
 
    public Task<QuerySuccess<T>> QueryAsync<T>(
        Query query,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
        where T : notnull =>
        QueryAsync<T>(query, Deserializer.Generate<T>(MappingCtx), queryOptions, cancel);
 
    public Task<QuerySuccess<object?>> QueryAsync(
        Query query,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default) =>
        QueryAsync<object?>(query, Deserializer.Dynamic, queryOptions, cancel);
 
    public Task<QuerySuccess<T>> QueryAsync<T>(
        Query query,
        IDeserializer<T> deserializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default) =>
        QueryAsyncInternal(query, deserializer, MappingCtx, queryOptions, cancel);
 
    public Task<QuerySuccess<object?>> QueryAsync(
        Query query,
        IDeserializer deserializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default) =>
        QueryAsync<object?>(query, (IDeserializer<object?>)deserializer, queryOptions, cancel);
 
    public IAsyncEnumerable<Page<T>> PaginateAsync<T>(
        Query query,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
        where T : notnull =>
        PaginateAsync(query, Deserializer.Generate<T>(MappingCtx), queryOptions, cancel);
 
    public IAsyncEnumerable<Page<T>> PaginateAsync<T>(
        Page<T> page,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
        where T : notnull =>
        PaginateAsync(page, Deserializer.Generate<T>(MappingCtx), queryOptions, cancel);
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Query query,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default) =>
        PaginateAsync(query, Deserializer.Dynamic, queryOptions, cancel);
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Page<object?> page,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default) =>
        PaginateAsync(page, Deserializer.Dynamic, queryOptions, cancel);
 
    public IAsyncEnumerable<Page<T>> PaginateAsync<T>(
        Query query,
        IDeserializer<T> elemDeserializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
    {
        var deserializer = new PageDeserializer<T>(elemDeserializer);
        return PaginateAsyncInternal(query, deserializer, queryOptions, cancel);
    }
 
    public IAsyncEnumerable<Page<T>> PaginateAsync<T>(
        Page<T> page,
        IDeserializer<T> elemDeserializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
    {
        var deserializer = new PageDeserializer<T>(elemDeserializer);
        return PaginateAsyncInternal(page, deserializer, queryOptions, cancel);
    }
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Query query,
        IDeserializer elemDeserializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
    {
        var elemObjDeser = (IDeserializer<object?>)elemDeserializer;
        var deserializer = new PageDeserializer<object?>(elemObjDeser);
        return PaginateAsyncInternal(query, deserializer, queryOptions, cancel);
    }
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Page<object?> page,
        IDeserializer elemDeserializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
    {
        var elemObjDeser = (IDeserializer<object?>)elemDeserializer;
        var deserializer = new PageDeserializer<object?>(elemObjDeser);
        return PaginateAsyncInternal(page, deserializer, queryOptions, cancel);
    }
 
    #endregion
 
    // Internally accessible for QuerySource use
    internal async IAsyncEnumerable<Page<T>> PaginateAsyncInternal<T>(
        Query query,
        PageDeserializer<T> deserializer,
        QueryOptions? queryOptions,
        [EnumeratorCancellation] CancellationToken cancel = default)
    {
        var p = await QueryAsyncInternal(query,
            deserializer,
            MappingCtx,
            queryOptions,
            cancel);
 
        await foreach (var page in PaginateAsyncInternal(p.Data, deserializer, queryOptions, cancel))
        {
            yield return page;
        }
    }
 
    private async IAsyncEnumerable<Page<T>> PaginateAsyncInternal<T>(
        Page<T> page,
        PageDeserializer<T> deserializer,
        QueryOptions? queryOptions,
        [EnumeratorCancellation] CancellationToken cancel = default)
    {
        yield return page;
 
        while (page.After is not null)
        {
            var q = new QueryExpr(new QueryLiteral($"Set.paginate('{page.After}')"));
 
            var response = await QueryAsyncInternal(q,
                deserializer,
                MappingCtx,
                queryOptions,
                cancel);
 
            page = response.Data;
            yield return page;
        }
    }
}