Support negative Content-Length in DataFromReader (#1981)

You can get an http.Response with ContentLength set to -1 (Chunked encoding), so
for DataFromReader to be useful for those we need to support that.
This commit is contained in:
Segev Finer
2019-08-05 04:42:59 +03:00
committed by thinkerou
parent 461df9320a
commit 20440b96b9
2 changed files with 26 additions and 1 deletions
+3 -1
View File
@@ -21,7 +21,9 @@ type Reader struct {
// Render (Reader) writes data with custom ContentType and headers.
func (r Reader) Render(w http.ResponseWriter) (err error) {
r.WriteContentType(w)
r.Headers["Content-Length"] = strconv.FormatInt(r.ContentLength, 10)
if r.ContentLength >= 0 {
r.Headers["Content-Length"] = strconv.FormatInt(r.ContentLength, 10)
}
r.writeHeaders(w, r.Headers)
_, err = io.Copy(w, r.Reader)
return