IClient.cs

Go to the documentation of this file.

using System.Runtime.CompilerServices;
using Fauna.Exceptions;
using Fauna.Mapping;
using Fauna.Serialization;
using Fauna.Types;
 
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,
        ISerializer<T> serializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
 
    public Task<QuerySuccess<object?>> QueryAsync(
        Query query,
        ISerializer serializer,
        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,
        ISerializer<T> elemSerializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
 
    public IAsyncEnumerable<Page<T>> PaginateAsync<T>(
        Page<T> page,
        ISerializer<T> elemSerializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Query query,
        ISerializer elemSerializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default);
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Page<object?> page,
        ISerializer elemSerializer,
        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,
        ISerializer<T> serializer,
        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, Serializer.Generate<T>(MappingCtx), queryOptions, cancel);
 
    public Task<QuerySuccess<object?>> QueryAsync(
        Query query,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default) =>
        QueryAsync<object?>(query, Serializer.Dynamic, queryOptions, cancel);
 
    public Task<QuerySuccess<T>> QueryAsync<T>(
        Query query,
        ISerializer<T> serializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default) =>
        QueryAsyncInternal(query, serializer, MappingCtx, queryOptions, cancel);
 
    public Task<QuerySuccess<object?>> QueryAsync(
        Query query,
        ISerializer serializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default) =>
        QueryAsync<object?>(query, (ISerializer<object?>)serializer, queryOptions, cancel);
 
    public IAsyncEnumerable<Page<T>> PaginateAsync<T>(
        Query query,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
        where T : notnull =>
        PaginateAsync(query, Serializer.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, Serializer.Generate<T>(MappingCtx), queryOptions, cancel);
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Query query,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default) =>
        PaginateAsync(query, Serializer.Dynamic, queryOptions, cancel);
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Page<object?> page,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default) =>
        PaginateAsync(page, Serializer.Dynamic, queryOptions, cancel);
 
    public IAsyncEnumerable<Page<T>> PaginateAsync<T>(
        Query query,
        ISerializer<T> elemSerializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
    {
        var serializer = new PageSerializer<T>(elemSerializer);
        return PaginateAsyncInternal(query, serializer, queryOptions, cancel);
    }
 
    public IAsyncEnumerable<Page<T>> PaginateAsync<T>(
        Page<T> page,
        ISerializer<T> elemSerializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
    {
        var serializer = new PageSerializer<T>(elemSerializer);
        return PaginateAsyncInternal(page, serializer, queryOptions, cancel);
    }
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Query query,
        ISerializer elemSerializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
    {
        var elemObjSer = (ISerializer<object?>)elemSerializer;
        var serializer = new PageSerializer<object?>(elemObjSer);
        return PaginateAsyncInternal(query, serializer, queryOptions, cancel);
    }
 
    public IAsyncEnumerable<Page<object?>> PaginateAsync(
        Page<object?> page,
        ISerializer elemSerializer,
        QueryOptions? queryOptions = null,
        CancellationToken cancel = default)
    {
        var elemObjSer = (ISerializer<object?>)elemSerializer;
        var serializer = new PageSerializer<object?>(elemObjSer);
        return PaginateAsyncInternal(page, serializer, queryOptions, cancel);
    }
 
    #endregion
 
    // Internally accessible for QuerySource use
    internal async IAsyncEnumerable<Page<T>> PaginateAsyncInternal<T>(
        Query query,
        PageSerializer<T> serializer,
        QueryOptions? queryOptions,
        [EnumeratorCancellation] CancellationToken cancel = default)
    {
        var p = await QueryAsyncInternal(query,
            serializer,
            MappingCtx,
            queryOptions,
            cancel);
 
        await foreach (var page in PaginateAsyncInternal(p.Data, serializer, queryOptions, cancel))
        {
            yield return page;
        }
    }
 
    private async IAsyncEnumerable<Page<T>> PaginateAsyncInternal<T>(
        Page<T> page,
        PageSerializer<T> serializer,
        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,
                serializer,
                MappingCtx,
                queryOptions,
                cancel);
 
            page = response.Data;
            yield return page;
        }
    }
}