feat: 为镜像构建脚本增加了一键导出镜像

AicsKnowledgeBase_file/Controllers/FileController.cs

@@ -19,9 +19,10 @@ public class FileController : ControllerBase {
     public async Task<IActionResult> GetFileStatus(string fileId) {
         try {
             var metadata = await _knowledgeFileHandler.GetFileMetadata(fileId);
-            var existingRanges = await _knowledgeFileHandler.GetAllPartFileRanges(fileId);
+            var existingRanges = (await _knowledgeFileHandler.GetAllPartFileRanges(fileId))
+                .Select((range, _) => $"{range.Item1}-{range.Item2}");
 
-            return Ok(new FileStatus(metadata.IsCompleted, metadata.Md5, metadata.Size, existingRanges));
+            return Ok(new FileStatus(metadata.IsCompleted, metadata.Md5, metadata.Size, existingRanges.ToList()));
         } catch (FileNotFoundException) {
             return NotFound();
         }
@@ -83,5 +84,5 @@ internal record FileStatus(
     bool IsCompleted,
     string Md5,
     ulong Size,
-    List<Tuple<ulong, ulong>> ExistingRanges
+    List<string> ExistingRanges
 );

AicsKnowledgeBase_file/Handlers/KnowledgeFileHandler.cs

@@ -20,11 +20,11 @@ public class KnowledgeFileHandler {
     }
 
     public async Task<string[]> GetAllPartFiles(string fileId) {
-        return Directory.GetFiles($"{FilePath}/{fileId}.part.*");
+        return Directory.GetFiles(FilePath, $"{fileId}.part.*");
     }
 
     public Tuple<ulong, ulong> ConvertPartFileNameToRange(string fileName) {
-        var range = fileName.Split('.')[2].Split('-');
+        var range = fileName.Split('.').Last().Split('-');
         return Tuple.Create(ulong.Parse(range[0]), ulong.Parse(range[1]));
     }
 
@@ -43,7 +43,8 @@ public class KnowledgeFileHandler {
     }
 
     public async Task SaveFileSlice(string fileId, IFormFile file, ulong rangeStart, ulong rangeEnd) {
-        await file.CopyToAsync(File.Create(GetSliceFileName(fileId, rangeStart, rangeEnd)));
+        await using var fileStream = File.Create(GetSliceFileName(fileId, rangeStart, rangeEnd));
+        await file.CopyToAsync(fileStream);
     }
 
     /// <exception cref="InvalidDataException">缺失分片时抛出，其Message内容是缺失分片的Range</exception>

AicsKnowledgeBase_file/Program.cs

@@ -8,6 +8,7 @@ Directory.CreateDirectory("./files");
 var builder = WebApplication.CreateBuilder(args);
 builder.WebHost.UseUrls("https://*:8081");
 builder.WebHost.UseKestrel((context, options) => {
+    options.Limits.MaxRequestBodySize = 1073741824L;
     options.ConfigureEndpointDefaults(listenOptions => {
         listenOptions.Protocols = HttpProtocols.Http1AndHttp2AndHttp3;
         listenOptions.UseHttps(builder.Configuration.GetSection(SslOptions.SectionName).Get<SslOptions>()!

buildDockerImage.ps1

@@ -1,2 +1,4 @@
 docker build -f .\AicsKnowledgeBase_file\Dockerfile -t auto/aics_file:latest .
+echo 'saving...'
+docker save -o ..\docker\aics_file.tar auto/aics_file:latest
 pause