fixup! Huobi: Add V2 websocket support

gbjk · Nov 6, 2024 · c2e73d5 · c2e73d5
1 parent c8231cf
commit c2e73d5
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/exchanges/huobi/huobi_test.go b/exchanges/huobi/huobi_test.go
@@ -1872,6 +1872,12 @@ func TestGenerateSubscriptions(t *testing.T) {
 	require.NoError(t, err, "generateSubscriptions must not error")
 	exp := subscription.List{}
 	for _, s := range h.Features.Subscriptions {
+		if s.Asset == asset.Empty {
+			s := s.Clone() //nolint:govet // Intentional lexical scope shadow
+			s.QualifiedChannel = channelName(s, nil)
+			exp = append(exp, s)
+			continue
+		}
 		for _, a := range h.GetAssetTypes(true) {
 			if s.Asset != asset.All && s.Asset != a {
 				continue

diff --git a/exchanges/huobi/huobi_websocket.go b/exchanges/huobi/huobi_websocket.go
@@ -690,7 +690,11 @@ func getErrResp(msg []byte) error {
 func channelName(s *subscription.Subscription, p *currency.Pair) string {
 	name := s.Channel
 	if n, ok := subscriptionNames[name]; ok {
-		return fmt.Sprintf(n, p)
+		if p != nil {
+			return fmt.Sprintf(n, p)
+		} else {
+			return n
+		}
 	}
 	panic(subscription.ErrPrivateChannelName)
 }