chore: adding timeout to ssh config

rudderlabs · Feb 20, 2023 · c93ac44 · c93ac44
1 parent 995cd3f
commit c93ac44
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/services/streammanager/kafka/client/client.go b/services/streammanager/kafka/client/client.go
@@ -72,7 +72,11 @@ func New(network string, addresses []string, conf Config) (*Client, error) {
 			return nil, fmt.Errorf("cannot parse SSH private key: %w", err)
 		}
 
-		sshConfig := &ssh.ClientConfig{User: conf.SSHConfig.User, Auth: []ssh.AuthMethod{ssh.PublicKeys(signer)}}
+		sshConfig := &ssh.ClientConfig{
+			User:    conf.SSHConfig.User,
+			Auth:    []ssh.AuthMethod{ssh.PublicKeys(signer)},
+			Timeout: conf.DialTimeout,
+		}
 		if conf.SSHConfig.AcceptAnyHostKey {
 			sshConfig.HostKeyCallback = ssh.InsecureIgnoreHostKey()
 		}