apply type

hirokisan · Apr 11, 2023 · 5f9611e · 5f9611e
1 parent 3aa742e
commit 5f9611e
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 9 deletions.
diff --git a/v5_ws_public.go b/v5_ws_public.go
@@ -64,15 +64,19 @@ type V5WebsocketPublicTopic string
 
 const (
 	// V5WebsocketPublicTopicOrderBook :
-	V5WebsocketPublicTopicOrderBook = "orderbook"
+	V5WebsocketPublicTopicOrderBook = V5WebsocketPublicTopic("orderbook")
 
 	// V5WebsocketPublicTopicKline :
-	V5WebsocketPublicTopicKline = "kline"
+	V5WebsocketPublicTopicKline = V5WebsocketPublicTopic("kline")
 
 	// V5WebsocketPublicTopicTicker :
-	V5WebsocketPublicTopicTicker = "tickers"
+	V5WebsocketPublicTopicTicker = V5WebsocketPublicTopic("tickers")
 )
 
+func (t V5WebsocketPublicTopic) String() string {
+	return string(t)
+}
+
 // judgeTopic :
 func (s *V5WebsocketPublicService) judgeTopic(respBody []byte) (V5WebsocketPublicTopic, error) {
 	parsedData := map[string]interface{}{}
@@ -81,11 +85,11 @@ func (s *V5WebsocketPublicService) judgeTopic(respBody []byte) (V5WebsocketPubli
 	}
 	if topic, ok := parsedData["topic"].(string); ok {
 		switch {
-		case strings.Contains(topic, "orderbook"):
+		case strings.Contains(topic, V5WebsocketPublicTopicOrderBook.String()):
 			return V5WebsocketPublicTopicOrderBook, nil
-		case strings.Contains(topic, "kline"):
+		case strings.Contains(topic, V5WebsocketPublicTopicKline.String()):
 			return V5WebsocketPublicTopicKline, nil
-		case strings.Contains(topic, "tickers"):
+		case strings.Contains(topic, V5WebsocketPublicTopicTicker.String()):
 			return V5WebsocketPublicTopicTicker, nil
 		}
 	}

diff --git a/v5_ws_public_kline.go b/v5_ws_public_kline.go
@@ -89,7 +89,7 @@ type V5WebsocketPublicKlineData struct {
 func (r *V5WebsocketPublicKlineResponse) Key() V5WebsocketPublicKlineParamKey {
 	topic := r.Topic
 	arr := strings.Split(topic, ".")
-	if arr[0] != V5WebsocketPublicTopicKline || len(arr) != 3 {
+	if arr[0] != V5WebsocketPublicTopicKline.String() || len(arr) != 3 {
 		return V5WebsocketPublicKlineParamKey{}
 	}
 

diff --git a/v5_ws_public_orderbook.go b/v5_ws_public_orderbook.go
@@ -134,7 +134,7 @@ func (b *V5WebsocketPublicOrderBookAsks) UnmarshalJSON(data []byte) error {
 func (r *V5WebsocketPublicOrderBookResponse) Key() V5WebsocketPublicOrderBookParamKey {
 	topic := r.Topic
 	arr := strings.Split(topic, ".")
-	if arr[0] != V5WebsocketPublicTopicOrderBook || len(arr) != 3 {
+	if arr[0] != V5WebsocketPublicTopicOrderBook.String() || len(arr) != 3 {
 		return V5WebsocketPublicOrderBookParamKey{}
 	}
 	depth, err := strconv.Atoi(arr[1])

diff --git a/v5_ws_public_ticker.go b/v5_ws_public_ticker.go
@@ -152,7 +152,7 @@ type V5WebsocketPublicTickerSpotResult struct {
 func (r *V5WebsocketPublicTickerResponse) Key() V5WebsocketPublicTickerParamKey {
 	topic := r.Topic
 	arr := strings.Split(topic, ".")
-	if arr[0] != V5WebsocketPublicTopicTicker || len(arr) != 2 {
+	if arr[0] != V5WebsocketPublicTopicTicker.String() || len(arr) != 2 {
 		return V5WebsocketPublicTickerParamKey{}
 	}