using DD.Persistence.Client.Clients.Interfaces;
using DD.Persistence.Models;
using DD.Persistence.Models.Common;
using System.Collections.Concurrent;

namespace DD.Persistence.Client.Clients.Mapping;
internal class TimestampedMappingClient(ITimestampedValuesClient client) : ITimestampedMappingClient
{
    /// <inheritdoc/>
    private readonly ConcurrentDictionary<Guid, TimestampedSetMapperBase> mapperCache = new();

    public async Task<int> AddRange(Guid discriminatorId, IEnumerable<TimestampedValuesDto> dtos, CancellationToken token)
    => await client.AddRange(discriminatorId, dtos, token);

    public async Task<int> Count(Guid discriminatorId, CancellationToken token)
    => await client.Count(discriminatorId, token);

    public void Dispose()
    {
        client.Dispose();
    }

    /// <inheritdoc/>
    public async Task<IEnumerable<T>> Get<T>(Guid discriminatorId, DateTimeOffset? geTimestamp, IEnumerable<string>? columnNames, int skip, int take, CancellationToken token)
    {
        var data = await Get([discriminatorId], geTimestamp, columnNames, skip, take, token);
        var mapper = GetMapper<T>(discriminatorId);

        return data.Select(mapper.DeserializeTimeStampedData);
    }

    /// <inheritdoc/>
    public async Task<IEnumerable<T>> GetLast<T>(Guid idDiscriminator, int take, CancellationToken token)
    {
        var data = await GetLast(idDiscriminator, take, token);
        var mapper = GetMapper<T>(idDiscriminator);

        return data.Select(mapper.DeserializeTimeStampedData);
    }


    /// <inheritdoc/>
    private TimestampedSetMapper<T> GetMapper<T>(Guid idDiscriminator)
    {
        return (TimestampedSetMapper<T>)mapperCache.GetOrAdd(idDiscriminator, name => new TimestampedSetMapper<T>(idDiscriminator));
    }

    public async Task<IEnumerable<TimestampedValuesDto>> Get(IEnumerable<Guid> discriminatorIds, DateTimeOffset? timestampBegin, IEnumerable<string>? columnNames, int skip, int take, CancellationToken token)
    => await client.Get(discriminatorIds, timestampBegin, columnNames, skip, take, token);

    public async Task<DatesRangeDto?> GetDatesRange(Guid discriminatorId, CancellationToken token)
    => await client.GetDatesRange(discriminatorId, token);

    public async Task<IEnumerable<TimestampedValuesDto>> GetFirst(Guid discriminatorId, int take, CancellationToken token)
    => await client.GetFirst(discriminatorId, take, token);

    public async Task<IEnumerable<TimestampedValuesDto>> GetGtDate(Guid discriminatorId, DateTimeOffset timestampBegin, CancellationToken token)
    => await client.GetGtDate(discriminatorId, timestampBegin, token);

    public async Task<IEnumerable<TimestampedValuesDto>> GetLast(Guid discriminatorId, int take, CancellationToken token)
    => await client.GetLast(discriminatorId, take, token);

    public async Task<IEnumerable<TimestampedValuesDto>> GetResampledData(Guid discriminatorId, DateTimeOffset timestampBegin, double intervalSec = 600, int approxPointsCount = 1024, CancellationToken token = default)
    => await client.GetResampledData(discriminatorId, timestampBegin, intervalSec, approxPointsCount, token);
}