set Content-Length header before writing body

gojek · Nov 4, 2020 · 7664d78 · 7664d78
1 parent 43246b9
commit 7664d78
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/handler/image.go b/internal/handler/image.go
@@ -58,8 +58,8 @@ func ImageHandler(deps *service.Dependencies) http.HandlerFunc {
 		w.Header().Set(CacheControlHeader, fmt.Sprintf("public,max-age=%d", config.CacheTime()))
 		// Ref to Google CDN we support: https://cloud.google.com/cdn/docs/caching#cacheability
 		w.Header().Set(VaryHeader, "Accept")
+		w.Header().Set(ContentLengthHeader, fmt.Sprintf("%d", len(data)))
 
-		cl, _ := w.Write(data)
-		w.Header().Set(ContentLengthHeader, fmt.Sprintf("%d", cl))
+		_, _ = w.Write(data)
 	}
 }