bug fix: ConnResponseWriter writes headers immediately

Author: orkunkaraduman

connresponsewriter.go

@@ -32,7 +32,7 @@ func (c *ConnResponseWriter) Header() http.Header {
 func (c *ConnResponseWriter) Write(body []byte) (int, error) {
 	c.mu.Lock()
 	defer c.mu.Unlock()
-	c.WriteHeader(http.StatusOK)
+	c.writeHeader(http.StatusOK)
 	if c.err != nil {
 		return 0, c.err
 	}
@@ -47,6 +47,15 @@ func (c *ConnResponseWriter) Write(body []byte) (int, error) {
 func (c *ConnResponseWriter) WriteHeader(statusCode int) {
 	c.mu.Lock()
 	defer c.mu.Unlock()
+	c.writeHeader(statusCode)
+}
+
+// Close closes network connection.
+func (c *ConnResponseWriter) Close() error {
+	return c.Conn.Close()
+}
+
+func (c *ConnResponseWriter) writeHeader(statusCode int) {
 	if c.err != nil {
 		return
 	}
@@ -71,8 +80,3 @@ func (c *ConnResponseWriter) WriteHeader(statusCode int) {
 	}
 	c.headersSent = true
 }
-
-// Close closes network connection.
-func (c *ConnResponseWriter) Close() error {
-	return c.Conn.Close()
-}