当前位置：首页 > news >正文

ASP.NET CORE WEBAPI文件下载

news 2024/5/15 7:38:34

最近要使用ASP.NET CORE WEBAPI用来下载文件，使用的.NET CORE 3.1。考虑如下场景：

文件是程序生成的。
文件应该能兼容各种格式。
浏览器可以感知进行下载。

准备

经过简单的调研，得到以下结论。

ASP.NET CORE 提供FileResult这种类型的ActionResult，可以直接返回文件结果，不需要直接处理HttpResponse。
通过Stream可以直接返回文件流供浏览器下载。
FileStreamResult是FileResult的具体实现，返回值应该是此类对象。
Stream有多种类型，适合直接内存中生成文件对象的是MemoryStream。
对目标有了基础的了解，就可以开始动手实现了。

实现

建立好ASP.NET CORE WEBAPI工程，把生成文件的代码独立出来一个函数。我这里需要是下载一个CSV格式的文件，因此生成一个CSV文件。
对于磁盘上的文件，可以使用FileStream对象，由于我这里需要运行中生成这个文件，需要使用MemoryStream。

using var stream = new MemoryStream();
using var writer = new StreamWriter(stream);
//生成标题
var propCollection = ttype.GetProperties();
foreach (var n in propCollection)
{writer.Write(n.Name);writer.Write(",");
}
writer.WriteLine();
//生成内容
foreach (var item in res)
{foreach (var n in propCollection){writer.Write(Convert.ToString(n.GetValue(item)));writer.Write(",");}writer.WriteLine();
}

请不要考虑里面反射的相关内容，按照自己的逻辑生成CSV即可，我只是懒得改代码而已。
代码中使用到了一些新的语法特性，请注意对低版本的.NET不一定适用。
直接返回Stream对象给Controller处理，处理代码如下：

var res = await info.GetAllQueryResult();
var actionresult = new FileStreamResult(res, new Microsoft.Net.Http.Headers.MediaTypeHeaderValue("text/csv"));
return actionresult;

CSV的Content-Type是text/csv，如果下载别的文件，请自行查询MIME格式。

调试

直接执行上面的代码，直接报错“无法读取已经关闭的流”。猜测是离开using语句块的时候，stream自动被关闭了。改动很简单，去掉using语句，不再报相同错误。

但是返回的文件长度一直是0，单步调试发现Writer执行完毕之后，stream返回的长度是0，内容实际上并没有写入，想起有一个Flush()，可以添加以确保数据写入。

单步显示stream长度有了，但是返回的长度还是0。继续单步调试发现Stream的Postion是停在文件结尾的，这个和直接开始读取文件完全不一样，文件读取一般是从开头开始的，于是直接设置Postion为0，问题解决。

下载能够成功了，但是文件名一直显示的是随机生成的，体验很差。设置一下FileDownloadName即可。

核心代码如下：

public async Task<Stream> GetAllQueryResult()
{var stream = new MemoryStream();var writer = new StreamWriter(stream);//生成标题var propCollection = ttype.GetProperties();foreach (var n in propCollection){writer.Write(n.Name);writer.Write(",");}writer.WriteLine();//生成内容foreach (var item in res){foreach (var n in propCollection){writer.Write(Convert.ToString(n.GetValue(item)));writer.Write(",");}writer.WriteLine();}writer.Flush();stream.Position = 0;return stream;
}

[HttpPost("file")]
[ProducesResponseType(typeof(FileResult), Status200OK)]
public async Task<FileResult> Download()
{var info = new Info();var res = await info.GetAllQueryResult();var actionresult = new FileStreamResult(res, new Microsoft.Net.Http.Headers.MediaTypeHeaderValue("text/csv"));actionresult.FileDownloadName = "Carinfos.csv";//Response.ContentLength = res.Length;return actionresult;
}

使用swagger调用，最后效果：