transport/local: configurable dial_timeout for connect, default 2s

transport/local/connect_local.go

@@ -12,6 +12,7 @@ import (
 type LocalConnecter struct {
 	listenerName   string
 	clientIdentity string
+	dialTimeout    time.Duration
 }
 
 func LocalConnecterFromConfig(in *config.LocalConnect) (*LocalConnecter, error) {
@@ -21,13 +22,24 @@ func LocalConnecterFromConfig(in *config.LocalConnect) (*LocalConnecter, error)
 	if in.ListenerName == "" {
 		return nil, fmt.Errorf("ListenerName must not be empty")
 	}
-	return &LocalConnecter{listenerName: in.ListenerName, clientIdentity: in.ClientIdentity}, nil
+	if in.DialTimeout < 0 {
+		return nil, fmt.Errorf("DialTimeout must be zero or positive")
+	}
+	cn := &LocalConnecter{
+		listenerName:   in.ListenerName,
+		clientIdentity: in.ClientIdentity,
+		dialTimeout:    in.DialTimeout,
+	}
+	return cn, nil
 }
 
 func (c *LocalConnecter) Connect(dialCtx context.Context) (transport.Wire, error) {
 	l := GetLocalListener(c.listenerName)
-	dialCtx, cancel := context.WithTimeout(dialCtx, 1*time.Second) // fail fast, config error by user is very likely
-	defer cancel()
+	if c.dialTimeout > 0 {
+		ctx, cancel := context.WithTimeout(dialCtx, c.dialTimeout)
+		defer cancel()
+		dialCtx = ctx // shadow
+	}
 	w, err := l.Connect(dialCtx, c.clientIdentity)
 	if err == context.DeadlineExceeded {
 		return nil, fmt.Errorf("local listener %q not reachable", c.listenerName)