* Fixed dial timeout processing

Author: asmyasnikov

CHANGELOG.md

@@ -10,6 +10,7 @@
 * Fixed default grpc dial options
 * Replaced single connection for discovery repeater into connection which creates each time for discovery request
 * Fixed retry of cluster discovery on initialization
+* Fixed dial timeout processing
 
 ## v3.42.5
 * Fixed closing of `database/sql` connection (aka `YDB` session)

connection.go

@@ -456,12 +456,6 @@ func connect(ctx context.Context, c *connection) error {
 		))
 	}
 
-	if t := c.config.DialTimeout(); t > 0 {
-		var cancel context.CancelFunc
-		ctx, cancel = context.WithTimeout(ctx, t)
-		defer cancel()
-	}
-
 	c.balancer, err = balancer.New(ctx,
 		c.config, c.pool,
 		append(

internal/balancer/balancer.go

@@ -90,7 +90,15 @@ func (b *Balancer) clusterDiscoveryAttempt(ctx context.Context) (err error) {
 		)
 	}()
 
-	childCtx, cancel := context.WithTimeout(ctx, b.driverConfig.DialTimeout())
+	var (
+		childCtx context.Context
+		cancel   context.CancelFunc
+	)
+	if dialTimeout := b.driverConfig.DialTimeout(); dialTimeout > 0 {
+		childCtx, cancel = context.WithTimeout(ctx, dialTimeout)
+	} else {
+		childCtx, cancel = context.WithCancel(ctx)
+	}
 	defer cancel()
 
 	client, err := b.discoveryClient(childCtx)
@@ -163,12 +171,6 @@ func New(
 	pool *conn.Pool,
 	opts ...discoveryConfig.Option,
 ) (b *Balancer, err error) {
-	if t := driverConfig.DialTimeout(); t > 0 {
-		var cancel context.CancelFunc
-		ctx, cancel = context.WithTimeout(ctx, t)
-		defer cancel()
-	}
-
 	var (
 		onDone = trace.DriverOnBalancerInit(
 			driverConfig.Trace(),