Add Response Headers to ASP.NET Core Middleware

Question

I want to add a processing time middleware to my ASP.NET Core WebApi like this

public class ProcessingTimeMiddleware   {     private readonly RequestDelegate _next;      public ProcessingTimeMiddleware(RequestDelegate next)     {         _next = next;     }      public async Task Invoke(HttpContext context)     {         var watch = new Stopwatch();         watch.Start();          await _next(context);          context.Response.Headers.Add("X-Processing-Time-Milliseconds", new[] { watch.ElapsedMilliseconds.ToString() });     } } 

But doing this throws an exception saying

 System.InvalidOperationException: Headers are readonly, reponse has already started. 

How can I add headers to the response?

Answer 1

Never mind, the code is here

    public async Task Invoke(HttpContext context)     {         var watch = new Stopwatch();         watch.Start();          //To add Headers AFTER everything you need to do this         context.Response.OnStarting(state => {             var httpContext = (HttpContext)state;             httpContext.Response.Headers.Add("X-Response-Time-Milliseconds", new[] { watch.ElapsedMilliseconds.ToString() });              return Task.CompletedTask;         }, context);          await _next(context);     } 

Answer 2

Response headers can't be set after anything has been written to the response body.Once you pass the request to next middleware and it writes to the Response, then the Middleware can't set the Response headers again.

However, there is a solution available using a Callback method.

Microsoft.AspNetCore.Http.HttpResponse defines the OnStarting Method, which Adds a delegate to be invoked just before response headers will be sent to the client. You can think this method as a callback method that will be called right before writing to the response starts.

public class ResponseTimeMiddleware     {         private const string RESPONSE_HEADER_RESPONSE_TIME = "X-Response-Time-ms";          private readonly RequestDelegate _next;          public ResponseTimeMiddleware(RequestDelegate next)         {             _next = next;         }          public Task InvokeAsync(HttpContext context)         {             var watch = new Stopwatch();             watch.Start();              context.Response.OnStarting(() =>              {                 watch.Stop();                 var responseTimeForCompleteRequest = watch.ElapsedMilliseconds;                 context.Response.Headers[RESPONSE_HEADER_RESPONSE_TIME] =  responseTimeForCompleteRequest.ToString();                  return Task.CompletedTask;             });              // Call the next delegate/middleware in the pipeline             return this._next(context);         }     }