56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
/*
|
||
* @FileName: body.go
|
||
* @Author: JJXu
|
||
* @CreateTime: 2022/3/2 上午1:23
|
||
* @Description:
|
||
*/
|
||
|
||
package simpleRequest
|
||
|
||
// 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__"
|
||
)
|
||
|
||
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
|
||
}
|