Dial without timeout if zero value is passed

gatewayd-io · Nov 17, 2023 · 1b53710 · 1b53710
1 parent 09bf6da
commit 1b53710
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/network/client.go b/network/client.go
@@ -86,7 +86,15 @@ func NewClient(ctx context.Context, clientConfig *config.Client, logger zerolog.
 	}
 
 	// Create a new connection.
-	conn, origErr := net.DialTimeout(client.Network, client.Address, client.DialTimeout)
+	var (
+		conn    net.Conn
+		origErr error
+	)
+	if client.DialTimeout == 0 {
+		conn, origErr = net.Dial(client.Network, client.Address)
+	} else {
+		conn, origErr = net.DialTimeout(client.Network, client.Address, client.DialTimeout)
+	}
 	if origErr != nil {
 		err := gerr.ErrClientConnectionFailed.Wrap(origErr)
 		logger.Error().Err(err).Msg("Failed to create a new connection")