refactor(render): remove headers parameter from writeHeader

This commit is contained in:
1911860538 2025-09-15 19:29:32 +08:00
parent f9bd00a6b7
commit 26b37c86ea

View File

@ -27,7 +27,7 @@ func (r Reader) Render(w http.ResponseWriter) (err error) {
} }
r.Headers["Content-Length"] = strconv.FormatInt(r.ContentLength, 10) r.Headers["Content-Length"] = strconv.FormatInt(r.ContentLength, 10)
} }
r.writeHeaders(w, r.Headers) r.writeHeaders(w)
_, err = io.Copy(w, r.Reader) _, err = io.Copy(w, r.Reader)
return return
} }
@ -37,10 +37,10 @@ func (r Reader) WriteContentType(w http.ResponseWriter) {
writeContentType(w, []string{r.ContentType}) writeContentType(w, []string{r.ContentType})
} }
// writeHeaders writes custom Header. // writeHeaders writes headers from r.Headers into response.
func (r Reader) writeHeaders(w http.ResponseWriter, headers map[string]string) { func (r Reader) writeHeaders(w http.ResponseWriter) {
header := w.Header() header := w.Header()
for k, v := range headers { for k, v := range r.Headers {
if header.Get(k) == "" { if header.Get(k) == "" {
header.Set(k, v) header.Set(k, v)
} }