From d81851e767804518ce782841cc1a57fb948eb0e8 Mon Sep 17 00:00:00 2001 From: Roman Vasilyev Date: Fri, 22 Feb 2019 19:51:04 -0800 Subject: [PATCH] ability to set Header --- recws.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/recws.go b/recws.go index b003118..cc1e1ca 100644 --- a/recws.go +++ b/recws.go @@ -162,6 +162,13 @@ func (rc *RecConn) setURL(url string) { rc.url = url } +func (rc *RecConn) setReqHeader(reqHeader http.Header) { + rc.mu.Lock() + defer rc.mu.Unlock() + + rc.reqHeader = reqHeader +} + // parseURL parses current url func (rc *RecConn) parseURL(urlStr string) (string, error) { if urlStr == "" { @@ -251,6 +258,7 @@ func (rc *RecConn) Dial(urlStr string, reqHeader http.Header) { // Config rc.setURL(urlStr) + rc.setReqHeader(reqHeader) rc.setDefaultRecIntvlMin() rc.setDefaultRecIntvlMax() rc.setDefaultRecIntvlFactor()