100 lines
2.6 KiB
Go
100 lines
2.6 KiB
Go
/*
|
||
*FileName: body.go
|
||
*Author: JJXu
|
||
*CreateTime: 2022/3/2 上午1:23
|
||
*Description:
|
||
*/
|
||
|
||
package simpleRequest
|
||
|
||
import (
|
||
"bytes"
|
||
"io"
|
||
"mime/multipart"
|
||
"strings"
|
||
)
|
||
|
||
// EntryMark 请求体条目标记，用于标记输入的body内容格式
|
||
type EntryMark string
|
||
|
||
func (b EntryMark) string() string {
|
||
return string(b)
|
||
}
|
||
|
||
const (
|
||
StringEntryType EntryMark = "__STRING_ENTRY__"
|
||
BytesEntryType EntryMark = "__BYTES_ENTRY__"
|
||
ModelEntryType EntryMark = "__MODEL_ENTRY__"
|
||
MapEntryType EntryMark = "__MAP_ENTRY__"
|
||
MultipartEntryType EntryMark = "__MULTIPART_ENTRY__"
|
||
FormFilePathKey EntryMark = "__FORM_FILE_PATH_KEY__"
|
||
)
|
||
|
||
func GetStringEntryTypeBody(bodyEntries map[string]any) io.Reader {
|
||
data, ok := bodyEntries[StringEntryType.string()]
|
||
if !ok || data == nil {
|
||
return nil
|
||
}
|
||
return strings.NewReader(data.(string))
|
||
}
|
||
func GetBytesEntryTypeBody(bodyEntries map[string]any) io.Reader {
|
||
data, ok := bodyEntries[BytesEntryType.string()]
|
||
if !ok || data == "" {
|
||
return nil
|
||
}
|
||
return bytes.NewReader(data.([]byte))
|
||
}
|
||
|
||
type BodyConf struct {
|
||
simpleReq *SimpleRequest
|
||
}
|
||
|
||
func (s *BodyConf) Set(key string, value any) *BodyConf {
|
||
s.simpleReq.BodyEntries[key] = value
|
||
return s
|
||
}
|
||
func (s *BodyConf) Sets(data map[string]any) *BodyConf {
|
||
s.simpleReq.BodyEntryMark = MapEntryType
|
||
for k, v := range data {
|
||
s.simpleReq.BodyEntries[k] = v
|
||
}
|
||
return s
|
||
}
|
||
func (s *BodyConf) SetString(strData string) *BodyConf {
|
||
s.simpleReq.BodyEntryMark = StringEntryType
|
||
s.simpleReq.BodyEntries[StringEntryType.string()] = strData
|
||
return s
|
||
}
|
||
|
||
func (s *BodyConf) SetBytes(byteData []byte) *BodyConf {
|
||
s.simpleReq.BodyEntryMark = BytesEntryType
|
||
s.simpleReq.BodyEntries[BytesEntryType.string()] = byteData
|
||
return s
|
||
}
|
||
|
||
func (s *BodyConf) SetModel(model any) *BodyConf {
|
||
s.simpleReq.BodyEntryMark = ModelEntryType
|
||
s.simpleReq.BodyEntries[ModelEntryType.string()] = model
|
||
return s
|
||
}
|
||
|
||
// 添加上传文件
|
||
func (s *BodyConf) SetFromDataFile(key, filePath string) *BodyConf {
|
||
s.simpleReq.BodyEntryMark = MultipartEntryType
|
||
s.simpleReq.BodyEntries[FormFilePathKey.string()+key] = filePath
|
||
if s.simpleReq.headers.Get(hdrContentTypeKey) == "" {
|
||
s.simpleReq.headers.Set(hdrContentTypeKey, formDataType)
|
||
}
|
||
return s
|
||
}
|
||
|
||
// 添加文件，适用于服务端文件转发场景，比如直接从c.FormFile("file")中获取FileHeader对象转发即可
|
||
func (s *BodyConf) SetFromDataMultipartFile(key string, multFile *multipart.FileHeader) *BodyConf {
|
||
s.simpleReq.BodyEntryMark = MultipartEntryType
|
||
s.simpleReq.BodyEntries[key] = multFile
|
||
if s.simpleReq.headers.Get(hdrContentTypeKey) == "" {
|
||
s.simpleReq.headers.Set(hdrContentTypeKey, formDataType)
|
||
}
|
||
return s
|
||
}
|