using System; using System.Linq; using System.Threading; using System.Threading.Tasks; using Data.Abstractions; using Data.Models.Client; using Data.Models.Server; using Microsoft.EntityFrameworkCore; using Microsoft.Extensions.Logging; using SharedLibraryCore; using SharedLibraryCore.Interfaces; using ILogger = Microsoft.Extensions.Logging.ILogger; namespace IW4MAdmin.Application.Misc { /// public class ServerDataViewer : IServerDataViewer { private readonly ILogger _logger; private readonly IDataValueCache _snapshotCache; private readonly IDataValueCache _serverStatsCache; private readonly TimeSpan? _cacheTimeSpan = Utilities.IsDevelopment ? TimeSpan.FromSeconds(1) : (TimeSpan?) TimeSpan.FromMinutes(1); public ServerDataViewer(ILogger logger, IDataValueCache snapshotCache, IDataValueCache serverStatsCache) { _logger = logger; _snapshotCache = snapshotCache; _serverStatsCache = serverStatsCache; } public async Task MaxConcurrentClientsAsync(long? serverId = null, TimeSpan? overPeriod = null, CancellationToken token = default) { _snapshotCache.SetCacheItem(async (snapshots, cancellationToken) => { var oldestEntry = overPeriod.HasValue ? DateTime.UtcNow - overPeriod.Value : DateTime.UtcNow.AddDays(-1); var maxClients = 0; if (serverId != null) { maxClients = await snapshots.Where(snapshot => snapshot.ServerId == serverId) .Where(snapshot => snapshot.CapturedAt >= oldestEntry) .MaxAsync(snapshot => (int?)snapshot.ClientCount, cancellationToken) ?? 0; } else { maxClients = await snapshots.Where(snapshot => snapshot.CapturedAt >= oldestEntry) .GroupBy(snapshot => snapshot.PeriodBlock) .Select(grp => grp.Sum(snapshot => (int?)snapshot.ClientCount)) .MaxAsync(cancellationToken) ?? 0; } _logger.LogDebug("Max concurrent clients since {Start} is {Clients}", oldestEntry, maxClients); return maxClients; }, nameof(MaxConcurrentClientsAsync), _cacheTimeSpan); try { return await _snapshotCache.GetCacheItem(nameof(MaxConcurrentClientsAsync), token); } catch (Exception ex) { _logger.LogError(ex, "Could not retrieve data for {Name}", nameof(MaxConcurrentClientsAsync)); return 0; } } public async Task<(int, int)> ClientCountsAsync(TimeSpan? overPeriod = null, CancellationToken token = default) { _serverStatsCache.SetCacheItem(async (set, cancellationToken) => { var count = await set.CountAsync(cancellationToken); var startOfPeriod = DateTime.UtcNow.AddHours(-overPeriod?.TotalHours ?? -24); var recentCount = await set.CountAsync(client => client.LastConnection >= startOfPeriod, cancellationToken); return (count, recentCount); }, nameof(_serverStatsCache), _cacheTimeSpan); try { return await _serverStatsCache.GetCacheItem(nameof(_serverStatsCache), token); } catch (Exception ex) { _logger.LogError(ex, "Could not retrieve data for {Name}", nameof(ClientCountsAsync)); return (0, 0); } } } }