added WithDialTimeout

networklore · Apr 16, 2021 · 5a4f8cb · 5a4f8cb
1 parent 116372e
commit 5a4f8cb
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/netrasp/netrasp.go b/pkg/netrasp/netrasp.go
@@ -7,6 +7,7 @@ package netrasp
 import (
 	"errors"
 	"fmt"
+	"time"
 
 	"golang.org/x/crypto/ssh"
 )
@@ -135,3 +136,10 @@ func WithSSHKeyExchange(name string) ConfigOpt {
 		c.SSHConfig.KeyExchanges = append(c.SSHConfig.KeyExchanges, name)
 	})
 }
+
+// WithDialTimeout allows you to configure timeout for dialing SSH server
+func WithDialTimeout(t time.Duration) ConfigOpt {
+	return newFuncConfigOpt(func(c *config) {
+		c.SSHConfig.Timeout = t
+	})
+}