Response#Write wites head in case it is not written

Signed-off-by: Vishal Rana <vr@labstack.com>
This commit is contained in:
Vishal Rana
2016-06-03 07:56:16 -07:00
parent 98a822e8d7
commit 02f12767fd
2 changed files with 6 additions and 0 deletions
+3
View File
@@ -52,6 +52,9 @@ func (r *Response) WriteHeader(code int) {
// Write implements `engine.Response#Write` function.
func (r *Response) Write(b []byte) (n int, err error) {
if !r.Committed() {
r.WriteHeader(http.StatusOK)
}
n, err = r.writer.Write(b)
r.size += int64(n)
return
+3
View File
@@ -56,6 +56,9 @@ func (r *Response) WriteHeader(code int) {
// Write implements `engine.Response#Write` function.
func (r *Response) Write(b []byte) (n int, err error) {
if !r.Committed() {
r.WriteHeader(http.StatusOK)
}
n, err = r.writer.Write(b)
r.size += int64(n)
return