add user-agent option for http/http2

This commit is contained in:
ginuerzh
2019-06-20 15:19:42 +08:00
parent bea815ec9a
commit e996e7c35b
6 changed files with 33 additions and 11 deletions
+5 -1
View File
@@ -37,6 +37,10 @@ func (c *httpConnector) Connect(conn net.Conn, addr string, options ...ConnectOp
if timeout <= 0 {
timeout = ConnectTimeout
}
ua := opts.UserAgent
if ua == "" {
ua = DefaultUserAgent
}
conn.SetDeadline(time.Now().Add(timeout))
defer conn.SetDeadline(time.Time{})
@@ -49,7 +53,7 @@ func (c *httpConnector) Connect(conn net.Conn, addr string, options ...ConnectOp
ProtoMinor: 1,
Header: make(http.Header),
}
req.Header.Set("User-Agent", DefaultUserAgent)
req.Header.Set("User-Agent", ua)
req.Header.Set("Proxy-Connection", "keep-alive")
user := opts.User