starainrt
732e81316c
- 分离 Request 的配置态与执行态，修复二次 Do、raw 模式网络配置失效和 body 来源互斥问题 - 新增 starnet trace 抽象，补齐 DNS/连接/TLS/重试事件，并优化动态 transport 缓存与代理解析路径 - 收紧非法代理为 fail-fast，多目标目标回退仅限幂等请求，修复 Host/TLS/SNI 等语义边界 - 补充防御性拷贝、专项回归测试、本地代理/TLS 用例与 README 行为说明
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
|
||
}
|
||
}
|