DD.Persistence.Client/Clients/Mapping/Abstractions/ITimestampedMappingClient.cs

@@ -3,12 +3,12 @@
 namespace DD.Persistence.Client.Clients.Mapping.Abstractions;
 
 /// <summary>
-/// 
+/// Маппинг - обертка для клиента по работе с данными
 /// </summary>
 public interface ITimestampedMappingClient : ITimestampedValuesClient
 {
     /// <summary>
-    /// Получить данные с фильтрацией для нескольких систем. Значение фильтра null - отключен
+    /// Получить данные с преобразованием к заданному типу
     /// </summary>
     /// <param name="discriminatorId"></param>
     /// <param name="geTimestamp"></param>
@@ -17,17 +17,29 @@ public interface ITimestampedMappingClient : ITimestampedValuesClient
     /// <param name="skip"></param>
     /// <param name="take"></param>
     /// <param name="token"></param>
-    Task<IEnumerable<T>> Get<T>(Guid discriminatorId, DateTimeOffset? geTimestamp, string? filterTree, IEnumerable<string>? columnNames, int skip, int take, CancellationToken token);
-    Task<IDictionary<Guid, object>> Gett(IEnumerable<Guid> discriminatorIds, string? filterTree, DateTimeOffset? timestampBegin, IEnumerable<string>? columnNames, int skip, int take, CancellationToken token);
+    Task<IEnumerable<T>> GetMapped<T>(Guid discriminatorId, DateTimeOffset? geTimestamp, string? filterTree, IEnumerable<string>? columnNames, int skip, int take, CancellationToken token);
+    
+    /// <summary>
+    /// Получить набор данных, преобразованных к соответствующим типам из заданного конфига
+    /// </summary>
+    /// <param name="discriminatorIds"></param>
+    /// <param name="timestampBegin"></param>
+    /// <param name="filterTree"></param>
+    /// <param name="columnNames"></param>
+    /// <param name="skip"></param>
+    /// <param name="take"></param>
+    /// <param name="token"></param>
+    /// <returns></returns>
+    Task<IDictionary<Guid, IEnumerable<object>>> GetMultiMapped(IEnumerable<Guid> discriminatorIds, DateTimeOffset? timestampBegin, string? filterTree, IEnumerable<string>? columnNames, int skip, int take, CancellationToken token);
 
 
     /// <summary>
-    /// 
+    /// Получить данные с конца с преобразованием к заданному типу
     /// </summary>
     /// <typeparam name="T"></typeparam>
     /// <param name="idDiscriminator"></param>
     /// <param name="take"></param>
     /// <param name="token"></param>
     /// <returns></returns>
-    Task<IEnumerable<T>> GetLast<T>(Guid idDiscriminator, int take, CancellationToken token);
+    Task<IEnumerable<T>> GetLastMapped<T>(Guid idDiscriminator, int take, CancellationToken token);
 }

DD.Persistence.Client/Clients/Mapping/Clients/SetpointMappingClient.cs

@@ -5,16 +5,11 @@ using DD.Persistence.Models.Common;
 using System.Text.Json;
 
 namespace DD.Persistence.Client.Clients.Mapping.Clients;
