From 14925171deeb68a17a3be098d79d48e7b54568e5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lucas=20L=C3=B6ffel?= Date: Tue, 9 Oct 2018 15:01:06 +0200 Subject: [PATCH] added hasSubscribeHandler --- recws.go | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/recws.go b/recws.go index ba97de5..5d0e351 100644 --- a/recws.go +++ b/recws.go @@ -280,6 +280,13 @@ func (rc *RecConn) getBackoff() *backoff.Backoff { } } +func (rc *RecConn) hasSubscribeHandler() bool { + rc.mu.RLock() + defer rc.mu.RUnlock() + + return rc.SubscribeHandler != nil +} + func (rc *RecConn) connect() { b := rc.getBackoff() @@ -302,7 +309,7 @@ func (rc *RecConn) connect() { if !rc.getNonVerbose() { log.Printf("Dial: connection was successfully established with %s\n", rc.url) - if rc.SubscribeHandler == nil { + if !rc.hasSubscribeHandler() { return }