From f87ca56e18134f80f483d1e4990ea999f4abac97 Mon Sep 17 00:00:00 2001
From: kruskal <99559985+kruskall@users.noreply.github.com>
Date: Fri, 19 Jan 2024 16:31:07 +0100
Subject: [PATCH] feat: add ConnectContext to transport client

---
 transport/client.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/transport/client.go b/transport/client.go
index 6cf8f7d5..b6f1d8c8 100644
--- a/transport/client.go
+++ b/transport/client.go
@@ -89,6 +89,10 @@ func NewClientWithDialer(d Dialer, c Config, network, host string, defaultPort i
 }
 
 func (c *Client) Connect() error {
+	return c.ConnectContext(context.Background())
+}
+
+func (c *Client) ConnectContext(ctx context.Context) error {
 	c.mutex.Lock()
 	defer c.mutex.Unlock()
 
@@ -97,7 +101,7 @@ func (c *Client) Connect() error {
 		c.conn = nil
 	}
 
-	conn, err := c.dialer.DialContext(context.Background(), c.network, c.host)
+	conn, err := c.dialer.DialContext(ctx, c.network, c.host)
 	if err != nil {
 		return err
 	}