ability to set Header

2019-02-22 19:51:04 -08:00
parent e4150e8085
commit d81851e767
1 changed files with 8 additions and 0 deletions
@@ -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()