QueryResponse.cs

Go to the documentation of this file.

using System.Net;
using System.Text.Json;
using Fauna.Mapping;
using Fauna.Serialization;
using static Fauna.Constants.ResponseFields;
 
namespace Fauna;
 
public abstract class QueryResponse
{
    public JsonElement RawJson { get; init; }
 
    public long LastSeenTxn { get; init; }
 
    public long SchemaVersion { get; init; }
 
    public string Summary { get; init; } = "";
 
    public Dictionary<string, string> QueryTags { get; init; } = new();
 
    public QueryStats Stats { get; init; }
 
    internal QueryResponse(JsonElement json)
    {
        RawJson = json;
 
        if (json.TryGetProperty(LastSeenTxnFieldName, out var elem))
        {
            if (elem.TryGetInt64(out var i)) LastSeenTxn = i;
        }
 
        if (json.TryGetProperty(SchemaVersionFieldName, out elem))
        {
            if (elem.TryGetInt64(out var i)) LastSeenTxn = i;
        }
 
        if (json.TryGetProperty(SummaryFieldName, out elem))
        {
            Summary = elem.GetString() ?? "";
        }
 
 
        if (json.TryGetProperty(QueryTagsFieldName, out elem))
        {
            var queryTagsString = elem.GetString();
 
            if (!string.IsNullOrEmpty(queryTagsString))
            {
                var tagPairs = queryTagsString.Split(',').Select(tag =>
                {
                    var tokens = tag.Split('=');
                    return KeyValuePair.Create(tokens[0], tokens[1]);
                });
 
                QueryTags = new Dictionary<string, string>(tagPairs);
            }
        }
 
        if (json.TryGetProperty(StatsFieldName, out elem))
        {
            Stats = elem.Deserialize<QueryStats>();
        }
    }
 
    public static QueryResponse? GetFromResponseBody<T>(
        MappingContext ctx,
        IDeserializer<T> deserializer,
        HttpStatusCode statusCode,
        string body)
    {
        try
        {
            var json = JsonSerializer.Deserialize<JsonElement>(body);
 
            if (statusCode is >= HttpStatusCode.OK and <= (HttpStatusCode)299)
            {
                return new QuerySuccess<T>(ctx, deserializer, json);
            }
 
            return new QueryFailure(statusCode, json);
        }
        catch (JsonException)
        {
            return null;
        }
    }
}
 
public sealed class QuerySuccess<T> : QueryResponse
{
    public T Data { get; init; }
 
    public string? StaticType { get; init; }
 
    public QuerySuccess(
        MappingContext ctx,
        IDeserializer<T> deserializer,
        JsonElement json)
        : base(json)
    {
        var dataText = json.GetProperty(DataFieldName).GetRawText();
        var reader = new Utf8FaunaReader(dataText);
        reader.Read();
        Data = deserializer.Deserialize(ctx, ref reader);
 
        if (json.TryGetProperty(StaticTypeFieldName, out var elem))
        {
            StaticType = elem.GetString();
        }
    }
}
 
public sealed class QueryFailure : QueryResponse
{
    public HttpStatusCode StatusCode { get; init; }
    public string ErrorCode { get; init; } = "";
    public string Message { get; init; } = "";
    public object? ConstraintFailures { get; init; }
    public object? Abort { get; init; }
 
    public QueryFailure(HttpStatusCode statusCode, JsonElement json) : base(json)
    {
        StatusCode = statusCode;
        if (!json.TryGetProperty(ErrorFieldName, out var elem)) return;
 
        var info = elem.Deserialize<ErrorInfo>();
        ErrorCode = info.Code ?? "";
        Message = info.Message ?? "";
        ConstraintFailures = info.ConstraintFailures;
        Abort = info.Abort;
    }
}