优化UserAgent的输入
This commit is contained in:
+2
-2
@@ -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{},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -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
@@ -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))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user