新增bodyentry解析模块,可自定义解析函数
parent
9bff109127
commit
f0327d0c93
@ -1,3 +1,3 @@
|
||||
module github.com/dorlolo/simpleRequest
|
||||
|
||||
go 1.17
|
||||
go 1.18
|
||||
|
@ -0,0 +1,17 @@
|
||||
// Package simpleRequest -----------------------------
|
||||
// @file : options.go
|
||||
// @author : JJXu
|
||||
// @contact : wavingBear@163.com
|
||||
// @time : 2022/12/10 01:45:37
|
||||
// -------------------------------------------
|
||||
package simpleRequest
|
||||
|
||||
type OPTION func(r *SimpleRequest) *SimpleRequest
|
||||
|
||||
//OptionNewBodyEntryParser 新增或覆盖BodyEntryParser
|
||||
func OptionNewBodyEntryParser(contentType string, parser IBodyEntryParser) OPTION {
|
||||
return func(r *SimpleRequest) *SimpleRequest {
|
||||
r.bodyEntryParsers[contentType] = parser
|
||||
return r
|
||||
}
|
||||
}
|
@ -0,0 +1,111 @@
|
||||
// Package simpleRequest -----------------------------
|
||||
// @file : parser.go
|
||||
// @author : JJXu
|
||||
// @contact : wavingBear@163.com
|
||||
// @time : 2022/12/10 00:48:45
|
||||
// -------------------------------------------
|
||||
package simpleRequest
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"io"
|
||||
"mime/multipart"
|
||||
"strings"
|
||||
)
|
||||
|
||||
var bodyEntryParsers = map[string]IBodyEntryParser{
|
||||
jsonContentType: new(JsonParser),
|
||||
formDataType: new(FormDataParser),
|
||||
}
|
||||
|
||||
type IBodyEntryParser interface {
|
||||
Unmarshal(bodyType EntryMark, BodyEntry map[string]any) io.Reader
|
||||
}
|
||||
|
||||
type JsonParser struct{}
|
||||
|
||||
func (JsonParser) Unmarshal(bodyType EntryMark, BodyEntry map[string]any) io.Reader {
|
||||
switch bodyType {
|
||||
case StringEntryType:
|
||||
return strings.NewReader(BodyEntry[StringEntryType.string()].(string))
|
||||
case BytesEntryType:
|
||||
return bytes.NewReader(BodyEntry[BytesEntryType.string()].([]byte))
|
||||
case ModelEntryType:
|
||||
jsonData, err := json.Marshal(BodyEntry)
|
||||
if err == nil {
|
||||
return bytes.NewReader(jsonData)
|
||||
} else {
|
||||
return strings.NewReader("{}")
|
||||
}
|
||||
case MapEntryType:
|
||||
jsonData, err := json.Marshal(BodyEntry)
|
||||
if err == nil {
|
||||
return bytes.NewReader(jsonData)
|
||||
} else {
|
||||
return strings.NewReader("{}")
|
||||
}
|
||||
default:
|
||||
if len(BodyEntry) > 0 {
|
||||
jsonData, err := json.Marshal(BodyEntry)
|
||||
if err == nil {
|
||||
return bytes.NewReader(jsonData)
|
||||
}
|
||||
}
|
||||
return strings.NewReader("{}")
|
||||
}
|
||||
}
|
||||
|
||||
type FormDataParser struct {
|
||||
ContentType string
|
||||
}
|
||||
|
||||
func (f *FormDataParser) Unmarshal(bodyType EntryMark, BodyEntry map[string]any) (body io.Reader) {
|
||||
switch bodyType {
|
||||
case MapEntryType:
|
||||
body, f.ContentType = multipartCommonParse(BodyEntry)
|
||||
case ModelEntryType:
|
||||
tb := BodyEntry[ModelEntryType.string()]
|
||||
buffer, err := json.Marshal(tb)
|
||||
if err != nil {
|
||||
panic(err.Error())
|
||||
}
|
||||
var mapper map[string]any
|
||||
err = json.Unmarshal(buffer, &mapper)
|
||||
if err != nil {
|
||||
panic(err.Error())
|
||||
}
|
||||
body, f.ContentType = multipartCommonParse(mapper)
|
||||
case StringEntryType:
|
||||
f.ContentType = formDataType
|
||||
return strings.NewReader(BodyEntry[StringEntryType.string()].(string))
|
||||
case BytesEntryType:
|
||||
f.ContentType = formDataType
|
||||
return bytes.NewReader(BodyEntry[StringEntryType.string()].([]byte))
|
||||
default:
|
||||
body, f.ContentType = multipartCommonParse(BodyEntry)
|
||||
}
|
||||
f.ContentType = formDataType
|
||||
return nil
|
||||
}
|
||||
func multipartCommonParse(BodyEntry map[string]any) (reader io.Reader, contentType string) {
|
||||
body := &bytes.Buffer{}
|
||||
writer := multipart.NewWriter(body)
|
||||
for k, sv := range BodyEntry {
|
||||
switch sv.(type) {
|
||||
case string:
|
||||
strSv, _ := sv.(string)
|
||||
_ = writer.WriteField(k, strSv)
|
||||
case []string:
|
||||
sss, _ := sv.([]string)
|
||||
for _, v := range sss {
|
||||
_ = writer.WriteField(k, v)
|
||||
}
|
||||
}
|
||||
}
|
||||
err := writer.Close()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return body, writer.FormDataContentType()
|
||||
}
|
Loading…
Reference in New Issue