diff --git a/client/websocket.go b/client/websocket.go index d4d51fdabb4..08c0f1384c8 100644 --- a/client/websocket.go +++ b/client/websocket.go @@ -39,11 +39,11 @@ func errorSubscription(err error) *Subscription { } } -func (p *Client) Websocket(query string, requestHeader http.Header, options ...Option) *Subscription { - return p.WebsocketWithPayload(query, nil, requestHeader, options...) +func (p *Client) Websocket(query string, reqHeaders http.Header, options ...Option) *Subscription { + return p.WebsocketWithPayload(query, nil, reqHeaders, options...) } -func (p *Client) WebsocketWithPayload(query string, initPayload map[string]interface{}, requestHeader http.Header, options ...Option) *Subscription { +func (p *Client) WebsocketWithPayload(query string, initPayload map[string]interface{}, reqHeaders http.Header, options ...Option) *Subscription { r := p.mkRequest(query, options...) requestBody, err := json.Marshal(r) if err != nil { @@ -53,7 +53,7 @@ func (p *Client) WebsocketWithPayload(query string, initPayload map[string]inter url := strings.Replace(p.url, "http://", "ws://", -1) url = strings.Replace(url, "https://", "wss://", -1) - c, resp, err := websocket.DefaultDialer.Dial(url, requestHeader) + c, resp, err := websocket.DefaultDialer.Dial(url, reqHeaders) if err != nil { return errorSubscription(fmt.Errorf("dial: %s", err.Error())) }