feat(protocol): allow override of dial func

2024-08-05 18:10:19 +02:00
parent b976bde363
commit 6842d4d88a
16 changed files with 158 additions and 18 deletions
--- a/reach/client/socketio/client.go
+++ b/reach/client/socketio/client.go
@ -32,12 +32,15 @@ func (c *Client) Connect() error {

 	wg.Add(1)

-	transport := &transport.WebsocketTransport{
-		PingInterval:   c.opts.PingInterval,
-		PingTimeout:    c.opts.PingTimeout,
-		ReceiveTimeout: c.opts.ReceiveTimeout,
-		SendTimeout:    c.opts.SendTimeout,
-		BufferSize:     c.opts.BufferSize,
+	transport := &Transport{
+		dial: c.opts.DialFunc,
+		ws: &transport.WebsocketTransport{
+			PingInterval:   c.opts.PingInterval,
+			PingTimeout:    c.opts.PingTimeout,
+			ReceiveTimeout: c.opts.ReceiveTimeout,
+			SendTimeout:    c.opts.SendTimeout,
+			BufferSize:     c.opts.BufferSize,
+		},
 	}

 	conn, err := gosocketio.Dial(c.endpoint, transport)