fix break of in dial

internal/pool/pool.go

@@ -319,26 +319,22 @@ func (p *ConnPool) dialConn(ctx context.Context, pooled bool) (*Conn, error) {
 	const backoffDuration = 100 * time.Millisecond
 
 	var lastErr error
-	for attempt := 0; attempt < maxRetries; attempt++ {
+	shouldLoop := true
-		// Add backoff delay for retry attempts
+	// when the timeout is reached, we should stop retrying
-		// (not for the first attempt, do at least one)
+	// but keep the lastErr to return to the caller
-		if attempt > 0 {
+	// instead of a generic context deadline exceeded error
-			select {
+	for attempt := 0; (attempt < maxRetries) && shouldLoop; attempt++ {
-			case <-ctx.Done():
-				// we should have lastErr set, but just in case
-				if lastErr == nil {
-					lastErr = ctx.Err()
-				}
-				break
-			case <-time.After(backoffDuration):
-				// Continue with retry
-			}
-		}
-
 		netConn, err := p.cfg.Dialer(ctx)
 		if err != nil {
 			lastErr = err
-			// Continue to next retry attempt
+			// Add backoff delay for retry attempts
+			// (not for the first attempt, do at least one)
+			select {
+			case <-ctx.Done():
+				shouldLoop = false
+			case <-time.After(backoffDuration):
+				// Continue with retry
+			}
 			continue
 		}