ASP.NET Core 中的请求超时中间件

这篇文章距离上次修改已过259天，其中的内容可能已经有所变动。

在ASP.NET Core中，你可以使用一个自定义的中间件来处理请求超时的情况。以下是一个简单的示例代码，演示了如何创建一个请求超时中间件：

using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Http;
using System;
using System.Threading;
using System.Threading.Tasks;

public class RequestTimeoutMiddleware
{
    private readonly RequestDelegate _next;
    private readonly TimeSpan _timeout;

    public RequestTimeoutMiddleware(RequestDelegate next, TimeSpan timeout)
    {
        _next = next;
        _timeout = timeout;
    }

    public async Task Invoke(HttpContext context)
    {
        var cancellationTokenSource = new CancellationTokenSource(_timeout);
        var cancellationToken = cancellationTokenSource.Token;

        // 注册超时处理
        cancellationToken.Register(() =>
        {
            context.Response.StatusCode = 408; // 超时响应码
            return Task.CompletedTask;
        });

        try
        {
            await _next(context);
        }
        catch (OperationCanceledException) when (!cancellationToken.IsCancellationRequested)
        {
            // 如果是由于超时抛出的OperationCanceledException，则不处理
        }
    }
}

// 在Startup.cs中使用中间件
public void Configure(IApplicationBuilder app)
{
    // 5秒超时
    var timeout = TimeSpan.FromSeconds(5);
    app.UseMiddleware<RequestTimeoutMiddleware>(timeout);

    // 其他中间件配置...
}

这段代码定义了一个名为RequestTimeoutMiddleware的类，它实现了请求超时的功能。在Invoke方法中，它使用CancellationTokenSource设置了一个指定的超时时间，并且注册了一个超时处理函数，该函数设置了响应的状态码为408（请求超时）。然后它将请求传递给下一个中间件，并捕获可能发生的OperationCanceledException，以确保在请求超时时不会处理异常。

在Startup.cs的Configure方法中，你可以通过app.UseMiddleware<RequestTimeoutMiddleware>(timeout);来应用这个超时中间件，其中timeout是你希望设置的超时时间。

ASP.NET Core 中的请求超时中间件

评论已关闭

推荐阅读