persistence/Persistence.Repository/Repositories/TechMessagesRepository.cs

using Mapster;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Caching.Memory;
using Newtonsoft.Json.Linq;
using Persistence.Database.Entity;
using Persistence.Models;
using Persistence.Repositories;
using Persistence.Repository.Extensions;

namespace Persistence.Repository.Repositories
{
	public class TechMessagesRepository : ITechMessagesRepository
	{
		private static readonly string SystemCacheKey = $"{typeof(ADSystem).FullName}CacheKey";
		private readonly IMemoryCache memoryCache;
		private DbContext db;

		public TechMessagesRepository(DbContext db, IMemoryCache memoryCache)
		{
			this.memoryCache = memoryCache;
			this.db = db;
		}

		protected virtual IQueryable<TechMessage> GetQueryReadOnly() => db.Set<TechMessage>();

		public async Task<PaginationContainer<TechMessageDto>> GetPage(RequestDto request, CancellationToken token)
		{
			var query = GetQueryReadOnly();
			var entities = await query
				.SortBy(request.SortSettings)
				.Skip(request.Skip)
				.Take(request.Take)
				.ToListAsync();
			var dto = new PaginationContainer<TechMessageDto>()
			{
				Skip = request.Skip,
				Take = request.Take,
				Count = entities.Count,
				Items = entities.Select(e => e.Adapt<TechMessageDto>())
			};

			return dto;
		}

		public async Task<Dictionary<string, int>> GetStatistics(int? importantId, string? autoDrillingSystem, CancellationToken token)
		{
			var query = GetQueryReadOnly();
			var count = await query
				.Where(e => importantId == null || e.ImportantId == importantId)
				.Where(e => autoDrillingSystem == null || e.System.Name == autoDrillingSystem)
				.GroupBy(e => e.System.Name)
				.ToDictionaryAsync(e => e.Key, v => v.Count());

			return count;
		}

		public async Task<IEnumerable<ADSystemDto>> GetSystems(CancellationToken token)
		{
			var systems = await memoryCache.GetOrCreateAsync(SystemCacheKey, async f =>
			{
				f.AbsoluteExpirationRelativeToNow = TimeSpan.FromMinutes(60);

				var query = db.Set<ADSystem>();
				var entities = await query.ToListAsync();
				var dtos = entities.Select(e => e.Adapt<ADSystemDto>());

				return dtos;
			});

			return systems ?? [];
		}

		public async Task<int> InsertRange(IEnumerable<TechMessageDto> dtos, CancellationToken token)
		{
			var entities = dtos.Select(dto =>
			{
				var task = Task.Run(async () =>
				{
					var entity = dto.Adapt<TechMessage>();
					var systems = await GetSystems(token);
					var systemId = systems.FirstOrDefault(e => e.Name == dto.System)?.SystemId
						?? await CreateSystem(dto.System);

					entity.SystemId = systemId;

					return entity;
				});
				task.Wait();

				return task.Result;
			});

			await db.Set<TechMessage>().AddRangeAsync(entities, token);
			var result = await db.SaveChangesAsync(token);

			return result;
		}

		private async Task<Guid> CreateSystem(string name)
		{
			memoryCache.Remove(SystemCacheKey);

			var systemId = Guid.NewGuid();
			var entity = new ADSystem()
			{
				SystemId = systemId,
				Name = name
			};

			await db.Set<ADSystem>().AddAsync(entity);
			await db.SaveChangesAsync();

			return Guid.NewGuid();
		}
	}
}