bithumb: Add websocket support (#745)

* bithumb: Add websocket support (WIP) * bithumb: cont impl. * bithumb: finish, issues with orderbook needs review * linter issues * bithumb: move to separate file * bithumb: change to pointer for book * bithumb: Add tests * bithumb: Address nits * websocket: Export subscription error and wrap returns * bithumb: cleanup/ fix tests * gctrpc/bithumb: fix misspelling * bithumb: gofmt * readme: update support table, regen doc * tradesReadme: update * readme: update template
2026-05-23 07:26:47 +00:00 · 2021-08-16 16:53:23 +10:00
parent e77baf3ad4
commit 736c92a99b
20 changed files with 3034 additions and 2112 deletions
--- a/exchanges/bithumb/bithumb_websocket_test.go
+++ b/exchanges/bithumb/bithumb_websocket_test.go
@@ -0,0 +1,101 @@
+package bithumb
+
+import (
+	"errors"
+	"sync"
+	"testing"
+
+	"github.com/thrasher-corp/gocryptotrader/currency"
+	exchange "github.com/thrasher-corp/gocryptotrader/exchanges"
+	"github.com/thrasher-corp/gocryptotrader/exchanges/asset"
+	"github.com/thrasher-corp/gocryptotrader/exchanges/stream"
+	"github.com/thrasher-corp/gocryptotrader/exchanges/ticker"
+)
+
+var (
+	wsTickerResp    = []byte(`{"type":"ticker","content":{"tickType":"24H","date":"20210811","time":"132017","openPrice":"33400","closePrice":"34010","lowPrice":"32660","highPrice":"34510","value":"45741663716.89916828275244531","volume":"1359398.496892086826189907","sellVolume":"198021.237915860451480504","buyVolume":"1161377.258976226374709403","prevClosePrice":"33530","chgRate":"1.83","chgAmt":"610","volumePower":"500","symbol":"UNI_KRW"}}`)
+	wsTransResp     = []byte(`{"type":"transaction","content":{"list":[{"buySellGb":"1","contPrice":"1166","contQty":"125.2400","contAmt":"146029.8400","contDtm":"2021-08-13 15:23:42.911273","updn":"dn","symbol":"DAI_KRW"}]}}`)
+	wsOrderbookResp = []byte(`{"type":"orderbookdepth","content":{"list":[{"symbol":"XLM_KRW","orderType":"ask","price":"401.2","quantity":"0","total":"0"},{"symbol":"XLM_KRW","orderType":"ask","price":"401.6","quantity":"21277.735","total":"1"},{"symbol":"XLM_KRW","orderType":"ask","price":"403.3","quantity":"4000","total":"1"},{"symbol":"XLM_KRW","orderType":"bid","price":"399.5","quantity":"0","total":"0"},{"symbol":"XLM_KRW","orderType":"bid","price":"398.2","quantity":"0","total":"0"},{"symbol":"XLM_KRW","orderType":"bid","price":"399.8","quantity":"31416.8779","total":"1"},{"symbol":"XLM_KRW","orderType":"bid","price":"398.5","quantity":"34328.387","total":"1"}],"datetime":"1628835823604483"}}`)
+)
+
+func TestWsHandleData(t *testing.T) {
+	t.Parallel()
+	pairs := currency.Pairs{
+		currency.Pair{
+			Base:  currency.BTC,
+			Quote: currency.USDT,
+		},
+	}
+
+	dummy := Bithumb{
+		Base: exchange.Base{
+			Name: "dummy",
+			Features: exchange.Features{
+				Enabled: exchange.FeaturesEnabled{SaveTradeData: true},
+			},
+			CurrencyPairs: currency.PairsManager{
+				Pairs: map[asset.Item]*currency.PairStore{
+					asset.Spot: {
+						Available: pairs,
+						Enabled:   pairs,
+						ConfigFormat: &currency.PairFormat{
+							Uppercase: true,
+						},
+					},
+				},
+			},
+			Websocket: &stream.Websocket{
+				Wg:          new(sync.WaitGroup),
+				DataHandler: make(chan interface{}, 1),
+			},
+		},
+	}
+
+	dummy.setupOrderbookManager()
+	dummy.API.Endpoints = b.NewEndpoints()
+
+	welcomeMsg := []byte(`{"status":"0000","resmsg":"Connected Successfully"}`)
+	err := dummy.wsHandleData(welcomeMsg)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = dummy.wsHandleData([]byte(`{"status":"1336","resmsg":"Failed"}`))
+	if !errors.Is(err, stream.ErrSubscriptionFailure) {
+		t.Fatalf("received: %v but expected: %v",
+			err,
+			stream.ErrSubscriptionFailure)
+	}
+
+	err = dummy.wsHandleData(wsTickerResp)
+	if !errors.Is(err, nil) {
+		t.Fatalf("received: %v but expected: %v", err, nil)
+	}
+
+	handled := <-dummy.Websocket.DataHandler
+	_, ok := handled.(*ticker.Price)
+	if !ok {
+		t.Fatal("unexpected value")
+	}
+
+	err = dummy.wsHandleData(wsTransResp) // This doesn't pipe to datahandler
+	if !errors.Is(err, nil) {
+		t.Fatalf("received: %v but expected: %v", err, nil)
+	}
+
+	err = dummy.wsHandleData(wsOrderbookResp) // This doesn't pipe to datahandler
+	if !errors.Is(err, nil) {
+		t.Fatalf("received: %v but expected: %v", err, nil)
+	}
+}
+
+func TestGenerateSubscriptions(t *testing.T) {
+	t.Parallel()
+	sub, err := b.GenerateSubscriptions()
+	if err != nil {
+		t.Fatal(err)
+	}
+	if sub == nil {
+		t.Fatal("unexpected value")
+	}
+}