-internal class SetpointMappingClient(ISetpointClient setpointClient, Dictionary<Guid, Type> mappingConfigs) : ISetpointMappingClient
-{
-    public async Task Add(Guid setpointKey, object newValue, CancellationToken token)
-    => await setpointClient.Add(setpointKey, newValue, token);
 
-    public void Dispose()
+/// <inheritdoc/>
+public class SetpointMappingClient(ISetpointClient setpointClient, Dictionary<Guid, Type> mappingConfigs) : ISetpointMappingClient
 {
-        setpointClient.Dispose();
-    }
-
+    /// <inheritdoc/>
     public async Task<IEnumerable<SetpointValueDto>> GetCurrent(IEnumerable<Guid> setpointKeys, CancellationToken token)
     => (await setpointClient.GetCurrent(setpointKeys, token))
         .Select(x => new SetpointValueDto
@@ -23,15 +18,16 @@ internal class SetpointMappingClient(ISetpointClient setpointClient, Dictionary<
             Value = DeserializeValue(x.Key, (JsonElement)x.Value)
         });
 
+    /// <inheritdoc/>
     public async Task<Dictionary<Guid, object>> GetCurrentDictionary(IEnumerable<Guid> setpointConfigs, CancellationToken token)
     {
-        return (await setpointClient.GetCurrent(setpointConfigs, token))
+        var result = (await setpointClient.GetCurrent(setpointConfigs, token))
             .ToDictionary(x => x.Key, x => DeserializeValue(x.Key, (JsonElement)x.Value));
+
+        return result;
     }
 
-    public async Task<DatesRangeDto> GetDatesRangeAsync(CancellationToken token)
-    => await setpointClient.GetDatesRangeAsync(token);
-
+    /// <inheritdoc/>
     public async Task<IEnumerable<SetpointValueDto>> GetHistory(IEnumerable<Guid> setpointKeys, DateTimeOffset historyMoment, CancellationToken token)
     {
         var result = await setpointClient.GetHistory(setpointKeys, historyMoment, token);
@@ -42,6 +38,7 @@ internal class SetpointMappingClient(ISetpointClient setpointClient, Dictionary<
         return result;
     }
 
+    /// <inheritdoc/>
     public async Task<Dictionary<Guid, IEnumerable<SetpointLogDto>>> GetLog(IEnumerable<Guid> setpointKeys, CancellationToken token)
     {
         var result = await setpointClient.GetLog(setpointKeys, token);
@@ -52,6 +49,7 @@ internal class SetpointMappingClient(ISetpointClient setpointClient, Dictionary<
         return result;
     }
 
+    /// <inheritdoc/>
     public async Task<IEnumerable<SetpointLogDto>> GetPart(DateTimeOffset dateBegin, int take, CancellationToken token)
     {
         var res = await setpointClient.GetPart(dateBegin, take, token);
@@ -61,7 +59,19 @@ internal class SetpointMappingClient(ISetpointClient setpointClient, Dictionary<
         return res;
     }
 
+    /// <inheritdoc/>
+    public async Task Add(Guid setpointKey, object newValue, CancellationToken token)
+    => await setpointClient.Add(setpointKey, newValue, token);
 
+    /// <inheritdoc/>
+    public async Task<DatesRangeDto> GetDatesRangeAsync(CancellationToken token)
+    => await setpointClient.GetDatesRangeAsync(token);
+
+    /// <inheritdoc/>
+    public void Dispose()
+    {
+        setpointClient.Dispose();
+    }
 
     private object DeserializeValue(Guid key, JsonElement value)
     {
@@ -72,8 +82,10 @@ internal class SetpointMappingClient(ISetpointClient setpointClient, Dictionary<
     }
     private void DeserializeList(IEnumerable<SetpointLogDto>? result)
     {
+        if (result is null)
+            return;
+
         foreach (var log in result)
             log.Value = DeserializeValue(log.Key, (JsonElement)log.Value);
-
     }
 }

DD.Persistence.Client/Clients/Mapping/Clients/TimestampedMappingClient.cs

@@ -3,93 +3,123 @@ using DD.Persistence.Client.Clients.Mapping.Abstractions;
 using DD.Persistence.Models;
 using DD.Persistence.Models.Common;
 using System.Collections.Concurrent;
-using System.Reflection;
 
 namespace DD.Persistence.Client.Clients.Mapping.Clients;
+
+/// <inheritdoc/>
 public class TimestampedMappingClient(ITimestampedValuesClient client, Dictionary<Guid, Type>? mappingConfigs) : ITimestampedMappingClient
 {
-    /// <inheritdoc/>
-    private readonly ConcurrentDictionary<Guid, TimestampedSetMapperBase> mapperCache = new();
+    private readonly ConcurrentDictionary<Guid, TimestampedSetMapper> mapperCache = new();
 
+    /// <inheritdoc/>
+    public async Task<IEnumerable<T>> GetMapped<T>(Guid discriminatorId, DateTimeOffset? geTimestamp,
+        string? filterTree, IEnumerable<string>? columnNames, int skip, int take, CancellationToken token)
+    {
+        var data = await Get([discriminatorId], geTimestamp, filterTree, columnNames, skip, take, token);
+        var mapper = GetMapper<T>(discriminatorId);
+
+        var mappedDtos = data.Select(mapper.DeserializeTimeStampedData).OfType<T>();
+
+        return mappedDtos;
+    }
+
+    /// <inheritdoc/>
+    public async Task<IEnumerable<T>> GetLastMapped<T>(Guid idDiscriminator, int take, CancellationToken token)
+    {
+        var data = await GetLast(idDiscriminator, take, token);
+        var mapper = GetMapper<T>(idDiscriminator);
+
+        var mappedDtos = data.Select(mapper.DeserializeTimeStampedData).OfType<T>();
+
+        return mappedDtos;
+    }
+
+    /// <inheritdoc/>
+    public async Task<IDictionary<Guid, IEnumerable<object>>> GetMultiMapped(IEnumerable<Guid> discriminatorIds, DateTimeOffset? geTimestamp,
+        string? filterTree,  IEnumerable<string>? columnNames, int skip, int take, CancellationToken token)
+    {
+        var data = await client.Get(discriminatorIds, geTimestamp, filterTree, columnNames, skip, take, token);
+
+        var result = discriminatorIds
+            .ToDictionary(discriminatorId => discriminatorId, discriminatorId =>
+            {
+                if (mappingConfigs is null)
+                    throw new ArgumentNullException(nameof(mappingConfigs));
+
+                if (!mappingConfigs.TryGetValue(discriminatorId, out var type))
+                    throw new InvalidCastException();
+
+                var mapper = GetMapper(discriminatorId, type);
+
+                var mappedDtos = data
+                    .Where(e => e.DiscriminatorId == discriminatorId)
+                    .Select(mapper.DeserializeTimeStampedData);
+
+                return mappedDtos;
+            });
+
+        //var genericMapperType = typeof(TimestampedSetMapper<>).MakeGenericType(type);
+
+        //var ttype = typeof(TimestampedSetMapper<object>);
+        //object b = 0;
+
+        //var mapper = MapperMethodInfo
+        //    .MakeGenericMethod([type])
+        //    .Invoke(this, [discriminatorId]);
+
+        //var deserializeMethod = genericMapperType
+        //    .GetMethod(nameof(TimestampedSetMapper<object>.DeserializeTimeStampedData))!;
+        // ToDo: приводим к Dictionary
+
+        return result;
+    }
+
+    /// <inheritdoc/>
     public async Task<int> AddRange(Guid discriminatorId, IEnumerable<TimestampedValuesDto> dtos, CancellationToken token)
     => await client.AddRange(discriminatorId, dtos, token);
 
+    /// <inheritdoc/>
     public async Task<int> Count(Guid discriminatorId, CancellationToken token)
     => await client.Count(discriminatorId, token);
 
+    /// <inheritdoc/>
+    public async Task<IEnumerable<TimestampedValuesDto>> Get(IEnumerable<Guid> discriminatorIds, DateTimeOffset? timestampBegin,
+        string? filterTree, IEnumerable<string>? columnNames, int skip, int take, CancellationToken token)
+    => await client.Get(discriminatorIds, timestampBegin, filterTree, columnNames, skip, take, token);
+
+    /// <inheritdoc/>
+    public async Task<DatesRangeDto?> GetDatesRange(Guid discriminatorId, CancellationToken token)
+    => await client.GetDatesRange(discriminatorId, token);
+
+    /// <inheritdoc/>
+    public async Task<IEnumerable<TimestampedValuesDto>> GetFirst(Guid discriminatorId, int take, CancellationToken token)
+    => await client.GetFirst(discriminatorId, take, token);
+
+    /// <inheritdoc/>
+    public async Task<IEnumerable<TimestampedValuesDto>> GetGtDate(Guid discriminatorId, DateTimeOffset timestampBegin, CancellationToken token)
+    => await client.GetGtDate(discriminatorId, timestampBegin, token);
+
+    /// <inheritdoc/>
+    public async Task<IEnumerable<TimestampedValuesDto>> GetLast(Guid discriminatorId, int take, CancellationToken token)
+    => await client.GetLast(discriminatorId, take, token);
+
+    /// <inheritdoc/>
+    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);
+
+    /// <inheritdoc/>
     public void Dispose()
     {
         client.Dispose();
     }
 
-    /// <inheritdoc/>
-    public async Task<IEnumerable<T>> Get<T>(Guid discriminatorId, DateTimeOffset? geTimestamp, string? filterTree, IEnumerable<string>? columnNames, int skip, int take, CancellationToken token)
+    private TimestampedSetMapper GetMapper(Guid idDiscriminator, Type type)
     {
-        var data = await Get([discriminatorId], geTimestamp, filterTree, columnNames, skip, take, token);
-        var mapper = GetMapper<T>(discriminatorId);
-
-        return data.Select(mapper.DeserializeTimeStampedData);
+        return mapperCache.GetOrAdd(idDiscriminator, name => new TimestampedSetMapper(idDiscriminator, type));
     }
 
-    /// <inheritdoc/>
-    public async Task<IEnumerable<T>> GetLast<T>(Guid idDiscriminator, int take, CancellationToken token)
+    private TimestampedSetMapper GetMapper<T>(Guid idDiscriminator)
     {
-        var data = await GetLast(idDiscriminator, take, token);
-        var mapper = GetMapper<T>(idDiscriminator);
-
-        return data.Select(mapper.DeserializeTimeStampedData);
-    }
-
-    public async Task<IDictionary<Guid, object>> Gett(IEnumerable<Guid> discriminatorIds, string? filterTree, DateTimeOffset? timestampBegin, IEnumerable<string>? columnNames, int skip, int take, CancellationToken token)
-    {
-        var data = await client.Get(discriminatorIds, timestampBegin, filterTree, columnNames, skip, take, token);
-
-        // ToDo: рефакторинг
-        foreach (var discriminatorId in discriminatorIds)
-        {
-            if (mappingConfigs.TryGetValue(discriminatorId, out var type))
-            {
-                var genericType = typeof(TimestampedSetMapper<>).MakeGenericType(type);
-
-                var mapper =
-                    typeof(TimestampedMappingClient)
-                    .GetMethod(nameof(GetMapper))!
-                    .MakeGenericMethod([type])
-                    .Invoke(this, [discriminatorId]);
-                var mapperInstance = Activator.CreateInstance(genericType, mapper); // ToDo: возможно не нужно
-
-                var deserializeMethod = genericType
-                    .GetMethod("DeserializeTimeStampedData")!;
-
-                var d = data.Select(e => deserializeMethod.Invoke(mapperInstance, [e]));
-                // ToDo: приводим к Dictionary
+        return mapperCache.GetOrAdd(idDiscriminator, name => new TimestampedSetMapper(idDiscriminator, typeof(T)));
     }
 }
-
-        return new Dictionary<Guid, object>();
-    }
-
-    /// <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, string? filterTree, IEnumerable<string>? columnNames, int skip, int take, CancellationToken token)
-    => await client.Get(discriminatorIds, timestampBegin, filterTree, 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);
-}

DD.Persistence.Client/Clients/Mapping/TimestampedSetMapper.cs

@@ -6,60 +6,53 @@ using System.Text.Json;
 
 namespace DD.Persistence.Client.Clients.Mapping;
 
-internal abstract class TimestampedSetMapperBase
+internal class TimestampedSetMapper
 {
-    public abstract object Map(TimestampedValuesDto data);
-
-}
-internal class TimestampedSetMapper<T> : TimestampedSetMapperBase
-{
-    private readonly Type entityType = typeof(T);
+    private readonly Type entityType;
     public Guid IdDiscriminator { get; }
     private readonly ConcurrentDictionary<string, PropertyInfo?> PropertyCache = new();
 
-    public TimestampedSetMapper(Guid idDiscriminator)
+    public TimestampedSetMapper(Guid idDiscriminator, Type entityType)
     {
         IdDiscriminator = idDiscriminator;
+        this.entityType = entityType;
     }
 
-    public override object Map(TimestampedValuesDto data)
+    public object DeserializeTimeStampedData(TimestampedValuesDto data)
     {
-        return DeserializeTimeStampedData(data)!;
-    }
-
-    public T DeserializeTimeStampedData(TimestampedValuesDto data)
-    {
-
         if (entityType.IsValueType)
             return MapStruct(data);
-        else
+        
         return MapClass(data);
     }
 
-    private T MapClass(TimestampedValuesDto data)
+    private object MapClass(TimestampedValuesDto data)
     {
-        var entity = (T)RuntimeHelpers.GetUninitializedObject(typeof(T));
+        var entity = RuntimeHelpers.GetUninitializedObject(entityType);
         foreach (var (propertyName, value) in data.Values)
         {
             if (value is JsonElement jsonElement)
                 SetPropertyValueFromJson(ref entity, propertyName, jsonElement);
         }
-        SetPropertyValue(ref entity, "Timestamp", data.Timestamp);
+        SetPropertyValue(ref entity, nameof(TimestampedValuesDto.Timestamp), data.Timestamp);
+        SetPropertyValue(ref entity, nameof(TimestampedValuesDto.DiscriminatorId), data.DiscriminatorId);
+
         return entity;
     }
 
-    private T MapStruct(TimestampedValuesDto data)
+    private object MapStruct(TimestampedValuesDto data)
     {
-        var entity = Activator.CreateInstance<T>();
+        var entity = Activator.CreateInstance(entityType);
         object boxedEntity = entity!;
         foreach (var (propertyName, value) in data.Values)
         {
             if (value is JsonElement jsonElement)
                 SetPropertyValueForStructFromJson(ref boxedEntity, propertyName, jsonElement);
         }
-        SetPropertyValueForStruct(ref boxedEntity, "Timestamp", data.Timestamp);
+        SetPropertyValueForStruct(ref boxedEntity, nameof(TimestampedValuesDto.Timestamp), data.Timestamp);
+        SetPropertyValueForStruct(ref boxedEntity, nameof(TimestampedValuesDto.DiscriminatorId), data.DiscriminatorId);
 
-        return (T)boxedEntity;
+        return boxedEntity;
     }
 
     private void SetPropertyValueForStructFromJson(ref object entity, string propertyName, JsonElement element)
@@ -73,9 +66,7 @@ internal class TimestampedSetMapper<T> : TimestampedSetMapperBase
             var value = element.Deserialize(property.PropertyType);
             property.SetValue(entity, value);
         }
-        catch (Exception ex)
-        {
-        }
+        catch (Exception) { }
     }
     private void SetPropertyValueForStruct(ref object entity, string propertyName, object value)
     {
@@ -88,13 +79,11 @@ internal class TimestampedSetMapper<T> : TimestampedSetMapperBase
             var convertedValue = Convert.ChangeType(value, property.PropertyType);
             property.SetValue(entity, convertedValue);
         }
-        catch (Exception ex)
-        {
-        }
+        catch (Exception) { }
     }
 
 
-    private void SetPropertyValueFromJson(ref T entity, string propertyName, JsonElement jsonElement)
+    private void SetPropertyValueFromJson(ref object entity, string propertyName, JsonElement jsonElement)
     {
         var property = GetPropertyInfo(propertyName);
 
@@ -106,13 +95,10 @@ internal class TimestampedSetMapper<T> : TimestampedSetMapperBase
             var value = jsonElement.Deserialize(property.PropertyType);
             property.SetValue(entity, value);
         }
-        catch (Exception ex)
-        {
-
-        }
+        catch (Exception) { }
     }
 
