optimize reading http request body

2020-08-10 18:27:08 +08:00
parent 952de5eacd
commit 40895ba8d9
1 changed files with 8 additions and 1 deletions
--- a/rest/handler/cryptionhandler.go
+++ b/rest/handler/cryptionhandler.go
@@ -42,7 +42,14 @@ func decryptBody(key []byte, r *http.Request) error {
 		return errContentLengthExceeded
 	}

-	content, err := ioutil.ReadAll(io.LimitReader(r.Body, maxBytes))
+	var content []byte
+	var err error
+	if r.ContentLength > 0 {
+		content = make([]byte, r.ContentLength, r.ContentLength)
+		_, err = io.ReadFull(r.Body, content)
+	} else {
+		content, err = ioutil.ReadAll(io.LimitReader(r.Body, maxBytes))
+	}
 	if err != nil {
 		return err
 	}