78 lines
2.8 KiB
C#
78 lines
2.8 KiB
C#
using MareSynchronosShared.Metrics;
|
|
using System.Collections.Concurrent;
|
|
|
|
namespace MareSynchronosStaticFilesServer;
|
|
|
|
public class FileStatisticsService : IHostedService
|
|
{
|
|
private readonly MareMetrics _metrics;
|
|
private readonly ILogger<FileStatisticsService> _logger;
|
|
private CancellationTokenSource _resetCancellationTokenSource;
|
|
private ConcurrentDictionary<string, long> _pastHourFiles = new(StringComparer.Ordinal);
|
|
private ConcurrentDictionary<string, long> _pastDayFiles = new(StringComparer.Ordinal);
|
|
|
|
public FileStatisticsService(MareMetrics metrics, ILogger<FileStatisticsService> logger)
|
|
{
|
|
_metrics = metrics;
|
|
_logger = logger;
|
|
}
|
|
|
|
public void LogFile(string fileHash, long length)
|
|
{
|
|
if (!_pastHourFiles.ContainsKey(fileHash))
|
|
{
|
|
_pastHourFiles[fileHash] = length;
|
|
_metrics.IncGauge(MetricsAPI.GaugeFilesUniquePastHour);
|
|
_metrics.IncGauge(MetricsAPI.GaugeFilesUniquePastHourSize, length);
|
|
}
|
|
if (!_pastDayFiles.ContainsKey(fileHash))
|
|
{
|
|
_pastDayFiles[fileHash] = length;
|
|
_metrics.IncGauge(MetricsAPI.GaugeFilesUniquePastDay);
|
|
_metrics.IncGauge(MetricsAPI.GaugeFilesUniquePastDaySize, length);
|
|
}
|
|
}
|
|
|
|
public Task StartAsync(CancellationToken cancellationToken)
|
|
{
|
|
_logger.LogInformation("Starting FileStatisticsService");
|
|
_resetCancellationTokenSource = new();
|
|
_ = ResetHourlyFileData();
|
|
_ = ResetDailyFileData();
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public async Task ResetHourlyFileData()
|
|
{
|
|
while (!_resetCancellationTokenSource.Token.IsCancellationRequested)
|
|
{
|
|
_logger.LogInformation("Resetting 1h Data");
|
|
|
|
_pastHourFiles = new(StringComparer.Ordinal);
|
|
_metrics.SetGaugeTo(MetricsAPI.GaugeFilesUniquePastHour, 0);
|
|
_metrics.SetGaugeTo(MetricsAPI.GaugeFilesUniquePastHourSize, 0);
|
|
await Task.Delay(TimeSpan.FromHours(1), _resetCancellationTokenSource.Token).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
public async Task ResetDailyFileData()
|
|
{
|
|
while (!_resetCancellationTokenSource.Token.IsCancellationRequested)
|
|
{
|
|
_logger.LogInformation("Resetting 24h Data");
|
|
|
|
_pastDayFiles = new(StringComparer.Ordinal);
|
|
_metrics.SetGaugeTo(MetricsAPI.GaugeFilesUniquePastDay, 0);
|
|
_metrics.SetGaugeTo(MetricsAPI.GaugeFilesUniquePastDaySize, 0);
|
|
await Task.Delay(TimeSpan.FromDays(1), _resetCancellationTokenSource.Token).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
public Task StopAsync(CancellationToken cancellationToken)
|
|
{
|
|
_resetCancellationTokenSource.Cancel();
|
|
_logger.LogInformation("Stopping FileStatisticsService");
|
|
return Task.CompletedTask;
|
|
}
|
|
}
|