Merge pull request #83 from Chia-Network/fix-initial-connect

Chia-Network · Feb 26, 2023 · d8ef573 · d8ef573
2 parents e46f471 + 230666c
commit d8ef573
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pkg/websocketclient/websocketclient.go b/pkg/websocketclient/websocketclient.go
@@ -157,9 +157,14 @@ func (c *WebsocketClient) Subscribe(service string) error {
 	if _, alreadySet := c.subscriptions[service]; alreadySet {
 		return nil
 	}
-	c.subscriptions[service] = true
 
-	return c.doSubscribe(service)
+	err := c.doSubscribe(service)
+	if err != nil {
+		return err
+	}
+
+	c.subscriptions[service] = true
+	return nil
 }
 
 func (c *WebsocketClient) doSubscribe(service string) error {