Kraken: Fix RemoveSubscription calls on sub err

thrasher-corp · Nov 2, 2023 · 119daca · 119daca
1 parent 1be835b
commit 119daca
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/exchanges/kraken/kraken_websocket.go b/exchanges/kraken/kraken_websocket.go
@@ -1222,12 +1222,14 @@ func (k *Kraken) subscribeToChan(c *stream.ChannelSubscription) error {
 
 	respRaw, err := conn.SendMessageReturnResponse(r.RequestID, r)
 	if err != nil {
+		k.Websocket.RemoveSubscriptions(*c)
 		return fmt.Errorf("%w Channel: %s Pair: %s Error: %w", stream.ErrSubscriptionFailure, c.Channel, c.Currency, err)
 	}
 
 	if err = k.getErrResp(respRaw); err != nil {
 		wErr := fmt.Errorf("%w Channel: %s Pair: %s; %w", stream.ErrSubscriptionFailure, c.Channel, c.Currency, err)
 		k.Websocket.DataHandler <- wErr
+		k.Websocket.RemoveSubscriptions(*c)
 		return wErr
 	}