Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

pkg/transport: pass dial timeout to NewTransport

Browse Source 
So we could set dial timeout for new transport, which makes it
customizable according to max RTT.
This commit is contained in:
Yicheng Qin 2015-10-10 22:46:21 -07:00
parent 233e717e2f
commit 9673eb625a
4 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
etcdctl/command/util.go
View File

@ -23,6 +23,7 @@ import (
"net/url"
"os"
"strings"
"time"
"github.com/coreos/etcd/Godeps/_workspace/src/github.com/bgentry/speakeasy"
"github.com/coreos/etcd/Godeps/_workspace/src/github.com/codegangsta/cli"
@ -33,6 +34,10 @@ import (
var (
ErrNoAvailSrc = errors.New("no available argument and stdin")
// the maximum amount of time a dial will wait for a connection to setup.
// 30s is long enough for most of the network conditions.
defaultDialTimeout = 30 * time.Second
)
// trimsplit slices s into all substrings separated by sep and returns a
@ -153,7 +158,7 @@ func getTransport(c *cli.Context) (*http.Transport, error) {
CertFile: certfile,
KeyFile: keyfile,
}
return transport.NewTransport(tls)
return transport.NewTransport(tls, defaultDialTimeout)
}
func getUsernamePasswordFromFlag(usernameFlag string) (username string, password string, err error) {

7
pkg/transport/listener.go
View File

@ -46,18 +46,19 @@ func NewListener(addr string, scheme string, info TLSInfo) (net.Listener, error)
return l, nil
}
func NewTransport(info TLSInfo) (*http.Transport, error) {
func NewTransport(info TLSInfo, dialtimeoutd time.Duration) (*http.Transport, error) {
cfg, err := info.ClientConfig()
if err != nil {
return nil, err
}
t := &http.Transport{
// timeouts taken from http.DefaultTransport
Dial: (&net.Dialer{
Timeout: 30 * time.Second,
Timeout: dialtimeoutd,
// value taken from http.DefaultTransport
KeepAlive: 30 * time.Second,
}).Dial,
// value taken from http.DefaultTransport
TLSHandshakeTimeout: 10 * time.Second,
TLSClientConfig: cfg,
}

3
pkg/transport/listener_test.go
View File

@ -21,6 +21,7 @@ import (
"net/http"
"os"
"testing"
"time"
)
func createTempFile(b []byte) (string, error) {
@ -115,7 +116,7 @@ func TestNewTransportTLSInfo(t *testing.T) {
for i, tt := range tests {
tt.parseFunc = fakeCertificateParserFunc(tls.Certificate{}, nil)
trans, err := NewTransport(tt)
trans, err := NewTransport(tt, time.Second)
if err != nil {
t.Fatalf("Received unexpected error from NewTransport: %v", err)
}

2
pkg/transport/timeout_transport.go
View File

@ -24,7 +24,7 @@ import (
// If read/write on the created connection blocks longer than its time limit,
// it will return timeout error.
func NewTimeoutTransport(info TLSInfo, dialtimeoutd, rdtimeoutd, wtimeoutd time.Duration) (*http.Transport, error) {
tr, err := NewTransport(info)
tr, err := NewTransport(info, dialtimeoutd)
if err != nil {
return nil, err
}