feat: add option to restore request body after reading

validate.go

@@ -6,6 +6,7 @@
 package validate
 
 import (
+	"bytes"
 	"encoding/json"
 	"io"
 	"net/http"
@@ -106,6 +107,10 @@ type GlobalOption struct {
 	//
 	// default: false
 	ValidatePrivateFields bool
+
+	// RestoreRequestBody Whether to restore the request body after reading it.
+	// default: false
+	RestoreRequestBody bool
 }
 
 // global options
@@ -396,6 +401,10 @@ func FromRequest(r *http.Request, maxMemoryLimit ...int64) (DataFace, error) {
 		if err != nil {
 			return nil, err
 		}
+		// restore request body
+		if gOpt.RestoreRequestBody {
+			r.Body = io.NopCloser(bytes.NewBuffer(bs))
+		}
 		return FromJSONBytes(bs)
 	}
 

validate_test.go

@@ -1,7 +1,10 @@
 package validate
 
 import (
+	"bytes"
 	"fmt"
+	"io"
+	"net/http"
 	"reflect"
 	"strings"
 	"testing"
@@ -274,3 +277,35 @@ func TestStruct_json_tag_name_parsing(t *testing.T) {
 	errStr = v.Errors["Field"]["email"]
 	assert.True(t, strings.HasPrefix(errStr, "Field "))
 }
+
+func TestValidation_RestoreRequestBody(t *testing.T) {
+	request, err := http.NewRequest(http.MethodPost, "/", bytes.NewBufferString(`{"test": "data"}`))
+	assert.Nil(t, err)
+	request.Header.Set("Content-Type", "application/json")
+
+	data, err := FromRequest(request)
+	assert.Nil(t, err)
+	assert.NotNil(t, data)
+
+	bs, err := io.ReadAll(request.Body)
+	assert.Nil(t, err)
+	assert.Empty(t, bs)
+
+	// restore body
+	Config(func(opt *GlobalOption) {
+		opt.RestoreRequestBody = true
+	})
+
+	request, err = http.NewRequest(http.MethodPost, "/", bytes.NewBufferString(`{"test": "data"}`))
+	assert.Nil(t, err)
+	request.Header.Set("Content-Type", "application/json")
+
+	data, err = FromRequest(request)
+	assert.Nil(t, err)
+	assert.NotNil(t, data)
+
+	bs, err = io.ReadAll(request.Body)
+	assert.Nil(t, err)
+	assert.Equal(t, `{"test": "data"}`, string(bs))
+
+}