DocSpace-buildtools/common/ASC.Api.Core/Middleware/DisposeMiddleware.cs

using System.Threading.Tasks;

using ASC.Common.Web;

using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Http;

namespace ASC.Api.Core.Middleware
{
    public class DisposeMiddleware
    {
        private readonly RequestDelegate _next;

        public DisposeMiddleware(RequestDelegate next) => _next = next;

        public async Task Invoke(HttpContext context)
        {
            context.Response.RegisterForDispose(new DisposableHttpContext(context));

            await _next.Invoke(context);
        }
    }

    public static class DisposeMiddlewareExtensions
    {
        public static IApplicationBuilder UseDisposeMiddleware(this IApplicationBuilder builder) =>
            builder.UseMiddleware<DisposeMiddleware>();
    }
}