113 lines
2.7 KiB
Go
113 lines
2.7 KiB
Go
|
package starnet
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"io"
|
|||
|
|
"os"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// WithBody 设置请求体（字节）
|
|||
|
|
func WithBody(body []byte) RequestOpt {
|
|||
|
|
return func(r *Request) error {
|
|||
|
|
setBytesBodyConfig(&r.config.Body, body)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// WithBodyString 设置请求体（字符串）
|
|||
|
|
func WithBodyString(body string) RequestOpt {
|
|||
|
|
return func(r *Request) error {
|
|||
|
|
setBytesBodyConfig(&r.config.Body, []byte(body))
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// WithBodyReader 设置请求体（Reader）。
|
|||
|
|
// 出于避免重复写的保守策略，Reader 形态的 body 在非幂等方法上不会自动参与 retry。
|
|||
|
|
func WithBodyReader(reader io.Reader) RequestOpt {
|
|||
|
|
return func(r *Request) error {
|
|||
|
|
setReaderBodyConfig(&r.config.Body, reader)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// WithJSON 设置 JSON 请求体
|
|||
|
|
func WithJSON(v interface{}) RequestOpt {
|
|||
|
|
return func(r *Request) error {
|
|||
|
|
data, err := json.Marshal(v)
|
|||
|
|
if err != nil {
|
|||
|
|
return wrapError(err, "marshal json")
|
|||
|
|
}
|
|||
|
|
r.config.Headers.Set("Content-Type", ContentTypeJSON)
|
|||
|
|
setBytesBodyConfig(&r.config.Body, data)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// WithFormData 设置表单数据
|
|||
|
|
func WithFormData(data map[string][]string) RequestOpt {
|
|||
|
|
return func(r *Request) error {
|
|||
|
|
setFormBodyConfig(&r.config.Body, data)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// WithFormDataMap 设置表单数据（简化版）
|
|||
|
|
func WithFormDataMap(data map[string]string) RequestOpt {
|
|||
|
|
return func(r *Request) error {
|
|||
|
|
setFormBodyConfig(&r.config.Body, nil)
|
|||
|
|
for key, value := range data {
|
|||
|
|
r.config.Body.FormData[key] = []string{value}
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// WithAddFormData 添加表单数据
|
|||
|
|
func WithAddFormData(key, value string) RequestOpt {
|
|||
|
|
return func(r *Request) error {
|
|||
|
|
ensureFormMode(&r.config.Body)
|
|||
|
|
r.config.Body.FormData[key] = append(r.config.Body.FormData[key], value)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// WithFile 添加文件
|
|||
|
|
func WithFile(formName, filePath string) RequestOpt {
|
|||
|
|
return func(r *Request) error {
|
|||
|
|
stat, err := os.Stat(filePath)
|
|||
|
|
if err != nil {
|
|||
|
|
return wrapError(ErrFileNotFound, "file: %s", filePath)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ensureMultipartMode(&r.config.Body)
|
|||
|
|
r.config.Body.Files = append(r.config.Body.Files, RequestFile{
|
|||
|
|
FormName: formName,
|
|||
|
|
FileName: stat.Name(),
|
|||
|
|
FilePath: filePath,
|
|||
|
|
FileSize: stat.Size(),
|
|||
|
|
FileType: ContentTypeOctetStream,
|
|||
|
|
})
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// WithFileStream 添加文件流
|
|||
|
|
func WithFileStream(formName, fileName string, size int64, reader io.Reader) RequestOpt {
|
|||
|
|
return func(r *Request) error {
|
|||
|
|
if reader == nil {
|
|||
|
|
return ErrNilReader
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ensureMultipartMode(&r.config.Body)
|
|||
|
|
r.config.Body.Files = append(r.config.Body.Files, RequestFile{
|
|||
|
|
FormName: formName,
|
|||
|
|
FileName: fileName,
|
|||
|
|
FileData: reader,
|
|||
|
|
FileSize: size,
|
|||
|
|
FileType: ContentTypeOctetStream,
|
|||
|
|
})
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
}
|