92 lines
2.9 KiB
C#
92 lines
2.9 KiB
C#
using System;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using MareSynchronosServer.Data;
|
|
using Microsoft.Extensions.Configuration;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Hosting;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace MareSynchronosServer
|
|
{
|
|
public class FileCleanupService : IHostedService, IDisposable
|
|
{
|
|
private readonly ILogger<FileCleanupService> _logger;
|
|
private readonly IServiceProvider _services;
|
|
private readonly IConfiguration _configuration;
|
|
private Timer _timer;
|
|
|
|
public FileCleanupService(ILogger<FileCleanupService> logger, IServiceProvider services, IConfiguration configuration)
|
|
{
|
|
_logger = logger;
|
|
_services = services;
|
|
_configuration = configuration;
|
|
}
|
|
|
|
public Task StartAsync(CancellationToken cancellationToken)
|
|
{
|
|
_logger.LogInformation("File Cleanup Service started");
|
|
|
|
_timer = new Timer(CleanUpFiles, null, TimeSpan.Zero, TimeSpan.FromMinutes(10));
|
|
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
private void CleanUpFiles(object state)
|
|
{
|
|
if (!int.TryParse(_configuration["UnusedFileRetentionPeriodInDays"], out var filesOlderThanDays))
|
|
{
|
|
filesOlderThanDays = 7;
|
|
}
|
|
|
|
_logger.LogInformation($"Cleaning up files older than {filesOlderThanDays} days");
|
|
|
|
using var scope = _services.CreateScope();
|
|
var dbContext = scope.ServiceProvider.GetService<MareDbContext>();
|
|
|
|
var prevTime = DateTime.Now.Subtract(TimeSpan.FromDays(filesOlderThanDays));
|
|
var filesToDelete =
|
|
dbContext.Files.Where(f => f.LastAccessTime < prevTime);
|
|
dbContext.RemoveRange(filesToDelete);
|
|
foreach (var file in filesToDelete)
|
|
{
|
|
var fileName = Path.Combine(_configuration["CacheDirectory"], file.Hash);
|
|
if (File.Exists(fileName))
|
|
{
|
|
_logger.LogInformation("Deleting: " + fileName);
|
|
File.Delete(fileName);
|
|
}
|
|
}
|
|
|
|
dbContext.SaveChanges();
|
|
|
|
var allFiles = dbContext.Files;
|
|
foreach (var file in allFiles)
|
|
{
|
|
var fileName = Path.Combine(_configuration["CacheDirectory"], file.Hash);
|
|
if (!File.Exists(fileName))
|
|
{
|
|
_logger.LogInformation("File does not exist anymore: " + fileName);
|
|
dbContext.Files.Remove(file);
|
|
}
|
|
}
|
|
|
|
dbContext.SaveChanges();
|
|
}
|
|
|
|
public Task StopAsync(CancellationToken cancellationToken)
|
|
{
|
|
_timer?.Change(Timeout.Infinite, 0);
|
|
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
_timer?.Dispose();
|
|
}
|
|
}
|
|
}
|