Set 10s TCP dial timeout on GorillaDialer via NetDialContext

transport/socket.go

@@ -3,6 +3,7 @@ package transport
 import (
 	"context"
 	"log/slog"
+	"net"
 	"net/http"
 	"time"
 
@@ -18,12 +19,13 @@ type GorillaDialer struct {
 	*websocket.Dialer
 }
 
+const dialTimeout = 10 * time.Second
+
 func NewGorillaDialer() *GorillaDialer {
+	netDialer := &net.Dialer{Timeout: dialTimeout}
 	return &GorillaDialer{
 		Dialer: &websocket.Dialer{
-			HandshakeTimeout: 45 * time.Second,
-			ReadBufferSize:   1024,
-			WriteBufferSize:  1024,
+			NetDialContext: netDialer.DialContext,
 		},
 	}
 }

transport/socket_test.go

@@ -27,9 +27,7 @@ func TestNewGorillaDialer(t *testing.T) {
 
 	assert.NotNil(t, dialer)
 	assert.NotNil(t, dialer.Dialer)
-	assert.Equal(t, 45*time.Second, dialer.Dialer.HandshakeTimeout)
-	assert.Equal(t, 1024, dialer.Dialer.ReadBufferSize)
-	assert.Equal(t, 1024, dialer.Dialer.WriteBufferSize)
+	assert.NotNil(t, dialer.Dialer.NetDialContext)
 }
 
 func TestAcquireSocket(t *testing.T) {