如何在ASP.NET核心3.1Web API中捕获当前状态代码

public class StatusCodeMiddleware : IMiddleware
{
    public async Task InvokeAsync(HttpContext context, RequestDelegate next)
    {
        await next(context);

        if (context.Response.StatusCode == 404)
        {
            // 处理状态代码为404的情况
        }
        else if (context.Response.StatusCode == 500)
        {
            // 处理状态代码为500的情况
        }
        // 其他状态代码的处理
    }
}

app.UseMiddleware<StatusCodeMiddleware>();

context.Response.ContentType = "application/json";
context.Response.StatusCode = 404;
await context.Response.WriteAsync("资源未找到");