-    private void SetPropertyValue(ref T entity, string propertyName, object value)
+    private void SetPropertyValue(ref object entity, string propertyName, object value)
     {
         var property = GetPropertyInfo(propertyName);
         if (property is null)
@@ -123,9 +109,7 @@ internal class TimestampedSetMapper<T> : TimestampedSetMapperBase
             var convertedValue = Convert.ChangeType(value, property.PropertyType);
             property.SetValue(entity, convertedValue);
         }
-        catch (Exception ex)
-        {
-        }
+        catch (Exception) { }
     }
 
     private PropertyInfo? GetPropertyInfo(string propertyName)

DD.Persistence.Models/TimestampedValuesDto.cs

@@ -7,6 +7,11 @@ namespace DD.Persistence.Models;
 /// </summary>
 public class TimestampedValuesDto : ITimestampAbstractDto
 {
+    /// <summary>
+    /// Дискриминатор
+    /// </summary>
+    public Guid DiscriminatorId { get; set; }
+
     /// <summary>
     /// Временная отметка
     /// </summary>

DD.Persistence.Test/MappingClientsTest.cs

@@ -1,29 +1,24 @@
 using DD.Persistence.Client.Clients.Interfaces;
-using DD.Persistence.Client.Clients.Mapping.Abstractions;
 using DD.Persistence.Client.Clients.Mapping.Clients;
-using DD.Persistence.Repositories;
-using DD.Persistence.Services.Interfaces;
+using DD.Persistence.Models;
 using NSubstitute;
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
+using System.Text.Json;
 
 namespace DD.Persistence.Test;
+
+public record FirstTestDto(Guid DiscriminatorId, DateTimeOffset Timestamp, int Id, string? Value);
+
+public record SecondTestDto(Guid DiscriminatorId, DateTimeOffset Timestamp, int Id, double Capacity);
+
 public class MappingClientsTest
 {
     private readonly ITimestampedValuesClient timestampedValuesClient = Substitute.For<ITimestampedValuesClient>();
     private readonly TimestampedMappingClient timestampedMappingClient;
-    public record TestDto
-    {
-        public int Id { get; set; }
-        public string? Value { get; set; }
-    }
 
     private readonly Dictionary<Guid, Type> mappingConfigs = new Dictionary<Guid, Type>()
     {
-        { Guid.NewGuid(), typeof(TestDto) }
+        { Guid.NewGuid(), typeof(FirstTestDto) },
+        { Guid.NewGuid(), typeof(SecondTestDto) }
     };
 
     public MappingClientsTest()
@@ -32,9 +27,59 @@ public class MappingClientsTest
     }
 
     [Fact]
-    public async void Test()
+    public async Task GetMultiMapped()
     {
+        // Arrange
         var discriminatorIds = mappingConfigs.Keys;
-        var result = await timestampedMappingClient.Gett(discriminatorIds, null, null, null, 0, 1, CancellationToken.None);
+        var firstDiscriminatorId = discriminatorIds.First();
+        var secondDiscriminatorId = discriminatorIds.Last();
+        var getResult = new[]
+        {
+            new TimestampedValuesDto()
+            {
+                DiscriminatorId = firstDiscriminatorId,
+                Timestamp = DateTime.UtcNow,
+                Values = new Dictionary<string, object>
+                {
+                    { nameof(FirstTestDto.Id), JsonDocument.Parse(JsonSerializer.Serialize(1)).RootElement },
+                    { nameof(FirstTestDto.Value), JsonDocument.Parse(JsonSerializer.Serialize("string1")).RootElement}
+                }
+            },
+            new TimestampedValuesDto()
+            {
+                DiscriminatorId = secondDiscriminatorId,
+                Timestamp = DateTime.UtcNow,
+                Values = new Dictionary<string, object>
+                {
+                    { nameof(SecondTestDto.Id), JsonDocument.Parse(JsonSerializer.Serialize(1)).RootElement },
+                    { nameof(SecondTestDto.Capacity), JsonDocument.Parse(JsonSerializer.Serialize(0.1)).RootElement}
+                }
+            }
+        };
+        timestampedValuesClient
+            .Get(discriminatorIds, null, null, null, 0, 1, CancellationToken.None)
+            .ReturnsForAnyArgs(getResult);
+
+        // Act
+        var result = await timestampedMappingClient.GetMultiMapped(discriminatorIds, null, null, null, 0, 1, CancellationToken.None);
+
+        // Assert
+        Assert.NotNull(result);
+        Assert.NotEmpty(result);
+        Assert.Equal(getResult.Count(), result.Count());
+
+        var firstActualDto = (FirstTestDto) result[firstDiscriminatorId].First();
+        Assert.NotNull(firstActualDto);
+
+        var actualId = firstActualDto.Id.ToString();
+        var expectedId = getResult[0].Values[nameof(FirstTestDto.Id)].ToString();
+        Assert.Equal(expectedId, actualId);
+
+        var secondActualDto = (SecondTestDto) result[secondDiscriminatorId].First();
+        Assert.NotNull(secondActualDto);
+
+        actualId = secondActualDto.Id.ToString();
+        expectedId = getResult[1].Values[nameof(SecondTestDto.Id)].ToString();
+        Assert.Equal(expectedId, actualId);
     }
 }

DD.Persistence/Services/TimestampedValuesService.cs

@@ -122,6 +122,7 @@ public class TimestampedValuesService : ITimestampedValuesService
             {
                 var dto = new TimestampedValuesDto()
                 {
+                    DiscriminatorId = keyValuePair.Key,
                     Timestamp = Timestamp.ToUniversalTime(),
                     Values = dataScheme
                         .ToDictionary(k => k.PropertyName, v => Values[v.Index])