优化UserAgent的输入

This commit is contained in:
Yun
2024-01-24 23:27:20 +08:00
parent eeef2edd6f
commit e1eec65937
3 changed files with 17 additions and 4 deletions
+2 -2
View File
@@ -14,8 +14,8 @@ type clientOptions struct {
func defaultOptions() clientOptions {
return clientOptions{
TimeOut: time.Second * 120, // 默认超时120
Logger: defaultLogger{},
TimeOut: time.Second * 120, // 默认超时120
Logger: defaultLogger{},
}
}
+12 -1
View File
@@ -89,7 +89,18 @@ func SetParamsFormFile(fieldName, fileName string, fileBytes []byte) Param {
*/
func SetParamsHeaders(h map[string]interface{}) Param {
return func(param *ClientParams) {
param.Headers = h
for key, _ := range h {
param.Headers[key] = h[key]
}
}
}
/**
* 设置UserAgent
*/
func SetUserAgent(userAgent UserAgent) Param {
return func(param *ClientParams) {
param.Headers["User-Agent"] = string(userAgent)
}
}
+3 -1
View File
@@ -40,7 +40,7 @@ func (p *ClientParams) parseUrl() error {
func (p *ClientParams) parseHeaders(r *http.Request) {
if p.Headers != nil {
if r.Header.Get("User-Agent") == "" {
r.Header.Add("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:87.0) Gecko/20100101 Firefox/87.0 Send By Golang")
r.Header.Add("User-Agent", string(UserAgentChrome))
}
for k, v := range p.Headers {
switch value := v.(type) {
@@ -52,6 +52,8 @@ func (p *ClientParams) parseHeaders(r *http.Request) {
}
case ContentType:
r.Header.Set(k, string(value))
case UserAgent:
r.Header.Set(k, string(value))
}
}
}