faas-rs/faas
1
0
Fork 0
mirror of https://github.com/openfaas/faas.git synced 2025-06-16 21:06:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement header pass-through

Browse Source 
Signed-off-by: Alex Ellis <alexellis2@gmail.com>
This commit is contained in:
Alex Ellis 2017-10-20 18:49:06 +02:00
parent 6d5ae1af5b
commit 4d4a09fa19
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
gateway/handlers/proxy.go
View File

@ -168,14 +168,22 @@ func invokeService(w http.ResponseWriter, r *http.Request, metrics metrics.Metri
clientHeader := w.Header() clientHeader := w.Header()
copyHeaders(&clientHeader, &response.Header) copyHeaders(&clientHeader, &response.Header)
header := response.Header.Get("Content-Type") responseHeader := response.Header.Get("Content-Type")
if len(header) > 0 { requestHeader := r.Header.Get("Content-Type")
w.Header().Set("Content-Type", response.Header.Get("Content-Type")) defaultHeader := "text/plain"
contentTypeField := "Content-Type"
fmt.Printf("Req %s Res %s\n", requestHeader, responseHeader)
if len(responseHeader) > 0 {
w.Header().Set(contentTypeField, responseHeader)
} else if len(requestHeader) > 0 {
w.Header().Set(contentTypeField, requestHeader)
} else { } else {
w.Header().Set("Content-Type", r.Header.Get("Content-Type")) w.Header().Set(contentTypeField, defaultHeader)
} }
writeHead(service, metrics, http.StatusOK, w) writeHead(service, metrics, response.StatusCode, w)
w.Write(responseBody) w.Write(responseBody)
} }