fix(timeout): make set/read timeout and write/put buffer atomic (#65)

writer.go

@@ -27,20 +27,23 @@ func NewWriter(w gin.ResponseWriter, buf *bytes.Buffer) *Writer {
 
 // Write will write data to response body
 func (w *Writer) Write(data []byte) (int, error) {
+	w.mu.Lock()
+	defer w.mu.Unlock()
+
 	if w.timeout || w.body == nil {
 		return 0, nil
 	}
 
-	w.mu.Lock()
-	defer w.mu.Unlock()
-
 	return w.body.Write(data)
 }
 
 // WriteHeader sends an HTTP response header with the provided status code.
 // If the response writer has already written headers or if a timeout has occurred,
 // this method does nothing.
 func (w *Writer) WriteHeader(code int) {
+	w.mu.Lock()
+	defer w.mu.Unlock()
+
 	if w.timeout || w.wroteHeaders {
 		return
 	}
@@ -53,9 +56,6 @@ func (w *Writer) WriteHeader(code int) {
 
 	checkWriteHeaderCode(code)
 
-	w.mu.Lock()
-	defer w.mu.Unlock()
-
 	w.writeHeader(code)
 	w.ResponseWriter.WriteHeader(code)
 }