update multpart form parser

This commit is contained in:
徐俊杰 2023-10-27 17:12:51 +08:00
parent db761d975d
commit 3795aecb68
2 changed files with 34 additions and 16 deletions

View File

@ -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()
} }

View File

@ -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]