ToxicMushroom/mare.server
1
0
Fork 0
mirror of https://github.com/Caraxi/mare.server.git synced 2025-12-12 22:17:22 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

rework static files server

Browse source
This commit is contained in:
rootdarkarchon 2022-12-20 14:20:24 +01:00
parent ea19231851
commit 25aa01c819
12 changed files with 315 additions and 224 deletions
Download patch file Download diff file Expand all files Collapse all files

107
MareSynchronosServer/MareSynchronosStaticFilesServer/FileService.cs
View file

@ -1,107 +0,0 @@
using Grpc.Core;
using MareSynchronosShared.Data;
using MareSynchronosShared.Metrics;
using MareSynchronosShared.Protos;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.Logging;
using System;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
namespace MareSynchronosStaticFilesServer;
public class FileService : MareSynchronosShared.Protos.FileService.FileServiceBase
{
private readonly string _basePath;
private readonly MareDbContext _mareDbContext;
private readonly ILogger<FileService> _logger;
private readonly MareMetrics _metricsClient;
public FileService(MareDbContext mareDbContext, IConfiguration configuration, ILogger<FileService> logger, MareMetrics metricsClient)
{
_basePath = configuration.GetRequiredSection("MareSynchronos")["CacheDirectory"];
_mareDbContext = mareDbContext;
_logger = logger;
_metricsClient = metricsClient;
}
public override async Task<Empty> UploadFile(IAsyncStreamReader<UploadFileRequest> requestStream, ServerCallContext context)
{
await requestStream.MoveNext();
var uploadMsg = requestStream.Current;
var filePath = Path.Combine(_basePath, uploadMsg.Hash);
using var fileWriter = File.OpenWrite(filePath);
var file = await _mareDbContext.Files.SingleOrDefaultAsync(f => f.Hash == uploadMsg.Hash && f.UploaderUID == uploadMsg.Uploader);
if (file != null)
{
await fileWriter.WriteAsync(uploadMsg.FileData.ToArray());
while (await requestStream.MoveNext())
{
await fileWriter.WriteAsync(requestStream.Current.FileData.ToArray());
}
await fileWriter.FlushAsync();
fileWriter.Close();
var fileSize = new FileInfo(filePath).Length;
file.Uploaded = true;
_metricsClient.IncGauge(MetricsAPI.GaugeFilesTotal, 1);
_metricsClient.IncGauge(MetricsAPI.GaugeFilesTotalSize, fileSize);
await _mareDbContext.SaveChangesAsync().ConfigureAwait(false);
_logger.LogInformation("User {user} uploaded file {hash}", uploadMsg.Uploader, uploadMsg.Hash);
}
return new Empty();
}
public override async Task<Empty> DeleteFiles(DeleteFilesRequest request, ServerCallContext context)
{
foreach (var hash in request.Hash)
{
try
{
FileInfo fi = new FileInfo(Path.Combine(_basePath, hash));
fi.Delete();
var file = await _mareDbContext.Files.SingleOrDefaultAsync(f => f.Hash == hash);
if (file != null)
{
_mareDbContext.Files.Remove(file);
_metricsClient.DecGauge(MetricsAPI.GaugeFilesTotal, 1);
_metricsClient.DecGauge(MetricsAPI.GaugeFilesTotalSize, fi.Length);
}
}
catch (Exception ex)
{
_logger.LogError(ex, "Could not delete file for hash {hash}", hash);
}
}
await _mareDbContext.SaveChangesAsync().ConfigureAwait(false);
return new Empty();
}
public override Task<FileSizeResponse> GetFileSizes(FileSizeRequest request, ServerCallContext context)
{
FileSizeResponse response = new();
foreach (var hash in request.Hash.Distinct())
{
FileInfo fi = new(Path.Combine(_basePath, hash));
if (fi.Exists)
{
response.HashToFileSize.Add(hash, fi.Length);
}
else
{
response.HashToFileSize.Add(hash, 0);
}
}
return Task.FromResult(response);
}
}