mirror
/
gin
mirror of https://github.com/gin-gonic/gin.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

improve the context.MultipartForm method

This commit is contained in:
cncal 2023-03-11 15:51:52 +08:00
parent a889c58de7
commit 04e75f4712
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
context.go
View File

@ -591,8 +591,12 @@ func (c *Context) FormFile(name string) (*multipart.FileHeader, error) {
// MultipartForm is the parsed multipart form, including file uploads.
func (c *Context) MultipartForm() (*multipart.Form, error) {
err := c.Request.ParseMultipartForm(c.engine.MaxMultipartMemory)
return c.Request.MultipartForm, err
if c.Request.MultipartForm == nil {
if err := c.Request.ParseMultipartForm(c.engine.MaxMultipartMemory); err != nil {
return nil, err
}
}
return c.Request.MultipartForm, nil
}
// SaveUploadedFile uploads the form file to specific dst.

2
context_test.go
View File

@ -107,7 +107,7 @@ func TestContextMultipartForm(t *testing.T) {
if assert.NoError(t, err) {
assert.NotNil(t, f)
}
assert.Equal(t, c.PostForm("foo"), "bar")
assert.NoError(t, c.SaveUploadedFile(f.File["file"][0], "test"))
}