Fixing issue where response type was not getting set

wshaddix · Jul 16, 2018 · cffb94d · cffb94d
1 parent a49f3a1
commit cffb94d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/ExampleHandlers/Tracing.cs b/src/ExampleHandlers/Tracing.cs
@@ -21,6 +21,7 @@ public Task<MicroserviceMessage> HandleAsync(MicroserviceMessage msg)
                 // add a trace header to the msg
                 var traceId = Guid.NewGuid().ToString("N");
                 msg.RequestHeaders.Add(headerName, traceId);
+                msg.ResponseHeaders.Add(headerName, traceId);
 
                 Log.Information("Added trace id of {TraceId} to the message", traceId);
             }

diff --git a/src/Proxy/HttpResponseFactory.cs b/src/Proxy/HttpResponseFactory.cs
@@ -12,6 +12,9 @@ internal static async void PrepareResponseAsync(HttpResponse httpResponse,
             // set the status code
             httpResponse.StatusCode = message.ResponseStatusCode;
 
+            // set the response type
+            httpResponse.ContentType = message.ResponseContentType;
+
             // set any response headers
             foreach (var header in message.ResponseHeaders)
             {