From 9010d067c5ec2cd9e108b9e78033eed38cd9e13b Mon Sep 17 00:00:00 2001 From: Daniel Cormier Date: Fri, 18 Nov 2016 14:38:22 -0500 Subject: [PATCH] Allow request headers to be included in initial connection to server --- transport/websocket.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/transport/websocket.go b/transport/websocket.go index 42772f7..58da673 100644 --- a/transport/websocket.go +++ b/transport/websocket.go @@ -88,11 +88,13 @@ type WebsocketTransport struct { SendTimeout time.Duration BufferSize int + + RequestHeader http.Header } func (wst *WebsocketTransport) Connect(url string) (conn Connection, err error) { dialer := websocket.Dialer{} - socket, _, err := dialer.Dial(url, nil) + socket, _, err := dialer.Dial(url, wst.RequestHeader) if err != nil { return nil, err }