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 }