persistence/Persistence.Benchmark/Tests/WitsDataBenchmark.cs
2024-12-25 13:54:07 +05:00

103 lines
3.4 KiB
C#

using DD.Persistence.Client;
using DD.Persistence.Models;
using Microsoft.Extensions.DependencyInjection;
using System.Diagnostics;
namespace Persistence.Benchmark.Tests;
public static class WitsDataBenchmark
{
private const string discriminatorId = "fef21bfd-e924-473d-b6c8-e4377e38245d";
public static async Task ExecuteTest(int count)
{
try
{
var factory = new WebAppFactoryFixture();
var scope = factory.Services.CreateScope();
var persistenceClientFactory = scope.ServiceProvider
.GetRequiredService<PersistenceClientFactory>();
var client = persistenceClientFactory.GetWitsDataClient();
var source = new CancellationTokenSource(TimeSpan.FromSeconds(6000));
var data = GenerateData(count);
var sw = new Stopwatch();
var abscissaData = new List<double>() { };
var ordinateData = new List<double>() { };
var saved = 0;
foreach (var item in data)
{
var time = sw.Elapsed;
sw.Start();
var response = await client.AddRange(item, source.Token);
sw.Stop();
Console.WriteLine($"Сохранено: {response.ToString()}");
saved = saved + response;
abscissaData.Add(saved / 100_000);
Console.WriteLine($"Затрачено времени на сохранение части: {sw.Elapsed - time}");
ordinateData.Add((double)(sw.Elapsed - time).TotalSeconds);
}
Console.WriteLine($"Затрачено времени на сохранение: {sw.Elapsed}");
var dis = Guid.Parse(discriminatorId);
var date = DateTime.Now.AddDays(-1);
sw = new Stopwatch();
sw.Start();
var get = await client.GetPart(dis, date, count, source.Token);
sw.Stop();
Console.WriteLine($"Затрачено времени на вычитку: {sw.Elapsed}");
}
catch (Exception ex) {
Console.WriteLine(ex.Message);
}
}
private static IEnumerable<IEnumerable<WitsDataDto>> GenerateData(int countToCreate)
{
var result = new List<List<WitsDataDto>>();
int enumerableCount = countToCreate / 25_000 + (countToCreate % 25_000 == 0 ? 0 : 1);
for (var k = 0; k < enumerableCount; k++)
{
var dtos = new List<WitsDataDto>();
for (var j = 0; j < 1000; j++)
{
for (var i = 0; i < 25; i++)
{
var timestamped = DateTimeOffset.UtcNow;
var random = new Random();
dtos.Add(new WitsDataDto()
{
DiscriminatorId = Guid.Parse(discriminatorId),
Timestamped = timestamped.AddSeconds(i),
Values = new List<WitsValueDto>()
{
new WitsValueDto()
{
RecordId = i + 1,
ItemId = i + 1,
Value = random.Next(1, 100)
}
}
});
}
}
result.Add(dtos);
}
return result;
}
}