Procházet zdrojové kódy

Add a default handshake timeout of 5 seconds

Pablo Carranza před 7 roky
rodič
revize
196b8d0585
1 změnil soubory, kde provedl 5 přidání a 2 odebrání
  1. 5 2
      client.go

+ 5 - 2
client.go

@@ -41,6 +41,7 @@ func NewClient(netConn net.Conn, u *url.URL, requestHeader http.Header, readBufS
 		NetDial: func(net, addr string) (net.Conn, error) {
 			return netConn, nil
 		},
+		HandshakeTimeout: 5 * time.Second,
 	}
 	return d.Dial(u.String(), requestHeader)
 }
@@ -106,7 +107,8 @@ func hostPortNoPort(u *url.URL) (hostPort, hostNoPort string) {
 
 // DefaultDialer is a dialer with all fields set to the default values.
 var DefaultDialer = &Dialer{
-	Proxy: http.ProxyFromEnvironment,
+	Proxy:            http.ProxyFromEnvironment,
+	HandshakeTimeout: 5 * time.Second,
 }
 
 // Dial creates a new client connection. Use requestHeader to specify the
@@ -122,7 +124,8 @@ func (d *Dialer) Dial(urlStr string, requestHeader http.Header) (*Conn, *http.Re
 
 	if d == nil {
 		d = &Dialer{
-			Proxy: http.ProxyFromEnvironment,
+			Proxy:            http.ProxyFromEnvironment,
+			HandshakeTimeout: 5 * time.Second,
 		}
 	}