add v5 public SubscribeKlines (#202)

Author: Sagleft

v5_ws_public.go

@@ -30,6 +30,11 @@ type V5WebsocketPublicServiceI interface {
 		func(V5WebsocketPublicKlineResponse) error,
 	) (func() error, error)
 
+	SubscribeKlines(
+		keys []V5WebsocketPublicKlineParamKey,
+		f func(V5WebsocketPublicKlineResponse) error,
+	) (func() error, error)
+
 	SubscribeTicker(
 		V5WebsocketPublicTickerParamKey,
 		func(V5WebsocketPublicTickerResponse) error,

v5_ws_public_kline.go

@@ -9,20 +9,41 @@ import (
 	"github.com/gorilla/websocket"
 )
 
+// SubscribeKlines :
+func (s *V5WebsocketPublicService) SubscribeKlines(
+	keys []V5WebsocketPublicKlineParamKey,
+	f func(V5WebsocketPublicKlineResponse) error,
+) (func() error, error) {
+	return s.subscribeKlines(keys, f)
+}
+
 // SubscribeKline :
 func (s *V5WebsocketPublicService) SubscribeKline(
 	key V5WebsocketPublicKlineParamKey,
 	f func(V5WebsocketPublicKlineResponse) error,
 ) (func() error, error) {
-	if err := s.addParamKlineFunc(key, f); err != nil {
-		return nil, err
+	return s.subscribeKlines([]V5WebsocketPublicKlineParamKey{key}, f)
+}
+
+func (s *V5WebsocketPublicService) subscribeKlines(
+	keys []V5WebsocketPublicKlineParamKey,
+	f func(V5WebsocketPublicKlineResponse) error,
+) (func() error, error) {
+	var args []interface{}
+	for _, key := range keys {
+		if err := s.addParamKlineFunc(key, f); err != nil {
+			return nil, err
+		}
+
+		args = append(args, key.Topic())
 	}
+
 	param := struct {
 		Op   string        `json:"op"`
 		Args []interface{} `json:"args"`
 	}{
 		Op:   "subscribe",
-		Args: []interface{}{key.Topic()},
+		Args: args,
 	}
 	buf, err := json.Marshal(param)
 	if err != nil {
@@ -37,7 +58,7 @@ func (s *V5WebsocketPublicService) SubscribeKline(
 			Args []interface{} `json:"args"`
 		}{
 			Op:   "unsubscribe",
-			Args: []interface{}{key.Topic()},
+			Args: args,
 		}
 		buf, err := json.Marshal(param)
 		if err != nil {
@@ -46,7 +67,10 @@ func (s *V5WebsocketPublicService) SubscribeKline(
 		if err := s.writeMessage(websocket.TextMessage, []byte(buf)); err != nil {
 			return err
 		}
-		s.removeParamKlineFunc(key)
+
+		for _, key := range keys {
+			s.removeParamKlineFunc(key)
+		}
 		return nil
 	}, nil
 }