blackteay/gin
3
0
Fork 0
mirror of https://github.com/gin-gonic/gin.git synced 2025-10-14 20:22:20 +08:00
Code Issues Projects Releases Wiki Activity

improve the context.MultipartForm method

Browse Source
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. // MultipartForm is the parsed multipart form, including file uploads.
func (c *Context) MultipartForm() (*multipart.Form, error) { func (c *Context) MultipartForm() (*multipart.Form, error) {
err := c.Request.ParseMultipartForm(c.engine.MaxMultipartMemory) if c.Request.MultipartForm == nil {
return c.Request.MultipartForm, err 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. // 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) { if assert.NoError(t, err) {
assert.NotNil(t, f) assert.NotNil(t, f)
} }
assert.Equal(t, c.PostForm("foo"), "bar")
assert.NoError(t, c.SaveUploadedFile(f.File["file"][0], "test")) assert.NoError(t, c.SaveUploadedFile(f.File["file"][0], "test"))
} }