add wrapper function DialWithTLSDialer to allow use of custom `net.…

…Dialer` (#336) Co-authored-by: John Weldon <[email protected]>
go-ldap · Aug 15, 2021 · a10e112 · a10e112
1 parent b09d2f3
commit a10e112
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/conn.go b/conn.go
@@ -130,6 +130,15 @@ func DialWithTLSConfig(tc *tls.Config) DialOpt {
 	}
 }
 
+// DialWithTLSDialer is a wrapper for DialWithTLSConfig with the option to
+// specify a net.Dialer to for example define a timeout or a custom resolver.
+func DialWithTLSDialer(tlsConfig *tls.Config, dialer *net.Dialer) DialOpt {
+	return func(dc *DialContext) {
+		dc.tc = tlsConfig
+		dc.d = dialer
+	}
+}
+
 // DialContext contains necessary parameters to dial the given ldap URL.
 type DialContext struct {
 	d  *net.Dialer

diff --git a/v3/conn.go b/v3/conn.go
@@ -130,6 +130,15 @@ func DialWithTLSConfig(tc *tls.Config) DialOpt {
 	}
 }
 
+// DialWithTLSDialer is a wrapper for DialWithTLSConfig with the option to
+// specify a net.Dialer to for example define a timeout or a custom resolver.
+func DialWithTLSDialer(tlsConfig *tls.Config, dialer *net.Dialer) DialOpt {
+	return func(dc *DialContext) {
+		dc.tc = tlsConfig
+		dc.d = dialer
+	}
+}
+
 // DialContext contains necessary parameters to dial the given ldap URL.
 type DialContext struct {
 	d  *net.Dialer