refactor file server
This commit is contained in:
2
MareAPI
2
MareAPI
Submodule MareAPI updated: cd8934a4ab...b529a101ae
@@ -5,6 +5,7 @@ namespace MareSynchronosStaticFilesServer;
|
|||||||
|
|
||||||
public class StaticFilesServerConfiguration : MareConfigurationBase
|
public class StaticFilesServerConfiguration : MareConfigurationBase
|
||||||
{
|
{
|
||||||
|
public bool IsDistributionNode { get; set; } = false;
|
||||||
public Uri? MainFileServerAddress { get; set; } = null;
|
public Uri? MainFileServerAddress { get; set; } = null;
|
||||||
public int ForcedDeletionOfFilesAfterHours { get; set; } = -1;
|
public int ForcedDeletionOfFilesAfterHours { get; set; } = -1;
|
||||||
public double CacheSizeHardLimitInGiB { get; set; } = -1;
|
public double CacheSizeHardLimitInGiB { get; set; } = -1;
|
||||||
|
|||||||
@@ -0,0 +1,29 @@
|
|||||||
|
using MareSynchronos.API.Routes;
|
||||||
|
using MareSynchronosStaticFilesServer.Services;
|
||||||
|
using Microsoft.AspNetCore.Authorization;
|
||||||
|
using Microsoft.AspNetCore.Mvc;
|
||||||
|
|
||||||
|
namespace MareSynchronosStaticFilesServer.Controllers;
|
||||||
|
|
||||||
|
[Route(MareFiles.Distribution)]
|
||||||
|
public class DistributionController : ControllerBase
|
||||||
|
{
|
||||||
|
private readonly CachedFileProvider _cachedFileProvider;
|
||||||
|
|
||||||
|
public DistributionController(ILogger<DistributionController> logger, CachedFileProvider cachedFileProvider) : base(logger)
|
||||||
|
{
|
||||||
|
_cachedFileProvider = cachedFileProvider;
|
||||||
|
}
|
||||||
|
|
||||||
|
[HttpGet(MareFiles.Distribution_Get)]
|
||||||
|
[Authorize(Policy = "Internal")]
|
||||||
|
public async Task<IActionResult> GetFile(string file)
|
||||||
|
{
|
||||||
|
_logger.LogInformation($"GetFile:{MareUser}:{file}");
|
||||||
|
|
||||||
|
var fs = await _cachedFileProvider.GetAndDownloadFileStream(file);
|
||||||
|
if (fs == null) return NotFound();
|
||||||
|
|
||||||
|
return File(fs, "application/octet-stream");
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,25 @@
|
|||||||
|
using MareSynchronos.API.Routes;
|
||||||
|
using MareSynchronosStaticFilesServer.Services;
|
||||||
|
using Microsoft.AspNetCore.Authorization;
|
||||||
|
using Microsoft.AspNetCore.Mvc;
|
||||||
|
|
||||||
|
namespace MareSynchronosStaticFilesServer.Controllers;
|
||||||
|
|
||||||
|
[Route(MareFiles.Main)]
|
||||||
|
public class MainController : ControllerBase
|
||||||
|
{
|
||||||
|
private readonly IClientReadyMessageService _messageService;
|
||||||
|
|
||||||
|
public MainController(ILogger<MainController> logger, IClientReadyMessageService mareHub) : base(logger)
|
||||||
|
{
|
||||||
|
_messageService = mareHub;
|
||||||
|
}
|
||||||
|
|
||||||
|
[HttpGet(MareFiles.Main_SendReady)]
|
||||||
|
[Authorize(Policy = "Internal")]
|
||||||
|
public IActionResult SendReadyToClients(string uid, Guid requestId)
|
||||||
|
{
|
||||||
|
_messageService.SendDownloadReady(uid, requestId);
|
||||||
|
return Ok();
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -10,7 +10,6 @@ using MareSynchronosShared.Services;
|
|||||||
using MareSynchronosShared.Utils;
|
using MareSynchronosShared.Utils;
|
||||||
using MareSynchronosStaticFilesServer.Services;
|
using MareSynchronosStaticFilesServer.Services;
|
||||||
using MareSynchronosStaticFilesServer.Utils;
|
using MareSynchronosStaticFilesServer.Utils;
|
||||||
using Microsoft.AspNetCore.Authorization;
|
|
||||||
using Microsoft.AspNetCore.Mvc;
|
using Microsoft.AspNetCore.Mvc;
|
||||||
using Microsoft.AspNetCore.SignalR;
|
using Microsoft.AspNetCore.SignalR;
|
||||||
using Microsoft.EntityFrameworkCore;
|
using Microsoft.EntityFrameworkCore;
|
||||||
@@ -165,18 +164,6 @@ public class ServerFilesController : ControllerBase
|
|||||||
return Ok(JsonSerializer.Serialize(notCoveredFiles.Values.ToList()));
|
return Ok(JsonSerializer.Serialize(notCoveredFiles.Values.ToList()));
|
||||||
}
|
}
|
||||||
|
|
||||||
[HttpGet(MareFiles.ServerFiles_Get + "/{fileId}")]
|
|
||||||
[Authorize(Policy = "Internal")]
|
|
||||||
public IActionResult GetFile(string fileId)
|
|
||||||
{
|
|
||||||
_logger.LogInformation($"GetFile:{MareUser}:{fileId}");
|
|
||||||
|
|
||||||
var fs = _cachedFileProvider.GetLocalFileStream(fileId);
|
|
||||||
if (fs == null) return NotFound();
|
|
||||||
|
|
||||||
return File(fs, "application/octet-stream");
|
|
||||||
}
|
|
||||||
|
|
||||||
[HttpPost(MareFiles.ServerFiles_Upload + "/{hash}")]
|
[HttpPost(MareFiles.ServerFiles_Upload + "/{hash}")]
|
||||||
[RequestSizeLimit(200 * 1024 * 1024)]
|
[RequestSizeLimit(200 * 1024 * 1024)]
|
||||||
public async Task<IActionResult> UploadFile(string hash, CancellationToken requestAborted)
|
public async Task<IActionResult> UploadFile(string hash, CancellationToken requestAborted)
|
||||||
|
|||||||
@@ -21,7 +21,8 @@ public sealed class CachedFileProvider : IDisposable
|
|||||||
private readonly SemaphoreSlim _downloadSemaphore = new(1);
|
private readonly SemaphoreSlim _downloadSemaphore = new(1);
|
||||||
private bool _disposed;
|
private bool _disposed;
|
||||||
|
|
||||||
private bool IsMainServer => _remoteCacheSourceUri == null;
|
private bool IsMainServer => _remoteCacheSourceUri == null && _isDistributionServer;
|
||||||
|
private bool _isDistributionServer;
|
||||||
|
|
||||||
public CachedFileProvider(IConfigurationService<StaticFilesServerConfiguration> configuration, ILogger<CachedFileProvider> logger, FileStatisticsService fileStatisticsService, MareMetrics metrics, ServerTokenGenerator generator)
|
public CachedFileProvider(IConfigurationService<StaticFilesServerConfiguration> configuration, ILogger<CachedFileProvider> logger, FileStatisticsService fileStatisticsService, MareMetrics metrics, ServerTokenGenerator generator)
|
||||||
{
|
{
|
||||||
@@ -30,8 +31,10 @@ public sealed class CachedFileProvider : IDisposable
|
|||||||
_metrics = metrics;
|
_metrics = metrics;
|
||||||
_generator = generator;
|
_generator = generator;
|
||||||
_remoteCacheSourceUri = configuration.GetValueOrDefault<Uri>(nameof(StaticFilesServerConfiguration.MainFileServerAddress), null);
|
_remoteCacheSourceUri = configuration.GetValueOrDefault<Uri>(nameof(StaticFilesServerConfiguration.MainFileServerAddress), null);
|
||||||
|
_isDistributionServer = configuration.GetValueOrDefault(nameof(StaticFilesServerConfiguration.IsDistributionNode), false);
|
||||||
_basePath = configuration.GetValue<string>(nameof(StaticFilesServerConfiguration.CacheDirectory));
|
_basePath = configuration.GetValue<string>(nameof(StaticFilesServerConfiguration.CacheDirectory));
|
||||||
_httpClient = new();
|
_httpClient = new();
|
||||||
|
_httpClient.DefaultRequestHeaders.UserAgent.Add(new ProductInfoHeaderValue("MareSynchronosServer"));
|
||||||
}
|
}
|
||||||
|
|
||||||
public void Dispose()
|
public void Dispose()
|
||||||
@@ -48,7 +51,7 @@ public sealed class CachedFileProvider : IDisposable
|
|||||||
private async Task DownloadTask(string hash)
|
private async Task DownloadTask(string hash)
|
||||||
{
|
{
|
||||||
// download file from remote
|
// download file from remote
|
||||||
var downloadUrl = MareFiles.ServerFilesGetFullPath(_remoteCacheSourceUri, hash);
|
var downloadUrl = MareFiles.DistributionGetFullPath(_remoteCacheSourceUri, hash);
|
||||||
_logger.LogInformation("Did not find {hash}, downloading from {server}", hash, downloadUrl);
|
_logger.LogInformation("Did not find {hash}, downloading from {server}", hash, downloadUrl);
|
||||||
|
|
||||||
using var requestMessage = new HttpRequestMessage(HttpMethod.Get, downloadUrl);
|
using var requestMessage = new HttpRequestMessage(HttpMethod.Get, downloadUrl);
|
||||||
@@ -129,12 +132,4 @@ public sealed class CachedFileProvider : IDisposable
|
|||||||
|
|
||||||
return GetLocalFileStream(hash);
|
return GetLocalFileStream(hash);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void ThrowIfDisposed()
|
|
||||||
{
|
|
||||||
if (_disposed)
|
|
||||||
{
|
|
||||||
throw new ObjectDisposedException(GetType().FullName);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,6 @@
|
|||||||
|
namespace MareSynchronosStaticFilesServer.Services;
|
||||||
|
|
||||||
|
public interface IClientReadyMessageService
|
||||||
|
{
|
||||||
|
void SendDownloadReady(string uid, Guid requestId);
|
||||||
|
}
|
||||||
@@ -0,0 +1,26 @@
|
|||||||
|
using Microsoft.AspNetCore.SignalR;
|
||||||
|
using MareSynchronos.API.SignalR;
|
||||||
|
using MareSynchronosServer.Hubs;
|
||||||
|
|
||||||
|
namespace MareSynchronosStaticFilesServer.Services;
|
||||||
|
|
||||||
|
public class MainClientReadyMessageService : IClientReadyMessageService
|
||||||
|
{
|
||||||
|
private readonly ILogger<MainClientReadyMessageService> _logger;
|
||||||
|
private readonly IHubContext<MareHub> _mareHub;
|
||||||
|
|
||||||
|
public MainClientReadyMessageService(ILogger<MainClientReadyMessageService> logger, IHubContext<MareHub> mareHub)
|
||||||
|
{
|
||||||
|
_logger = logger;
|
||||||
|
_mareHub = mareHub;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void SendDownloadReady(string uid, Guid requestId)
|
||||||
|
{
|
||||||
|
_ = Task.Run(async () =>
|
||||||
|
{
|
||||||
|
_logger.LogInformation("Sending Client Ready for {uid}:{requestId} to SignalR", uid, requestId);
|
||||||
|
await _mareHub.Clients.User(uid).SendAsync(nameof(IMareHub.Client_DownloadReady), requestId).ConfigureAwait(false);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -8,23 +8,21 @@ using Microsoft.EntityFrameworkCore;
|
|||||||
|
|
||||||
namespace MareSynchronosStaticFilesServer.Services;
|
namespace MareSynchronosStaticFilesServer.Services;
|
||||||
|
|
||||||
public class FileCleanupService : IHostedService
|
public class MainFileCleanupService : IHostedService
|
||||||
{
|
{
|
||||||
private readonly string _cacheDir;
|
private readonly string _cacheDir;
|
||||||
private readonly IConfigurationService<StaticFilesServerConfiguration> _configuration;
|
private readonly IConfigurationService<StaticFilesServerConfiguration> _configuration;
|
||||||
private readonly bool _isMainServer;
|
private readonly ILogger<MainFileCleanupService> _logger;
|
||||||
private readonly ILogger<FileCleanupService> _logger;
|
|
||||||
private readonly MareMetrics _metrics;
|
private readonly MareMetrics _metrics;
|
||||||
private readonly IServiceProvider _services;
|
private readonly IServiceProvider _services;
|
||||||
private CancellationTokenSource _cleanupCts;
|
private CancellationTokenSource _cleanupCts;
|
||||||
|
|
||||||
public FileCleanupService(MareMetrics metrics, ILogger<FileCleanupService> logger, IServiceProvider services, IConfigurationService<StaticFilesServerConfiguration> configuration)
|
public MainFileCleanupService(MareMetrics metrics, ILogger<MainFileCleanupService> logger, IServiceProvider services, IConfigurationService<StaticFilesServerConfiguration> configuration)
|
||||||
{
|
{
|
||||||
_metrics = metrics;
|
_metrics = metrics;
|
||||||
_logger = logger;
|
_logger = logger;
|
||||||
_services = services;
|
_services = services;
|
||||||
_configuration = configuration;
|
_configuration = configuration;
|
||||||
_isMainServer = configuration.IsMain;
|
|
||||||
_cacheDir = _configuration.GetValue<string>(nameof(StaticFilesServerConfiguration.CacheDirectory));
|
_cacheDir = _configuration.GetValue<string>(nameof(StaticFilesServerConfiguration.CacheDirectory));
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -48,13 +46,10 @@ public class FileCleanupService : IHostedService
|
|||||||
|
|
||||||
CleanUpFilesBeyondSizeLimit(dbContext, ct);
|
CleanUpFilesBeyondSizeLimit(dbContext, ct);
|
||||||
|
|
||||||
if (_isMainServer)
|
|
||||||
{
|
|
||||||
CleanUpStuckUploads(dbContext);
|
CleanUpStuckUploads(dbContext);
|
||||||
|
|
||||||
await dbContext.SaveChangesAsync(ct).ConfigureAwait(false);
|
await dbContext.SaveChangesAsync(ct).ConfigureAwait(false);
|
||||||
}
|
}
|
||||||
}
|
|
||||||
catch (Exception e)
|
catch (Exception e)
|
||||||
{
|
{
|
||||||
_logger.LogError(e, "Error during cleanup task");
|
_logger.LogError(e, "Error during cleanup task");
|
||||||
@@ -113,13 +108,10 @@ public class FileCleanupService : IHostedService
|
|||||||
_metrics.DecGauge(MetricsAPI.GaugeFilesTotal);
|
_metrics.DecGauge(MetricsAPI.GaugeFilesTotal);
|
||||||
_logger.LogInformation("Deleting {oldestFile} with size {size}MiB", oldestFile.FullName, ByteSize.FromBytes(oldestFile.Length).MebiBytes);
|
_logger.LogInformation("Deleting {oldestFile} with size {size}MiB", oldestFile.FullName, ByteSize.FromBytes(oldestFile.Length).MebiBytes);
|
||||||
oldestFile.Delete();
|
oldestFile.Delete();
|
||||||
if (_isMainServer)
|
|
||||||
{
|
|
||||||
FileCache f = new() { Hash = oldestFile.Name.ToUpperInvariant() };
|
FileCache f = new() { Hash = oldestFile.Name.ToUpperInvariant() };
|
||||||
dbContext.Entry(f).State = EntityState.Deleted;
|
dbContext.Entry(f).State = EntityState.Deleted;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
catch (Exception ex)
|
catch (Exception ex)
|
||||||
{
|
{
|
||||||
_logger.LogWarning(ex, "Error during cache size limit cleanup");
|
_logger.LogWarning(ex, "Error during cache size limit cleanup");
|
||||||
@@ -127,8 +119,6 @@ public class FileCleanupService : IHostedService
|
|||||||
}
|
}
|
||||||
|
|
||||||
private void CleanUpOrphanedFiles(List<FileCache> allFiles, FileInfo[] allPhysicalFiles, CancellationToken ct)
|
private void CleanUpOrphanedFiles(List<FileCache> allFiles, FileInfo[] allPhysicalFiles, CancellationToken ct)
|
||||||
{
|
|
||||||
if (_isMainServer)
|
|
||||||
{
|
{
|
||||||
var allFilesHashes = new HashSet<string>(allFiles.Select(a => a.Hash.ToUpperInvariant()), StringComparer.Ordinal);
|
var allFilesHashes = new HashSet<string>(allFiles.Select(a => a.Hash.ToUpperInvariant()), StringComparer.Ordinal);
|
||||||
foreach (var file in allPhysicalFiles)
|
foreach (var file in allPhysicalFiles)
|
||||||
@@ -144,7 +134,6 @@ public class FileCleanupService : IHostedService
|
|||||||
ct.ThrowIfCancellationRequested();
|
ct.ThrowIfCancellationRequested();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
private async Task CleanUpOutdatedFiles(MareDbContext dbContext, CancellationToken ct)
|
private async Task CleanUpOutdatedFiles(MareDbContext dbContext, CancellationToken ct)
|
||||||
{
|
{
|
||||||
@@ -164,13 +153,15 @@ public class FileCleanupService : IHostedService
|
|||||||
var prevTimeForcedDeletion = DateTime.Now.Subtract(TimeSpan.FromHours(forcedDeletionAfterHours));
|
var prevTimeForcedDeletion = DateTime.Now.Subtract(TimeSpan.FromHours(forcedDeletionAfterHours));
|
||||||
DirectoryInfo dir = new(_cacheDir);
|
DirectoryInfo dir = new(_cacheDir);
|
||||||
var allFilesInDir = dir.GetFiles("*", SearchOption.AllDirectories);
|
var allFilesInDir = dir.GetFiles("*", SearchOption.AllDirectories);
|
||||||
var allFiles = await dbContext.Files.ToListAsync().ConfigureAwait(false);
|
var files = dbContext.Files.OrderBy(f => f.Hash);
|
||||||
|
List<FileCache> allFiles = await dbContext.Files.ToListAsync(ct).ConfigureAwait(false);
|
||||||
int fileCounter = 0;
|
int fileCounter = 0;
|
||||||
foreach (var fileCache in allFiles.Where(f => f.Uploaded))
|
foreach (var fileCache in allFiles.Where(f => f.Uploaded))
|
||||||
{
|
{
|
||||||
var file = FilePathUtil.GetFileInfoForHash(_cacheDir, fileCache.Hash);
|
|
||||||
bool fileDeleted = false;
|
bool fileDeleted = false;
|
||||||
if (file == null && _isMainServer)
|
|
||||||
|
var file = FilePathUtil.GetFileInfoForHash(_cacheDir, fileCache.Hash);
|
||||||
|
if (file == null)
|
||||||
{
|
{
|
||||||
_logger.LogInformation("File does not exist anymore: {fileName}", fileCache.Hash);
|
_logger.LogInformation("File does not exist anymore: {fileName}", fileCache.Hash);
|
||||||
dbContext.Files.Remove(fileCache);
|
dbContext.Files.Remove(fileCache);
|
||||||
@@ -182,26 +173,20 @@ public class FileCleanupService : IHostedService
|
|||||||
_metrics.DecGauge(MetricsAPI.GaugeFilesTotal);
|
_metrics.DecGauge(MetricsAPI.GaugeFilesTotal);
|
||||||
_logger.LogInformation("File outdated: {fileName}, {fileSize}MiB", file.Name, ByteSize.FromBytes(file.Length).MebiBytes);
|
_logger.LogInformation("File outdated: {fileName}, {fileSize}MiB", file.Name, ByteSize.FromBytes(file.Length).MebiBytes);
|
||||||
file.Delete();
|
file.Delete();
|
||||||
if (_isMainServer)
|
|
||||||
{
|
|
||||||
fileDeleted = true;
|
fileDeleted = true;
|
||||||
dbContext.Files.Remove(fileCache);
|
dbContext.Files.Remove(fileCache);
|
||||||
}
|
}
|
||||||
}
|
|
||||||
else if (file != null && forcedDeletionAfterHours > 0 && file.LastWriteTime < prevTimeForcedDeletion)
|
else if (file != null && forcedDeletionAfterHours > 0 && file.LastWriteTime < prevTimeForcedDeletion)
|
||||||
{
|
{
|
||||||
_metrics.DecGauge(MetricsAPI.GaugeFilesTotalSize, file.Length);
|
_metrics.DecGauge(MetricsAPI.GaugeFilesTotalSize, file.Length);
|
||||||
_metrics.DecGauge(MetricsAPI.GaugeFilesTotal);
|
_metrics.DecGauge(MetricsAPI.GaugeFilesTotal);
|
||||||
_logger.LogInformation("File forcefully deleted: {fileName}, {fileSize}MiB", file.Name, ByteSize.FromBytes(file.Length).MebiBytes);
|
_logger.LogInformation("File forcefully deleted: {fileName}, {fileSize}MiB", file.Name, ByteSize.FromBytes(file.Length).MebiBytes);
|
||||||
file.Delete();
|
file.Delete();
|
||||||
if (_isMainServer)
|
|
||||||
{
|
|
||||||
fileDeleted = true;
|
fileDeleted = true;
|
||||||
dbContext.Files.Remove(fileCache);
|
dbContext.Files.Remove(fileCache);
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
if (_isMainServer && !fileDeleted && file != null && fileCache.Size == 0)
|
if (!fileDeleted && file != null && fileCache.Size == 0)
|
||||||
{
|
{
|
||||||
_logger.LogInformation("Setting File Size of " + fileCache.Hash + " to " + file.Length);
|
_logger.LogInformation("Setting File Size of " + fileCache.Hash + " to " + file.Length);
|
||||||
fileCache.Size = file.Length;
|
fileCache.Size = file.Length;
|
||||||
@@ -1,32 +1,27 @@
|
|||||||
using MareSynchronosShared.Metrics;
|
using MareSynchronosShared.Metrics;
|
||||||
using MareSynchronosShared.Services;
|
using MareSynchronosShared.Services;
|
||||||
using MareSynchronosStaticFilesServer.Utils;
|
using MareSynchronosStaticFilesServer.Utils;
|
||||||
using Microsoft.AspNetCore.SignalR;
|
|
||||||
using System.Collections.Concurrent;
|
using System.Collections.Concurrent;
|
||||||
using System.Timers;
|
using System.Timers;
|
||||||
using MareSynchronos.API.SignalR;
|
|
||||||
|
|
||||||
namespace MareSynchronosStaticFilesServer.Services;
|
namespace MareSynchronosStaticFilesServer.Services;
|
||||||
|
|
||||||
public class RequestQueueService : IHostedService
|
public class RequestQueueService : IHostedService
|
||||||
{
|
{
|
||||||
private record PriorityEntry(bool IsHighPriority, DateTime LastChecked);
|
private readonly IClientReadyMessageService _clientReadyMessageService;
|
||||||
|
|
||||||
private readonly IHubContext<MareSynchronosServer.Hubs.MareHub> _hubContext;
|
|
||||||
private readonly ILogger<RequestQueueService> _logger;
|
private readonly ILogger<RequestQueueService> _logger;
|
||||||
private readonly MareMetrics _metrics;
|
private readonly MareMetrics _metrics;
|
||||||
private readonly ConcurrentQueue<UserRequest> _queue = new();
|
private readonly ConcurrentQueue<UserRequest> _queue = new();
|
||||||
private readonly ConcurrentQueue<UserRequest> _priorityQueue = new();
|
private readonly ConcurrentQueue<UserRequest> _priorityQueue = new();
|
||||||
private readonly int _queueExpirationSeconds;
|
private readonly int _queueExpirationSeconds;
|
||||||
private readonly SemaphoreSlim _queueProcessingSemaphore = new(1);
|
private readonly SemaphoreSlim _queueProcessingSemaphore = new(1);
|
||||||
private readonly SemaphoreSlim _queueSemaphore = new(1);
|
|
||||||
private readonly UserQueueEntry[] _userQueueRequests;
|
private readonly UserQueueEntry[] _userQueueRequests;
|
||||||
private int _queueLimitForReset;
|
private int _queueLimitForReset;
|
||||||
private readonly int _queueReleaseSeconds;
|
private readonly int _queueReleaseSeconds;
|
||||||
private System.Timers.Timer _queueTimer;
|
private System.Timers.Timer _queueTimer;
|
||||||
|
|
||||||
public RequestQueueService(MareMetrics metrics, IConfigurationService<StaticFilesServerConfiguration> configurationService,
|
public RequestQueueService(MareMetrics metrics, IConfigurationService<StaticFilesServerConfiguration> configurationService,
|
||||||
ILogger<RequestQueueService> logger, IHubContext<MareSynchronosServer.Hubs.MareHub> hubContext)
|
ILogger<RequestQueueService> logger, IClientReadyMessageService hubContext)
|
||||||
{
|
{
|
||||||
_userQueueRequests = new UserQueueEntry[configurationService.GetValueOrDefault(nameof(StaticFilesServerConfiguration.DownloadQueueSize), 50)];
|
_userQueueRequests = new UserQueueEntry[configurationService.GetValueOrDefault(nameof(StaticFilesServerConfiguration.DownloadQueueSize), 50)];
|
||||||
_queueExpirationSeconds = configurationService.GetValueOrDefault(nameof(StaticFilesServerConfiguration.DownloadTimeoutSeconds), 5);
|
_queueExpirationSeconds = configurationService.GetValueOrDefault(nameof(StaticFilesServerConfiguration.DownloadTimeoutSeconds), 5);
|
||||||
@@ -34,7 +29,7 @@ public class RequestQueueService : IHostedService
|
|||||||
_queueReleaseSeconds = configurationService.GetValueOrDefault(nameof(StaticFilesServerConfiguration.DownloadQueueReleaseSeconds), 15);
|
_queueReleaseSeconds = configurationService.GetValueOrDefault(nameof(StaticFilesServerConfiguration.DownloadQueueReleaseSeconds), 15);
|
||||||
_metrics = metrics;
|
_metrics = metrics;
|
||||||
_logger = logger;
|
_logger = logger;
|
||||||
_hubContext = hubContext;
|
_clientReadyMessageService = hubContext;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void ActivateRequest(Guid request)
|
public void ActivateRequest(Guid request)
|
||||||
@@ -125,7 +120,7 @@ public class RequestQueueService : IHostedService
|
|||||||
{
|
{
|
||||||
_logger.LogDebug("Dequeueing {req} into {i}: {user} with {file}", userRequest.RequestId, slot, userRequest.User, string.Join(", ", userRequest.FileIds));
|
_logger.LogDebug("Dequeueing {req} into {i}: {user} with {file}", userRequest.RequestId, slot, userRequest.User, string.Join(", ", userRequest.FileIds));
|
||||||
_userQueueRequests[slot] = new(userRequest, DateTime.UtcNow.AddSeconds(_queueExpirationSeconds));
|
_userQueueRequests[slot] = new(userRequest, DateTime.UtcNow.AddSeconds(_queueExpirationSeconds));
|
||||||
_ = _hubContext.Clients.User(userRequest.User).SendAsync(nameof(IMareHub.Client_DownloadReady), userRequest.RequestId).ConfigureAwait(false);
|
_clientReadyMessageService.SendDownloadReady(userRequest.User, userRequest.RequestId);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void ProcessQueue(object src, ElapsedEventArgs e)
|
private void ProcessQueue(object src, ElapsedEventArgs e)
|
||||||
|
|||||||
@@ -0,0 +1,47 @@
|
|||||||
|
using MareSynchronos.API.Routes;
|
||||||
|
using MareSynchronosShared.Services;
|
||||||
|
using MareSynchronosShared.Utils;
|
||||||
|
using System.Net.Http.Headers;
|
||||||
|
|
||||||
|
namespace MareSynchronosStaticFilesServer.Services;
|
||||||
|
|
||||||
|
public class ShardClientReadyMessageService : IClientReadyMessageService
|
||||||
|
{
|
||||||
|
private readonly ILogger<ShardClientReadyMessageService> _logger;
|
||||||
|
private readonly ServerTokenGenerator _tokenGenerator;
|
||||||
|
private readonly IConfigurationService<StaticFilesServerConfiguration> _configurationService;
|
||||||
|
private readonly HttpClient _httpClient;
|
||||||
|
|
||||||
|
public ShardClientReadyMessageService(ILogger<ShardClientReadyMessageService> logger, ServerTokenGenerator tokenGenerator, IConfigurationService<StaticFilesServerConfiguration> configurationService)
|
||||||
|
{
|
||||||
|
_logger = logger;
|
||||||
|
_tokenGenerator = tokenGenerator;
|
||||||
|
_configurationService = configurationService;
|
||||||
|
_httpClient = new();
|
||||||
|
_httpClient.DefaultRequestHeaders.UserAgent.Add(new ProductInfoHeaderValue("MareSynchronosServer"));
|
||||||
|
}
|
||||||
|
|
||||||
|
public void SendDownloadReady(string uid, Guid requestId)
|
||||||
|
{
|
||||||
|
var mainUrl = _configurationService.GetValue<Uri>(nameof(StaticFilesServerConfiguration.MainFileServerAddress));
|
||||||
|
var path = MareFiles.MainSendReadyFullPath(mainUrl, uid, requestId);
|
||||||
|
using HttpRequestMessage msg = new()
|
||||||
|
{
|
||||||
|
RequestUri = path
|
||||||
|
};
|
||||||
|
msg.Headers.Authorization = new AuthenticationHeaderValue("Bearer", _tokenGenerator.Token);
|
||||||
|
|
||||||
|
_ = Task.Run(async () =>
|
||||||
|
{
|
||||||
|
_logger.LogInformation("Sending Client Ready for {uid}:{requestId} to {path}", uid, requestId, path);
|
||||||
|
try
|
||||||
|
{
|
||||||
|
using var result = await _httpClient.SendAsync(msg).ConfigureAwait(false);
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
_logger.LogError(ex, "Failure to send for {uid}:{requestId}", uid, requestId);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,153 @@
|
|||||||
|
using ByteSizeLib;
|
||||||
|
using MareSynchronosShared.Metrics;
|
||||||
|
using MareSynchronosShared.Services;
|
||||||
|
using Microsoft.EntityFrameworkCore;
|
||||||
|
|
||||||
|
namespace MareSynchronosStaticFilesServer.Services;
|
||||||
|
|
||||||
|
public class ShardFileCleanupService : IHostedService
|
||||||
|
{
|
||||||
|
private readonly string _cacheDir;
|
||||||
|
private readonly IConfigurationService<StaticFilesServerConfiguration> _configuration;
|
||||||
|
private readonly ILogger<MainFileCleanupService> _logger;
|
||||||
|
private readonly MareMetrics _metrics;
|
||||||
|
private CancellationTokenSource _cleanupCts;
|
||||||
|
|
||||||
|
public ShardFileCleanupService(MareMetrics metrics, ILogger<MainFileCleanupService> logger, IConfigurationService<StaticFilesServerConfiguration> configuration)
|
||||||
|
{
|
||||||
|
_metrics = metrics;
|
||||||
|
_logger = logger;
|
||||||
|
_configuration = configuration;
|
||||||
|
_cacheDir = _configuration.GetValue<string>(nameof(StaticFilesServerConfiguration.CacheDirectory));
|
||||||
|
}
|
||||||
|
|
||||||
|
public async Task CleanUpTask(CancellationToken ct)
|
||||||
|
{
|
||||||
|
_logger.LogInformation("Starting periodic cleanup task");
|
||||||
|
|
||||||
|
while (!ct.IsCancellationRequested)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
DirectoryInfo dir = new(_cacheDir);
|
||||||
|
var allFiles = dir.GetFiles("*", SearchOption.AllDirectories);
|
||||||
|
_metrics.SetGaugeTo(MetricsAPI.GaugeFilesTotalSize, allFiles.Sum(f => f.Length));
|
||||||
|
_metrics.SetGaugeTo(MetricsAPI.GaugeFilesTotal, allFiles.Length);
|
||||||
|
|
||||||
|
CleanUpOutdatedFiles(ct);
|
||||||
|
|
||||||
|
CleanUpFilesBeyondSizeLimit(ct);
|
||||||
|
}
|
||||||
|
catch (Exception e)
|
||||||
|
{
|
||||||
|
_logger.LogError(e, "Error during cleanup task");
|
||||||
|
}
|
||||||
|
|
||||||
|
var now = DateTime.Now;
|
||||||
|
TimeOnly currentTime = new(now.Hour, now.Minute, now.Second);
|
||||||
|
TimeOnly futureTime = new(now.Hour, now.Minute - now.Minute % 15, 0);
|
||||||
|
var span = futureTime.AddMinutes(15) - currentTime;
|
||||||
|
|
||||||
|
_logger.LogInformation("File Cleanup Complete, next run at {date}", now.Add(span));
|
||||||
|
await Task.Delay(span, ct).ConfigureAwait(false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Task StartAsync(CancellationToken cancellationToken)
|
||||||
|
{
|
||||||
|
_logger.LogInformation("Cleanup Service started");
|
||||||
|
|
||||||
|
_cleanupCts = new();
|
||||||
|
|
||||||
|
_ = CleanUpTask(_cleanupCts.Token);
|
||||||
|
|
||||||
|
return Task.CompletedTask;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Task StopAsync(CancellationToken cancellationToken)
|
||||||
|
{
|
||||||
|
_cleanupCts.Cancel();
|
||||||
|
|
||||||
|
return Task.CompletedTask;
|
||||||
|
}
|
||||||
|
|
||||||
|
private void CleanUpFilesBeyondSizeLimit(CancellationToken ct)
|
||||||
|
{
|
||||||
|
var sizeLimit = _configuration.GetValueOrDefault<double>(nameof(StaticFilesServerConfiguration.CacheSizeHardLimitInGiB), -1);
|
||||||
|
if (sizeLimit <= 0)
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
_logger.LogInformation("Cleaning up files beyond the cache size limit of {cacheSizeLimit} GiB", sizeLimit);
|
||||||
|
var allLocalFiles = Directory.EnumerateFiles(_cacheDir, "*", SearchOption.AllDirectories)
|
||||||
|
.Select(f => new FileInfo(f)).ToList()
|
||||||
|
.OrderBy(f => f.LastAccessTimeUtc).ToList();
|
||||||
|
var totalCacheSizeInBytes = allLocalFiles.Sum(s => s.Length);
|
||||||
|
long cacheSizeLimitInBytes = (long)ByteSize.FromGibiBytes(sizeLimit).Bytes;
|
||||||
|
while (totalCacheSizeInBytes > cacheSizeLimitInBytes && allLocalFiles.Any() && !ct.IsCancellationRequested)
|
||||||
|
{
|
||||||
|
var oldestFile = allLocalFiles[0];
|
||||||
|
allLocalFiles.Remove(oldestFile);
|
||||||
|
totalCacheSizeInBytes -= oldestFile.Length;
|
||||||
|
_metrics.DecGauge(MetricsAPI.GaugeFilesTotalSize, oldestFile.Length);
|
||||||
|
_metrics.DecGauge(MetricsAPI.GaugeFilesTotal);
|
||||||
|
_logger.LogInformation("Deleting {oldestFile} with size {size}MiB", oldestFile.FullName, ByteSize.FromBytes(oldestFile.Length).MebiBytes);
|
||||||
|
oldestFile.Delete();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
_logger.LogWarning(ex, "Error during cache size limit cleanup");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void CleanUpOutdatedFiles(CancellationToken ct)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
var unusedRetention = _configuration.GetValueOrDefault(nameof(StaticFilesServerConfiguration.UnusedFileRetentionPeriodInDays), 14);
|
||||||
|
var forcedDeletionAfterHours = _configuration.GetValueOrDefault(nameof(StaticFilesServerConfiguration.ForcedDeletionOfFilesAfterHours), -1);
|
||||||
|
|
||||||
|
_logger.LogInformation("Cleaning up files older than {filesOlderThanDays} days", unusedRetention);
|
||||||
|
if (forcedDeletionAfterHours > 0)
|
||||||
|
{
|
||||||
|
_logger.LogInformation("Cleaning up files written to longer than {hours}h ago", forcedDeletionAfterHours);
|
||||||
|
}
|
||||||
|
|
||||||
|
var prevTime = DateTime.Now.Subtract(TimeSpan.FromDays(unusedRetention));
|
||||||
|
var prevTimeForcedDeletion = DateTime.Now.Subtract(TimeSpan.FromHours(forcedDeletionAfterHours));
|
||||||
|
DirectoryInfo dir = new(_cacheDir);
|
||||||
|
var allFilesInDir = dir.GetFiles("*", SearchOption.AllDirectories);
|
||||||
|
int fileCounter = 0;
|
||||||
|
|
||||||
|
foreach (var file in allFilesInDir)
|
||||||
|
{
|
||||||
|
if (file.LastAccessTime < prevTime)
|
||||||
|
{
|
||||||
|
_metrics.DecGauge(MetricsAPI.GaugeFilesTotalSize, file.Length);
|
||||||
|
_metrics.DecGauge(MetricsAPI.GaugeFilesTotal);
|
||||||
|
_logger.LogInformation("File outdated: {fileName}, {fileSize}MiB", file.Name, ByteSize.FromBytes(file.Length).MebiBytes);
|
||||||
|
file.Delete();
|
||||||
|
}
|
||||||
|
else if (forcedDeletionAfterHours > 0 && file.LastWriteTime < prevTimeForcedDeletion)
|
||||||
|
{
|
||||||
|
_metrics.DecGauge(MetricsAPI.GaugeFilesTotalSize, file.Length);
|
||||||
|
_metrics.DecGauge(MetricsAPI.GaugeFilesTotal);
|
||||||
|
_logger.LogInformation("File forcefully deleted: {fileName}, {fileSize}MiB", file.Name, ByteSize.FromBytes(file.Length).MebiBytes);
|
||||||
|
file.Delete();
|
||||||
|
}
|
||||||
|
|
||||||
|
fileCounter++;
|
||||||
|
|
||||||
|
ct.ThrowIfCancellationRequested();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
_logger.LogWarning(ex, "Error during file cleanup of old files");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -26,6 +26,7 @@ namespace MareSynchronosStaticFilesServer;
|
|||||||
public class Startup
|
public class Startup
|
||||||
{
|
{
|
||||||
private bool _isMain;
|
private bool _isMain;
|
||||||
|
private bool _isDistributionNode;
|
||||||
private readonly ILogger<Startup> _logger;
|
private readonly ILogger<Startup> _logger;
|
||||||
|
|
||||||
public Startup(IConfiguration configuration, ILogger<Startup> logger)
|
public Startup(IConfiguration configuration, ILogger<Startup> logger)
|
||||||
@@ -33,7 +34,8 @@ public class Startup
|
|||||||
Configuration = configuration;
|
Configuration = configuration;
|
||||||
_logger = logger;
|
_logger = logger;
|
||||||
var mareSettings = Configuration.GetRequiredSection("MareSynchronos");
|
var mareSettings = Configuration.GetRequiredSection("MareSynchronos");
|
||||||
_isMain = string.IsNullOrEmpty(mareSettings.GetValue(nameof(StaticFilesServerConfiguration.MainFileServerAddress), string.Empty));
|
_isDistributionNode = mareSettings.GetValue(nameof(StaticFilesServerConfiguration.IsDistributionNode), false);
|
||||||
|
_isMain = string.IsNullOrEmpty(mareSettings.GetValue(nameof(StaticFilesServerConfiguration.MainFileServerAddress), string.Empty)) && _isDistributionNode;
|
||||||
}
|
}
|
||||||
|
|
||||||
public IConfiguration Configuration { get; }
|
public IConfiguration Configuration { get; }
|
||||||
@@ -51,6 +53,7 @@ public class Startup
|
|||||||
|
|
||||||
var mareConfig = Configuration.GetRequiredSection("MareSynchronos");
|
var mareConfig = Configuration.GetRequiredSection("MareSynchronos");
|
||||||
|
|
||||||
|
// metrics configuration
|
||||||
services.AddSingleton(m => new MareMetrics(m.GetService<ILogger<MareMetrics>>(), new List<string>
|
services.AddSingleton(m => new MareMetrics(m.GetService<ILogger<MareMetrics>>(), new List<string>
|
||||||
{
|
{
|
||||||
MetricsAPI.CounterFileRequests,
|
MetricsAPI.CounterFileRequests,
|
||||||
@@ -72,13 +75,24 @@ public class Startup
|
|||||||
MetricsAPI.GaugeQueueInactive,
|
MetricsAPI.GaugeQueueInactive,
|
||||||
MetricsAPI.GaugeQueueActive,
|
MetricsAPI.GaugeQueueActive,
|
||||||
}));
|
}));
|
||||||
|
|
||||||
|
// generic services
|
||||||
services.AddSingleton<CachedFileProvider>();
|
services.AddSingleton<CachedFileProvider>();
|
||||||
services.AddSingleton<FileStatisticsService>();
|
services.AddSingleton<FileStatisticsService>();
|
||||||
services.AddSingleton<RequestFileStreamResultFactory>();
|
services.AddSingleton<RequestFileStreamResultFactory>();
|
||||||
|
services.AddSingleton<ServerTokenGenerator>();
|
||||||
|
services.AddSingleton<RequestQueueService>();
|
||||||
|
services.AddHostedService(p => p.GetService<RequestQueueService>());
|
||||||
services.AddHostedService(m => m.GetService<FileStatisticsService>());
|
services.AddHostedService(m => m.GetService<FileStatisticsService>());
|
||||||
services.AddHostedService<FileCleanupService>();
|
services.AddSingleton<IConfigurationService<MareConfigurationAuthBase>, MareConfigurationServiceClient<MareConfigurationAuthBase>>();
|
||||||
|
services.AddHostedService(p => (MareConfigurationServiceClient<MareConfigurationAuthBase>)p.GetService<IConfigurationService<MareConfigurationAuthBase>>());
|
||||||
|
|
||||||
|
// specific services
|
||||||
|
if (_isMain)
|
||||||
|
{
|
||||||
|
services.AddSingleton<IClientReadyMessageService, MainClientReadyMessageService>();
|
||||||
|
services.AddHostedService<MainFileCleanupService>();
|
||||||
|
services.AddSingleton<IConfigurationService<StaticFilesServerConfiguration>, MareConfigurationServiceServer<StaticFilesServerConfiguration>>();
|
||||||
services.AddDbContextPool<MareDbContext>(options =>
|
services.AddDbContextPool<MareDbContext>(options =>
|
||||||
{
|
{
|
||||||
options.UseNpgsql(Configuration.GetConnectionString("DefaultConnection"), builder =>
|
options.UseNpgsql(Configuration.GetConnectionString("DefaultConnection"), builder =>
|
||||||
@@ -88,63 +102,6 @@ public class Startup
|
|||||||
options.EnableThreadSafetyChecks(false);
|
options.EnableThreadSafetyChecks(false);
|
||||||
}, mareConfig.GetValue(nameof(MareConfigurationBase.DbContextPoolSize), 1024));
|
}, mareConfig.GetValue(nameof(MareConfigurationBase.DbContextPoolSize), 1024));
|
||||||
|
|
||||||
services.AddOptions<JwtBearerOptions>(JwtBearerDefaults.AuthenticationScheme)
|
|
||||||
.Configure<IConfigurationService<MareConfigurationAuthBase>>((o, s) =>
|
|
||||||
{
|
|
||||||
o.TokenValidationParameters = new()
|
|
||||||
{
|
|
||||||
ValidateIssuer = false,
|
|
||||||
ValidateLifetime = false,
|
|
||||||
ValidateAudience = false,
|
|
||||||
ValidateIssuerSigningKey = true,
|
|
||||||
IssuerSigningKey = new SymmetricSecurityKey(Encoding.ASCII.GetBytes(s.GetValue<string>(nameof(MareConfigurationAuthBase.Jwt)))),
|
|
||||||
};
|
|
||||||
});
|
|
||||||
|
|
||||||
services.AddAuthentication(o =>
|
|
||||||
{
|
|
||||||
o.DefaultAuthenticateScheme = JwtBearerDefaults.AuthenticationScheme;
|
|
||||||
o.DefaultChallengeScheme = JwtBearerDefaults.AuthenticationScheme;
|
|
||||||
o.DefaultScheme = JwtBearerDefaults.AuthenticationScheme;
|
|
||||||
}).AddJwtBearer();
|
|
||||||
|
|
||||||
services.AddAuthorization(options =>
|
|
||||||
{
|
|
||||||
options.FallbackPolicy = new AuthorizationPolicyBuilder().RequireAuthenticatedUser().Build();
|
|
||||||
options.AddPolicy("Internal", new AuthorizationPolicyBuilder().RequireClaim(MareClaimTypes.Internal, "true").Build());
|
|
||||||
});
|
|
||||||
|
|
||||||
if (_isMain)
|
|
||||||
{
|
|
||||||
services.AddSingleton<IConfigurationService<StaticFilesServerConfiguration>, MareConfigurationServiceServer<StaticFilesServerConfiguration>>();
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
services.AddSingleton<IConfigurationService<StaticFilesServerConfiguration>, MareConfigurationServiceClient<StaticFilesServerConfiguration>>();
|
|
||||||
services.AddHostedService(p => (MareConfigurationServiceClient<StaticFilesServerConfiguration>)p.GetService<IConfigurationService<StaticFilesServerConfiguration>>());
|
|
||||||
}
|
|
||||||
|
|
||||||
services.AddSingleton<IConfigurationService<MareConfigurationAuthBase>, MareConfigurationServiceClient<MareConfigurationAuthBase>>();
|
|
||||||
|
|
||||||
services.AddSingleton<ServerTokenGenerator>();
|
|
||||||
services.AddSingleton<RequestQueueService>();
|
|
||||||
services.AddHostedService(p => p.GetService<RequestQueueService>());
|
|
||||||
services.AddControllers().ConfigureApplicationPartManager(a =>
|
|
||||||
{
|
|
||||||
a.FeatureProviders.Remove(a.FeatureProviders.OfType<ControllerFeatureProvider>().First());
|
|
||||||
if (_isMain)
|
|
||||||
{
|
|
||||||
a.FeatureProviders.Add(new AllowedControllersFeatureProvider(typeof(MareStaticFilesServerConfigurationController), typeof(CacheController), typeof(RequestController), typeof(ServerFilesController)));
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
a.FeatureProviders.Add(new AllowedControllersFeatureProvider(typeof(CacheController), typeof(RequestController)));
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
services.AddHostedService(p => (MareConfigurationServiceClient<MareConfigurationAuthBase>)p.GetService<IConfigurationService<MareConfigurationAuthBase>>());
|
|
||||||
|
|
||||||
services.AddSingleton<IUserIdProvider, IdBasedUserIdProvider>();
|
|
||||||
var signalRServiceBuilder = services.AddSignalR(hubOptions =>
|
var signalRServiceBuilder = services.AddSignalR(hubOptions =>
|
||||||
{
|
{
|
||||||
hubOptions.MaximumReceiveMessageSize = long.MaxValue;
|
hubOptions.MaximumReceiveMessageSize = long.MaxValue;
|
||||||
@@ -206,6 +163,60 @@ public class Startup
|
|||||||
};
|
};
|
||||||
|
|
||||||
services.AddStackExchangeRedisExtensions<SystemTextJsonSerializer>(redisConfiguration);
|
services.AddStackExchangeRedisExtensions<SystemTextJsonSerializer>(redisConfiguration);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
services.AddSingleton<IClientReadyMessageService, ShardClientReadyMessageService>();
|
||||||
|
services.AddHostedService<ShardFileCleanupService>();
|
||||||
|
services.AddSingleton<IConfigurationService<StaticFilesServerConfiguration>, MareConfigurationServiceClient<StaticFilesServerConfiguration>>();
|
||||||
|
services.AddHostedService(p => (MareConfigurationServiceClient<StaticFilesServerConfiguration>)p.GetService<IConfigurationService<StaticFilesServerConfiguration>>());
|
||||||
|
}
|
||||||
|
|
||||||
|
// controller setup
|
||||||
|
services.AddControllers().ConfigureApplicationPartManager(a =>
|
||||||
|
{
|
||||||
|
a.FeatureProviders.Remove(a.FeatureProviders.OfType<ControllerFeatureProvider>().First());
|
||||||
|
if (_isMain)
|
||||||
|
{
|
||||||
|
a.FeatureProviders.Add(new AllowedControllersFeatureProvider(typeof(MareStaticFilesServerConfigurationController),
|
||||||
|
typeof(CacheController), typeof(RequestController), typeof(ServerFilesController),
|
||||||
|
typeof(DistributionController), typeof(MainController)));
|
||||||
|
}
|
||||||
|
else if (_isDistributionNode)
|
||||||
|
{
|
||||||
|
a.FeatureProviders.Add(new AllowedControllersFeatureProvider(typeof(CacheController), typeof(RequestController), typeof(DistributionController)));
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
a.FeatureProviders.Add(new AllowedControllersFeatureProvider(typeof(CacheController), typeof(RequestController)));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// authentication and authorization
|
||||||
|
services.AddOptions<JwtBearerOptions>(JwtBearerDefaults.AuthenticationScheme)
|
||||||
|
.Configure<IConfigurationService<MareConfigurationAuthBase>>((o, s) =>
|
||||||
|
{
|
||||||
|
o.TokenValidationParameters = new()
|
||||||
|
{
|
||||||
|
ValidateIssuer = false,
|
||||||
|
ValidateLifetime = false,
|
||||||
|
ValidateAudience = false,
|
||||||
|
ValidateIssuerSigningKey = true,
|
||||||
|
IssuerSigningKey = new SymmetricSecurityKey(Encoding.ASCII.GetBytes(s.GetValue<string>(nameof(MareConfigurationAuthBase.Jwt))))
|
||||||
|
};
|
||||||
|
});
|
||||||
|
services.AddAuthentication(o =>
|
||||||
|
{
|
||||||
|
o.DefaultAuthenticateScheme = JwtBearerDefaults.AuthenticationScheme;
|
||||||
|
o.DefaultChallengeScheme = JwtBearerDefaults.AuthenticationScheme;
|
||||||
|
o.DefaultScheme = JwtBearerDefaults.AuthenticationScheme;
|
||||||
|
}).AddJwtBearer();
|
||||||
|
services.AddAuthorization(options =>
|
||||||
|
{
|
||||||
|
options.FallbackPolicy = new AuthorizationPolicyBuilder().RequireAuthenticatedUser().Build();
|
||||||
|
options.AddPolicy("Internal", new AuthorizationPolicyBuilder().RequireClaim(MareClaimTypes.Internal, "true").Build());
|
||||||
|
});
|
||||||
|
services.AddSingleton<IUserIdProvider, IdBasedUserIdProvider>();
|
||||||
|
|
||||||
services.AddHealthChecks();
|
services.AddHealthChecks();
|
||||||
services.AddHttpLogging(e => e = new Microsoft.AspNetCore.HttpLogging.HttpLoggingOptions());
|
services.AddHttpLogging(e => e = new Microsoft.AspNetCore.HttpLogging.HttpLoggingOptions());
|
||||||
@@ -228,8 +239,12 @@ public class Startup
|
|||||||
app.UseAuthorization();
|
app.UseAuthorization();
|
||||||
|
|
||||||
app.UseEndpoints(e =>
|
app.UseEndpoints(e =>
|
||||||
|
{
|
||||||
|
if (_isMain)
|
||||||
{
|
{
|
||||||
e.MapHub<MareSynchronosServer.Hubs.MareHub>("/dummyhub");
|
e.MapHub<MareSynchronosServer.Hubs.MareHub>("/dummyhub");
|
||||||
|
}
|
||||||
|
|
||||||
e.MapControllers();
|
e.MapControllers();
|
||||||
e.MapHealthChecks("/health").WithMetadata(new AllowAnonymousAttribute());
|
e.MapHealthChecks("/health").WithMetadata(new AllowAnonymousAttribute());
|
||||||
});
|
});
|
||||||
|
|||||||
Reference in New Issue
Block a user