优化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 { func defaultOptions() clientOptions {
return clientOptions{ return clientOptions{
TimeOut: time.Second * 120, // 默认超时120 TimeOut: time.Second * 120, // 默认超时120
Logger: defaultLogger{}, 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 { func SetParamsHeaders(h map[string]interface{}) Param {
return func(param *ClientParams) { 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) { func (p *ClientParams) parseHeaders(r *http.Request) {
if p.Headers != nil { if p.Headers != nil {
if r.Header.Get("User-Agent") == "" { 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 { for k, v := range p.Headers {
switch value := v.(type) { switch value := v.(type) {
@@ -52,6 +52,8 @@ func (p *ClientParams) parseHeaders(r *http.Request) {
} }
case ContentType: case ContentType:
r.Header.Set(k, string(value)) r.Header.Set(k, string(value))
case UserAgent:
r.Header.Set(k, string(value))
} }
} }
} }