处理ASP.NET Core 1.0上的大文件上传

Martin 发表于 Dev

马丁

当我将大文件上传到ASP.NET Core中的Web api时，运行时会在将我的处理和存储上传功能触发之前将文件加载到内存中。对于较大的上载，由于速度慢且需要更多的内存，因此这成为一个问题。对于早期版本的ASP.NET ，有一些文章介绍了如何禁用请求缓冲，但是我找不到关于如何使用ASP.NET Core进行处理的任何信息。是否可以禁用请求缓冲，这样我就不会一直用完服务器上的内存？

肖恩·卢丁（Shaun Luttin）

使用Microsoft.AspNetCore.WebUtilities.MultipartReader它，因为它...

可以使用最少的缓冲来解析任何流。它一次给您每个节的标题和主体，然后您可以对该节的主体进行所需的操作（缓冲，丢弃，写入磁盘等）。

这是一个中间件示例。

app.Use(async (context, next) =>
{
    if (!IsMultipartContentType(context.Request.ContentType))
    {
        await next();
        return;
    }

    var boundary = GetBoundary(context.Request.ContentType);
    var reader = new MultipartReader(boundary, context.Request.Body);
    var section = await reader.ReadNextSectionAsync();

    while (section != null)
    {
        // process each image
        const int chunkSize = 1024;
        var buffer = new byte[chunkSize];
        var bytesRead = 0;
        var fileName = GetFileName(section.ContentDisposition);

        using (var stream = new FileStream(fileName, FileMode.Append))
        {
            do
            {
                bytesRead = await section.Body.ReadAsync(buffer, 0, buffer.Length);
                stream.Write(buffer, 0, bytesRead);

            } while (bytesRead > 0);
        }

        section = await reader.ReadNextSectionAsync();
    }

    context.Response.WriteAsync("Done.");
});

这是帮手。

private static bool IsMultipartContentType(string contentType)
{
    return 
        !string.IsNullOrEmpty(contentType) &&
        contentType.IndexOf("multipart/", StringComparison.OrdinalIgnoreCase) >= 0;
}

private static string GetBoundary(string contentType)
{
    var elements = contentType.Split(' ');
    var element = elements.Where(entry => entry.StartsWith("boundary=")).First();
    var boundary = element.Substring("boundary=".Length);
    // Remove quotes
    if (boundary.Length >= 2 && boundary[0] == '"' && 
        boundary[boundary.Length - 1] == '"')
    {
        boundary = boundary.Substring(1, boundary.Length - 2);
    }
    return boundary;
}

private string GetFileName(string contentDisposition)
{
    return contentDisposition
        .Split(';')
        .SingleOrDefault(part => part.Contains("filename"))
        .Split('=')
        .Last()
        .Trim('"');
}