From 5f9611e5c2cfe2518eb60f845190bacb280fc8c6 Mon Sep 17 00:00:00 2001 From: hirokisan Date: Tue, 11 Apr 2023 20:38:15 +0900 Subject: [PATCH] apply type --- v5_ws_public.go | 16 ++++++++++------ v5_ws_public_kline.go | 2 +- v5_ws_public_orderbook.go | 2 +- v5_ws_public_ticker.go | 2 +- 4 files changed, 13 insertions(+), 9 deletions(-) diff --git a/v5_ws_public.go b/v5_ws_public.go index d55c869..269d29a 100644 --- 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 index 1c6e379..c9f6af2 100644 --- 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 index 04288ae..e97af6f 100644 --- 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 index 3848b1d..8902ce2 100644 --- 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{} }