decouple fileservice to be able to run standalone

2022-08-23 02:28:04 +02:00
parent bdfe51c15c
commit a63174009f
11 changed files with 297 additions and 76 deletions
--- a/MareSynchronosServer/MareSynchronosStaticFilesServer/FileService.cs
+++ b/MareSynchronosServer/MareSynchronosStaticFilesServer/FileService.cs
@@ -0,0 +1,95 @@
+using Grpc.Core;
+using MareSynchronosShared.Data;
+using MareSynchronosShared.Metrics;
+using MareSynchronosShared.Protos;
+using Microsoft.EntityFrameworkCore;
+using System.Security.Policy;
+
+namespace MareSynchronosStaticFilesServer;
+
+public class FileService : MareSynchronosShared.Protos.FileService.FileServiceBase
+{
+    private readonly string _basePath;
+    private readonly MareDbContext _mareDbContext;
+    private readonly ILogger<FileService> _logger;
+    private readonly MetricsService.MetricsServiceClient _metricsClient;
+
+    public FileService(MareDbContext mareDbContext, IConfiguration configuration, ILogger<FileService> logger, MetricsService.MetricsServiceClient metricsClient)
+    {
+        _basePath = configuration["CacheDirectory"];
+        _mareDbContext = mareDbContext;
+        _logger = logger;
+        _metricsClient = metricsClient;
+    }
+
+    public override async Task<Empty> UploadFile(UploadFileRequest request, ServerCallContext context)
+    {
+
+        var filePath = Path.Combine(_basePath, request.Hash);
+        var file = await _mareDbContext.Files.SingleOrDefaultAsync(f => f.Hash == request.Hash && f.UploaderUID == request.Uploader);
+        if (file != null)
+        {
+            var byteData = request.FileData.ToArray();
+            await File.WriteAllBytesAsync(filePath, byteData);
+            file.Uploaded = true;
+
+            await _metricsClient.IncGaugeAsync(new GaugeRequest()
+            { GaugeName = MetricsAPI.GaugeFilesTotalSize, Value = byteData.Length }).ConfigureAwait(false);
+            await _metricsClient.IncGaugeAsync(new GaugeRequest()
+            { GaugeName = MetricsAPI.GaugeFilesTotal, Value = 1 }).ConfigureAwait(false);
+
+            await _mareDbContext.SaveChangesAsync().ConfigureAwait(false);
+            _logger.LogInformation("User {user} uploaded file {hash}", request.Uploader, request.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);
+
+                    await _metricsClient.DecGaugeAsync(new GaugeRequest()
+                    { GaugeName = MetricsAPI.GaugeFilesTotalSize, Value = fi.Length }).ConfigureAwait(false);
+                    await _metricsClient.DecGaugeAsync(new GaugeRequest()
+                    { GaugeName = MetricsAPI.GaugeFilesTotal, Value = 1 }).ConfigureAwait(false);
+                }
+            }
+            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)
+        {
+            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);
+    }
+}