Как вернуть файл, сохраненный как байт [] в ASP. NET Core? - PullRequest
Новая
       75

Как вернуть файл, сохраненный как байт [] в ASP. NET Core?

0 голосов
/ 31 января 2020

Я долго искал, и хотя все должно быть просто, я просто не могу заставить его работать. На основании примеров, которые я видел, вот где я дошел:

SomeAppService.cs 

public async Task<FileStream> Download(long? id)
{
    var attachment = await _repository.FirstOrDefaultAsync(x => x.Id == id);

    var fileStream = new FileStream($"{attachment.FileName}.{attachment.FileExtension}", 
        FileMode.Create, FileAccess.Write);
    fileStream.Write(attachment.File, 0, attachment.File.Length);

    return fileStream;
}

Как можно заметить, "FileName", «FileExtension» и «File» (который является вышеупомянутым байтовым массивом) хранятся в базе данных. Вложение может быть любым видом файла, за исключением запрещенных расширений в методе Upload (не показан). Тогда в моем контроллере у меня есть:

SomeController.cs 

[AllowAnonymous]
[HttpGet("Download/{id}")]
public async Task<IActionResult> Download(long? id)
{
    var fileStream = await appService.Download(id);
    return new FileStreamResult(fileStream, "application/octet-stream");
}

Однако, когда я достигаю конечной точки загрузки, я получаю файл с именем "response" , без расширения, с 0 байтами.

Ресурсы:

Файл возврата в ASP. Net Core Web API

Возврат FileResult из байта []

Сохранение и загрузка MemoryStream в / из файла (Ответ с 255 ответами дал мне представление о том, как превратить массив байтов в поток файлов, но я не знаю, работает ли это)

Ответы [ 2 ]

0 голосов
/ 31 января 2020

Если вы хотите передавать файл из большого двоичного объекта базы данных из структуры сущностей, не загружая его в память. Сначала разбейте модель данных на две части: 

public class Attachment{
    public int Id { get; set; }
    public string Filename { get; set; }
    public string ContentType { get; set; }
    public virtual AttachmentBlob Blob { get; set; }
    //...
}

public class AttachmentBlob{
    public int Id { get; set; }
    public byte[] File { get; set; }
}

Отобразите их в одну и ту же таблицу, но не как собственный тип; 

   modelBuilder.Entity<Attachment>(e => {
       e.HasOne(a => a.Blob)
        .WithOne()
        .HasForeignKey<AttachmentBlob>(b => b.Id);
   });
   modelBuilder.Entity<AttachmentBlob>(e => {
       e.ToTable("Attachment");
   });

Затем вы можете читать и записывать их как байтовые массивы или как потоки; 

   public static async Task Read(DbContext db, Attachment attachment, Func<Stream,Task> callback)
   {
      await db.Database.OpenConnectionAsync();
      try {
         var conn = db.Database.GetDbConnection();
         var cmd = conn.CreateCommand();
         var parm = cmd.CreateParameter();
         cmd.Parameters.Add(parm);
         parm.ParameterName = "@id";
         parm.Value = attachment.Id;
         cmd.CommandText = "select File from Attachment where Id = @id";
         using (var reader = await cmd.ExecuteReaderAsync()){
            if (await reader.ReadAsync())
               await callback(reader.GetStream(0));
         }
      } finally {
         await db.Database.CloseConnectionAsync();
      }
   }

   public class AttachmentResult : FileStreamResult
   {
      private readonly DbContext db;
      private readonly Attachment attachment;

      public AttachmentResult(DbContext db, Attachment attachment) : base(new MemoryStream(), attachment.ContentType)
      {
         this.db = db;
         this.attachment = attachment;
      }

      public override async Task ExecuteResultAsync(ActionContext context)
      {
         await Read(db, attachment, async s => {
            FileStream = s;
            await base.ExecuteResultAsync(context);
         });
      }
   }

   public static async Task Write(DbContext db, Attachment attachment, Stream content)
   {
      await db.Database.OpenConnectionAsync();
      try {
         var conn = db.Database.GetDbConnection();
         var cmd = conn.CreateCommand();
         cmd.Transaction = db.Database.CurrentTransaction?.GetDbTransaction();
         var parm = cmd.CreateParameter();
         cmd.Parameters.Add(parm);
         parm.ParameterName = "@id";
         parm.Value = attachment.Id;
         parm = cmd.CreateParameter();
         cmd.Parameters.Add(parm);
         parm.ParameterName = "@content";
         parm.Value = content;
         cmd.CommandText = "update Attachment set File = @content where Id = @id";
         await cmd.ExecuteNonQueryAsync();
      } finally {
         await db.Database.CloseConnectionAsync();
      }
   }

   public static Task InsertAttachment(DbContext db, Attachment attachment, Stream content){
      var strat = db.Database.CreateExecutionStrategy();
      return strat.ExecuteAsync(async () => {
         using (var trans = await db.Database.BeginTransactionAsync())
         {
             db.Set<Attachment>.Add(attachment);
             await db.SaveChangesAsync();
             await Write(db, attachment, content);
             trans.Commit();
             db.ChangeTracker.AcceptAllChanges();
         }
      });
   }
0 голосов
/ 31 января 2020

Спасибо всем, в итоге FileContentResult сделал свое дело.

Вот как это выглядит:

Сервис 

public async Task<Attachment> Download(long? id)
{
    return await _repository.FirstOrDefaultAsync(x => x.Id == id);
}

Контроллер 

[AllowAnonymous]
[HttpGet("Download/{id}")]
public async Task<IActionResult> Download(long? id)
{
    var attachment = await appService.Download(id);
    return new FileContentResult(attachment.File, 
        MimeTypeMap.GetMimeType(attachment.FileExtension))
    {
        FileDownloadName = $"{attachment.NomeFile}.{attachment.FileExtension}"
    };
}

MimeTypeMap можно найти здесь

...