mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-23 17:20:24 +08:00
return ErrBodylessRequest on get method etc.
This commit is contained in:
parent
9428fface2
commit
05c8dd0b9c
@ -83,8 +83,13 @@ func ParseHeader(headerValue string) map[string]string {
|
||||
|
||||
// Parses the post request which contains json in body.
|
||||
func ParseJsonBody(r *http.Request, v interface{}) error {
|
||||
var reader io.Reader
|
||||
switch r.Method {
|
||||
case http.MethodDelete, http.MethodPatch, http.MethodPost, http.MethodPut:
|
||||
default:
|
||||
return ErrBodylessRequest
|
||||
}
|
||||
|
||||
var reader io.Reader
|
||||
if withJsonBody(r) {
|
||||
reader = io.LimitReader(r.Body, maxBodyLen)
|
||||
} else {
|
||||
|
@ -134,6 +134,26 @@ func BenchmarkParseRaw(b *testing.B) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestParseJsonBodyless(t *testing.T) {
|
||||
methods := []string{
|
||||
http.MethodConnect,
|
||||
http.MethodGet,
|
||||
http.MethodHead,
|
||||
http.MethodOptions,
|
||||
http.MethodTrace,
|
||||
}
|
||||
|
||||
for _, method := range methods {
|
||||
t.Run(method, func(t *testing.T) {
|
||||
r, err := http.NewRequest(http.MethodGet, "http://hello.com", nil)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
assert.Equal(t, ErrBodylessRequest, ParseJsonBody(r, nil))
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkParseAuto(b *testing.B) {
|
||||
r, err := http.NewRequest(http.MethodGet, "http://hello.com/a?name=hello&age=18&percent=3.4", nil)
|
||||
if err != nil {
|
||||
|
@ -1,5 +1,7 @@
|
||||
package httpx
|
||||
|
||||
import "errors"
|
||||
|
||||
const (
|
||||
ApplicationJson = "application/json"
|
||||
ContentEncoding = "Content-Encoding"
|
||||
@ -17,3 +19,5 @@ const (
|
||||
CodeSignatureWrongTime
|
||||
CodeSignatureInvalidToken
|
||||
)
|
||||
|
||||
var ErrBodylessRequest = errors.New("not a POST|PUT|PATCH request")
|
Loading…
Reference in New Issue
Block a user