fix: buffer reuse and release #917

Author: jeevatkm

client.go

@@ -1236,7 +1236,6 @@ func (c *Client) executeBefore(req *Request) error {
 		return wrapNoRetryErr(err)
 	}
 
-	req.RawRequest.Body = newRequestBodyReleaser(req.RawRequest.Body, req.bodyBuf)
 	return nil
 }
 

request.go

@@ -1030,6 +1030,7 @@ func (r *Request) Execute(method, url string) (*Response, error) {
 		r.Attempt = 1
 		resp, err = r.client.execute(r)
 		r.client.onErrorHooks(r, resp, unwrapNoRetryErr(err))
+		backToBufPool(r.bodyBuf)
 		return resp, unwrapNoRetryErr(err)
 	}
 
@@ -1059,6 +1060,7 @@ func (r *Request) Execute(method, url string) (*Response, error) {
 	}
 
 	r.client.onErrorHooks(r, resp, unwrapNoRetryErr(err))
+	backToBufPool(r.bodyBuf)
 	return resp, unwrapNoRetryErr(err)
 }
 

request_test.go

@@ -2203,10 +2203,8 @@ func TestRequestGH917(t *testing.T) {
 	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		b, err := io.ReadAll(r.Body)
 		assertError(t, err)
-		if len(b) > 0 {
-			// sometimes, the body is "testtest" instead of "test"
-			assertEqual(t, "test", string(b))
-		}
+		assertEqual(t, "test", string(b))
+
 		w.WriteHeader(http.StatusInternalServerError)
 	}))
 

util.go

@@ -19,7 +19,6 @@ import (
 	"runtime"
 	"sort"
 	"strings"
-	"sync"
 )
 
 //‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾
@@ -297,38 +296,17 @@ func acquireBuffer() *bytes.Buffer {
 
 func releaseBuffer(buf *bytes.Buffer) {
 	if buf != nil {
+		buf.Reset()
 		bufPool.Put(buf)
 	}
 }
 
-// requestBodyReleaser wraps requests's body and implements custom Close for it.
-// The Close method closes original body and releases request body back to sync.Pool.
-type requestBodyReleaser struct {
-	releaseOnce sync.Once
-	reqBuf      *bytes.Buffer
-	io.ReadCloser
-}
-
-func newRequestBodyReleaser(respBody io.ReadCloser, reqBuf *bytes.Buffer) io.ReadCloser {
-	if reqBuf == nil {
-		return respBody
-	}
-
-	return &requestBodyReleaser{
-		reqBuf:     reqBuf,
-		ReadCloser: respBody,
+func backToBufPool(buf *bytes.Buffer) {
+	if buf != nil {
+		bufPool.Put(buf)
 	}
 }
 
-func (rr *requestBodyReleaser) Close() error {
-	err := rr.ReadCloser.Close()
-	rr.releaseOnce.Do(func() {
-		releaseBuffer(rr.reqBuf)
-	})
-
-	return err
-}
-
 func closeq(v interface{}) {
 	if c, ok := v.(io.Closer); ok {
 		silently(c.Close())