Merge pull request #11 from dcormier/master

Allow request headers to be included in initial connection to server
This commit is contained in:
Gennadiy Kovalev 2017-03-17 16:52:37 +02:00 committed by GitHub
commit fe537904e1
1 changed files with 3 additions and 1 deletions

View File

@ -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
}