mirror of
https://github.com/infinite-iroha/touka.git
synced 2026-02-03 00:41:10 +08:00
fix limitMaxSizeReader non use body close & fix mergeCtx Value
This commit is contained in:
parent
cb86cb935a
commit
49508b49c1
2 changed files with 14 additions and 8 deletions
12
context.go
12
context.go
|
|
@ -456,6 +456,12 @@ func (c *Context) GetReqBodyFull() ([]byte, error) {
|
|||
}()
|
||||
} else {
|
||||
limitBytesReader = c.Request.Body
|
||||
defer func() {
|
||||
err := limitBytesReader.Close()
|
||||
if err != nil {
|
||||
c.AddError(fmt.Errorf("failed to close request body: %w", err))
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
data, err := copyb.ReadAll(limitBytesReader)
|
||||
|
|
@ -484,6 +490,12 @@ func (c *Context) GetReqBodyBuffer() (*bytes.Buffer, error) {
|
|||
}()
|
||||
} else {
|
||||
limitBytesReader = c.Request.Body
|
||||
defer func() {
|
||||
err := limitBytesReader.Close()
|
||||
if err != nil {
|
||||
c.AddError(fmt.Errorf("failed to close request body: %w", err))
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
data, err := copyb.ReadAll(limitBytesReader)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue