C#设置multipartReader.ReadNextSectionAsync()的超时时间

6

我正在使用C# .net core读取用户发送的多个文件的上传数据。在读取最后一个文件后,如何防止用户等待无限时间。

try
{
  var cancellationTokenSource = new CancellationTokenSource();
  cancellationTokenSource.CancelAfter(1000);
  section = await multipartReader.ReadNextSectionAsync(cancellationTokenSource.Token);
}
catch (Exception ex)
{
  throw;
}

虽然我已经设置了1秒的cancelationToken,但它变成了无限制的,如果我发送另一个请求,它不会进入下一行。

public static async Task<HttpRequest> FromHttpContextAsync(HttpContext httpContext)
        {
            bool multipart = false;
            HttpRequest retVal = new HttpRequest(httpContext);
            var sb = new StringBuilder();         
            var sr = new StreamReader(httpContext.Stream, Encoding.UTF8);
            {
                var line1 = await sr.ReadLineAsync();
                sb.AppendLine(line1);
                var Line1Parts = (line1).Split(' ');
                retVal.Methode = Line1Parts[0].ToLower();
                retVal.RawUrl = System.Net.WebUtility.UrlDecode(Line1Parts[1]).Replace("&amp;", "&");
                var urlPart = retVal.RawUrl.Split('?');
                retVal.Url = urlPart[0];
                if (urlPart.Length > 1)
                {
                    foreach (var part in urlPart[1].Split(new[] { '&' }, StringSplitOptions.RemoveEmptyEntries))
                    {
                        var tmp = part.Split(new[] { '=' }, StringSplitOptions.RemoveEmptyEntries);
                        try
                        {
                            retVal.QueryStrings.Add(tmp[0], tmp[1]);
                        }
                        catch (Exception ex)
                        {
                        }
                    }
                }
            string line = await sr.ReadLineAsync();
            sb.AppendLine(line);
            int contentLength = 0;
            while (!string.IsNullOrEmpty(line))
            {
                var tmp = line.Split(':');
                var key = tmp[0].Trim().ToLower();
                retVal.Header.Add(new KeyValuePair<string, string>(tmp[0], tmp[1]));

                switch (key)
                {

                    case "cookie":
                        {

                            foreach (var part in tmp[1].Split(new[] { ';' }, StringSplitOptions.RemoveEmptyEntries))
                            {
                                var pares = part.Split('=');
                                if (pares.Length == 2)
                                {
                                    retVal.Cookies.Add(new KeyValuePair<string, string>(pares[0], pares[1]));
                                }
                            }
                            break;
                        }
                    case "content-length":
                        {
                            contentLength = int.Parse(tmp[1]);
                            break;
                        }
                }
                line = await sr.ReadLineAsync();
                sb.AppendLine(line);
            }
if (sb.ToString().Contains("Content-Type: multipart/form-data"))
{
  string boundary = FindBoundary(sb.ToString());
  MultipartReader multipartReader = new MultipartReader(boundary, httpContext.Stream);
  var section = await multipartReader.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 MemoryStream())
    {
      do
      {
        try
        {
          bytesRead = await section.Body.ReadAsync(buffer, 0, buffer.Length);
        }
        catch (Exception ex)
        {
          Console.Write(ex);
        }

        stream.Write(buffer, 0, bytesRead);
      } while (bytesRead > 0);
      retVal.Files.Add(new Tuple<string, string, byte[]>("", fileName, stream.ToArray()));
    }

    try
    {
      var cancellationTokenSource = new CancellationTokenSource();
      cancellationTokenSource.CancelAfter(1000);
      section = await multipartReader.ReadNextSectionAsync(cancellationTokenSource.Token);
    }
    catch (Exception ex)
    {
      throw;
    }


  }
  foreach (var file in retVal.Files)
  {
    File.WriteAllBytes("d:\\" + file.Item2, file.Item3);
  }
}

HttpContext是此项目的内联类,以下是HttpContext的源代码:

public class HttpContext
    {
        public HttpRequest Request { get; private set; }
        public HttpResponse Response { get; private set; }
        public Stream Stream { private set; get; }
        private HttpContext(Stream networkStream)
        {
            Stream = networkStream;
        }

        public async static Task<HttpContext> FromHttpContextAsync(Stream networkStream)
        {
            var retVal = new HttpContext(networkStream);
            retVal.Request = await HttpRequest.FromHttpContextAsync(retVal);
            retVal.Response = HttpResponse.FromHttpContext(retVal);
            return retVal;
        }
    }

1
以上代码叫什么?您还没有包含封装方法签名。描述的症状听起来像死锁,建议首先检查死锁。 - Nkosi
@Nkosi,我从流中读取数据,直到达到\r\n\r\n为止,这样可以读取所有的头部信息。然后从导出的字符串中提取边界,并将流和边界传递给读取各个部分的函数。但是当想要读取最后一个部分时,如果没有该部分会出现什么问题呢? - Hamed_gibago
1个回答

0

由于缺乏细节和上下文,尝试重现此问题非常困难,我怀疑这里的问题是由于NetworkStreams(在您的MultipartReader实例下使用)尚未完全支持CancellationTokens。 实际上,在.NET Core上几乎每个Socket相关操作都只检查最终传递的CancellationToken - 在我看来这是无用的。

好消息是,.NET Core团队正在积极解决这个问题,我相信这个问题将在.NET Core 3.0中完全解决:

https://github.com/dotnet/corefx/issues/24430

作为一个临时的丑陋解决方案,您可以更改代码以等待一个虚构的延迟任务和您对ReadNextSectionAsync()的调用,假设您不想在此之后重新使用那个停滞的套接字/NetworkStream:
var timeoutTask = Task.Delay(TimeSpan.FromSeconds(1)).ConfigureAwait(false);
var readNextSectionTask = multipartReader.ReadNextSectionAsync().ConfigureAwait(false);

if (await Task.WhenAny(timeoutTask, readNextSectionTask).ConfigureAwait(false) == timeoutTask)
{
    // TODO: Handle the timeout
}
else
{
    section = await readNextSectionTask;
}

此外,您未配置可等待对象可能会导致潜在的死锁源(我不确定您是在 ASP.NET Core 上运行此代码还是在其他同步上下文提供程序上)。为了排除这种可能性,建议在 await 调用之后立即调用 ConfigureAwait(false),就像您在我的先前代码块中看到的那样。

@erfancobisi 我通过来自mono旧源的一个组件解决了这个问题:http://www.mono-project.com jgauffin/Griffin.WebServer - Hamed_gibago

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接