ASP.NET 6中间件：用Pipe读取并覆写其他中间件响应

核心实现思路 #

要在MiddlewareA中读取MiddlewareB返回的响应并按需覆写，同时添加响应头，核心步骤如下：

1. 拦截响应流：用Pipe替换原始响应流，让后续中间件（包括MiddlewareB）将响应写入Pipe。
2. 执行后续中间件：调用next()触发链式处理，待后续中间件完成响应写入后再读取Pipe内容。
3. 处理响应内容：根据自定义条件判断是否覆写响应，覆写则生成新响应，否则将原始内容写回原始流。
4. 设置响应头：通过context.Response.OnStarting()在响应即将发送前添加头，确保无论响应是否被覆写都生效。

完整代码实现 #

public class MiddlewareA
{
    private readonly RequestDelegate _next;

    public MiddlewareA(RequestDelegate next)
    {
        _next = next;
    }

    public async Task InvokeAsync(HttpContext context)
    {
        // 保存原始响应流与特性
        var originalBody = context.Response.Body;
        var originalBodyFeature = context.Features.Get<IHttpResponseBodyFeature>();

        try
        {
            // 创建Pipe用于捕获后续中间件的响应输出
            using var pipe = new Pipe();
            var pipeWriter = pipe.Writer;

            // 替换响应流为Pipe的Writer流，确保后续中间件写入Pipe
            context.Response.Body = pipeWriter.AsStream();
            context.Features.Set<IHttpResponseBodyFeature>(new StreamResponseBodyFeature(pipeWriter.AsStream()));

            // 在响应发送前添加自定义响应头
            context.Response.OnStarting(() =>
            {
                context.Response.Headers.TryAdd("X-Custom-MiddlewareA-Header", "MiddlewareA-Added");
                return Task.CompletedTask;
            });

            // 执行后续中间件（包含MiddlewareB），等待其写入响应到Pipe
            await _next(context);

            // 完成Pipe写入，准备读取内容
            await pipeWriter.CompleteAsync();

            // 从Pipe中读取响应内容
            var reader = pipe.Reader;
            var readResult = await reader.ReadAsync();
            var responseBytes = readResult.Buffer.ToArray();
            reader.AdvanceTo(readResult.Buffer.End);
            await reader.CompleteAsync();

            // 解析响应内容（示例为UTF-8字符串，可根据实际调整编码或格式）
            var responseContent = Encoding.UTF8.GetString(responseBytes);

            // 自定义判断条件：是否需要覆写响应
            bool shouldOverrideResponse = responseContent.Contains("TriggerOverride");

            if (shouldOverrideResponse)
            {
                // 重置响应状态，清除之前的响应头与内容标记
                context.Response.StatusCode = 200;
                context.Response.ContentLength = null;
                context.Response.Body = originalBody;

                // 写入自定义响应内容
                var customResponseBytes = Encoding.UTF8.GetBytes("Custom Response Generated by MiddlewareA");
                await context.Response.Body.WriteAsync(customResponseBytes);
            }
            else
            {
                // 将原始响应内容写回原始流，确保正常返回
                context.Response.Body = originalBody;
                await context.Response.Body.WriteAsync(responseBytes);
            }
        }
        finally
        {
            // 恢复原始响应流与特性，避免影响后续中间件或框架逻辑
            context.Response.Body = originalBody;
            if (originalBodyFeature != null)
            {
                context.Features.Set(originalBodyFeature);
            }
        }
    }
}

// 扩展方法：注册MiddlewareA
public static class MiddlewareAExtensions
{
    public static IApplicationBuilder UseMiddlewareA(this IApplicationBuilder app)
    {
        return app.UseMiddleware<MiddlewareA>();
    }
}

关键细节说明 #

• Pipe的使用时机：必须在调用await next()之后读取Pipe内容，因为next()才会触发后续中间件处理请求并生成响应。
• ResponseBodyFeature替换：替换IHttpResponseBodyFeature是为了适配ASP.NET Core内部的响应处理逻辑，避免部分场景下流不兼容的问题。
• 响应头设置位置：OnStarting()回调放在调用next()之前即可，该回调会在响应即将发送到客户端前执行，不受后续中间件响应处理的影响。
• 覆写响应的注意事项：覆写时必须重置StatusCode和ContentLength，并恢复原始响应流，避免客户端接收到混乱的响应数据。

内容的提问来源于stack exchange，提问作者user9297554