update multpart form parser
This commit is contained in:
parent
db761d975d
commit
3795aecb68
48
parser.go
48
parser.go
@ -9,6 +9,7 @@ package simpleRequest
|
|||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"mime/multipart"
|
"mime/multipart"
|
||||||
"os"
|
"os"
|
||||||
@ -88,32 +89,45 @@ func (f *FormDataParser) Unmarshal(bodyType EntryMark, BodyEntry map[string]any)
|
|||||||
}
|
}
|
||||||
func multipartCommonParse(BodyEntry map[string]any) (reader io.Reader, contentType string) {
|
func multipartCommonParse(BodyEntry map[string]any) (reader io.Reader, contentType string) {
|
||||||
body := &bytes.Buffer{}
|
body := &bytes.Buffer{}
|
||||||
writer := multipart.NewWriter(body)
|
formWriter := multipart.NewWriter(body)
|
||||||
for k, sv := range BodyEntry {
|
for k, sv := range BodyEntry {
|
||||||
if strings.Contains(k, FormFilePathKey.string()) {
|
if strings.Contains(k, FormFilePathKey.string()) {
|
||||||
key := k[len(FormFilePathKey):]
|
key := k[len(FormFilePathKey):]
|
||||||
path := sv.(string)
|
fp := sv.(string)
|
||||||
filename := filepath.Base(path)
|
filename := filepath.Base(fp)
|
||||||
filePart, _ := writer.CreateFormFile(key, filename)
|
filePart, _ := formWriter.CreateFormFile(key, filename)
|
||||||
content, err := os.ReadFile(path)
|
content, err := os.ReadFile(fp)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
_, _ = filePart.Write(content)
|
_, _ = filePart.Write(content)
|
||||||
|
|
||||||
|
// way 2
|
||||||
|
file, err := os.Open(fp)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
fieldName := k[len(FormFilePathKey.string()):]
|
||||||
|
formPart, err := formWriter.CreateFormFile(fieldName, filepath.Base(fp))
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if _, err = io.Copy(formPart, file); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
switch sv.(type) {
|
switch multValue := sv.(type) {
|
||||||
case string:
|
case string:
|
||||||
strSv, _ := sv.(string)
|
_ = formWriter.WriteField(k, multValue)
|
||||||
_ = writer.WriteField(k, strSv)
|
|
||||||
case []string:
|
case []string:
|
||||||
sss, _ := sv.([]string)
|
sss, _ := sv.([]string)
|
||||||
for _, v := range sss {
|
for _, v := range sss {
|
||||||
_ = writer.WriteField(k, v)
|
_ = formWriter.WriteField(k, v)
|
||||||
}
|
}
|
||||||
case *multipart.FileHeader:
|
case *multipart.FileHeader:
|
||||||
file, _ := sv.(*multipart.FileHeader)
|
filePart, _ := formWriter.CreateFormFile(k, multValue.Filename)
|
||||||
filePart, _ := writer.CreateFormFile(k, file.Filename)
|
src, err := multValue.Open()
|
||||||
src, err := file.Open()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
return
|
return
|
||||||
@ -124,13 +138,17 @@ func multipartCommonParse(BodyEntry map[string]any) (reader io.Reader, contentTy
|
|||||||
panic(err)
|
panic(err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
case []byte:
|
||||||
|
formWriter.WriteField(k, string(multValue))
|
||||||
|
case int:
|
||||||
|
formWriter.WriteField(k, fmt.Sprintf("%v", multValue))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
err := writer.Close()
|
err := formWriter.Close()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
return body, writer.FormDataContentType()
|
return body, formWriter.FormDataContentType()
|
||||||
}
|
}
|
@ -262,7 +262,7 @@ func (s *SimpleRequest) initBody() {
|
|||||||
s.body = parser.Unmarshal(s.BodyEntryMark, s.BodyEntries)
|
s.body = parser.Unmarshal(s.BodyEntryMark, s.BodyEntries)
|
||||||
fdParser := parser.(*FormDataParser)
|
fdParser := parser.(*FormDataParser)
|
||||||
s.headers.Set("Content-Type", fdParser.ContentType)
|
s.headers.Set("Content-Type", fdParser.ContentType)
|
||||||
|
|
||||||
case IsXMLType(contentTypeData):
|
case IsXMLType(contentTypeData):
|
||||||
//application/soap+xml ,application/xml
|
//application/soap+xml ,application/xml
|
||||||
var parser, ok = s.bodyEntryParsers[xmlDataType]
|
var parser, ok = s.bodyEntryParsers[xmlDataType]
|
||||||
|
Loading…
Reference in New Issue
Block a user