Files
gocryptotrader/exchanges/huobi/huobi_websocket.go
Gareth Kirwan f9437dbd08 Bitfinex: Websocket subscription improvements (#1353)
* Websockets: Add keys to websocket subscriptions

* This switches all RO uses of the mutex to use a RLock method.
* The mutex used for discrete field access has had scope drift from
  name 'connectionMutex' so rename to more appropriate fieldsMutex
* The mutex used for Set/CanUseAuthEndpoints moves from the
  subscriptions endpoint to the fieldsMutex
* Add GetSubscription by key
* Expose stream.Matcher type

* Bitfinex: Subscribe and Unsubscribe atomicly

* Fix Auth failures ignored
* This change makes it so that Subscribe and Unsubscribe wait for success
** Tells the DataHandler about errors
** Errors are returned to consumers
* Subscribes concurrently to the channels
* It also simplifies the chanId to stream mapping
* Removes unable to locate chanID: %d errors which are just noise
* Paves the way for unified channelSubscription id handling
* Adds support for subId for Book subscriptions, which is more robust

* Vastly simplifies what we need to test TestWsSubscribedResponse
This test was working to ensure that the various fancy key parsing
mechanisms all worked. Now that we use subId, we just need a thorough
test of that
* Expose Match.Set in order to capture websocket incoming data
Can't see another way of doing this. Doesn't seem too bad

* Allow tests to run with auth or WS
These flags made it difficult to run the tests whilst working on
websockets

* Enable API auth and WS in testconfig
This change minimises the changes requires for a full test run against
live endpoints, so that new contributors have a clearer testing path.
I cannot see any reason to turn WS off and Auth endpoints off when we're
not going to run API tests without Creds being set, and we're not going
to do live fire tests without canManipulateRealOrders

* TestWsSubscribe and various fixes
** Enables the websocket for live non-authed integration tests by default
** Adds an integration test for subscriptions
** Changes the Ws tests to respect canManipulateRealOrders
** Uses WsConnect instead of setupWs; fixes seqNo config not sent for WS tests
** Allows api creds to live in config/testdata.json which might be
  less likely to accidentally commit, and less obtrusive

* Bitfinex: Support period and timeframe for Candles

* Fixes manual Subscribe() symbol or key formatting
* Unifies handling of params for DefaultSubscriptions and manual
  subsrciptions

* Bitfinex: Handle conf and info WS channel events

* Bitfinex: Better tests for subscriptions

* fixup! Websockets: Add keys to websocket subscriptions

* fixup! Bitfinex: Subscribe and Unsubscribe atomicly

* fixup! Websockets: Add keys to websocket subscriptions

* Websockets: Add Pending subscription status

Add a status tracker so that Sub/Unsub can prevent duplicates,
and also fixes when first message comes before we have added the sub
to the tracker

* Websockets: Add State instead of pending

This change allows more clarity about the current state and
checks for specifically already Unsubing

* Bitfinex: Fix first sub message maybe lost

The only link we have between a sub req and the sub resp is the subID.
And the only link we have between a sub message and the sub is the chanID.
We can't derive a link using Pair or anything else.

This meant that by sending the resp and its chanID down the IncomingData
channel, we allowed the channel reader to maybe process the next
message, the first message on the channel, before the runtime executed
the switch back to subscribeToChan waiting on the chan.

To fix this, we key initially on subId.(string), and then replace it
with chanId.(int64) when we have it *inside* the wsHandleData so we
know we've procedurally handled it before the next message.

subscribeToChan is then free to remove the subId keyed Sub regardless of
error or not

If there's an error, we don't need to inline handling because there
won't be any second update.

Expands test coverage to make sure those subId keyed subscriptions are
removed.

* Websocket: Validate state in SetChanState

* fixup! Bitfinex: Fix first sub message maybe lost

* Websockets: Rename RemoveUnsuccessfulSubs

Implementation doesn't imply Unsuccessful or need to.
This change supports the registering of Pending subs

* Bitfinex: Fix race in Tests
2023-11-02 12:10:43 +11:00

824 lines
22 KiB
Go

package huobi
import (
"context"
"encoding/json"
"errors"
"fmt"
"net/http"
"net/url"
"strconv"
"strings"
"time"
"github.com/gorilla/websocket"
"github.com/thrasher-corp/gocryptotrader/common"
"github.com/thrasher-corp/gocryptotrader/common/crypto"
"github.com/thrasher-corp/gocryptotrader/currency"
"github.com/thrasher-corp/gocryptotrader/exchanges/account"
"github.com/thrasher-corp/gocryptotrader/exchanges/asset"
"github.com/thrasher-corp/gocryptotrader/exchanges/order"
"github.com/thrasher-corp/gocryptotrader/exchanges/orderbook"
"github.com/thrasher-corp/gocryptotrader/exchanges/stream"
"github.com/thrasher-corp/gocryptotrader/exchanges/ticker"
"github.com/thrasher-corp/gocryptotrader/exchanges/trade"
"github.com/thrasher-corp/gocryptotrader/log"
)
const (
baseWSURL = "wss://api.huobi.pro"
futuresWSURL = "wss://api.hbdm.com/"
wsMarketURL = baseWSURL + "/ws"
wsMarketKline = "market.%s.kline.1min"
wsMarketDepth = "market.%s.depth.step0"
wsMarketTrade = "market.%s.trade.detail"
wsMarketTicker = "market.%s.detail"
wsAccountsOrdersEndPoint = "/ws/v1"
wsAccountsList = "accounts.list"
wsOrdersList = "orders.list"
wsOrdersDetail = "orders.detail"
wsAccountsOrdersURL = baseWSURL + wsAccountsOrdersEndPoint
wsAccountListEndpoint = wsAccountsOrdersEndPoint + "/" + wsAccountsList
wsOrdersListEndpoint = wsAccountsOrdersEndPoint + "/" + wsOrdersList
wsOrdersDetailEndpoint = wsAccountsOrdersEndPoint + "/" + wsOrdersDetail
wsDateTimeFormatting = "2006-01-02T15:04:05"
signatureMethod = "HmacSHA256"
signatureVersion = "2"
requestOp = "req"
authOp = "auth"
loginDelay = 50 * time.Millisecond
rateLimit = 20
)
// Instantiates a communications channel between websocket connections
var comms = make(chan WsMessage)
// WsConnect initiates a new websocket connection
func (h *HUOBI) WsConnect() error {
if !h.Websocket.IsEnabled() || !h.IsEnabled() {
return errors.New(stream.WebsocketNotEnabled)
}
var dialer websocket.Dialer
err := h.wsDial(&dialer)
if err != nil {
return err
}
if h.Websocket.CanUseAuthenticatedEndpoints() {
err = h.wsAuthenticatedDial(&dialer)
if err != nil {
log.Errorf(log.ExchangeSys,
"%v - authenticated dial failed: %v\n",
h.Name,
err)
}
err = h.wsLogin(context.TODO())
if err != nil {
log.Errorf(log.ExchangeSys,
"%v - authentication failed: %v\n",
h.Name,
err)
h.Websocket.SetCanUseAuthenticatedEndpoints(false)
}
}
h.Websocket.Wg.Add(1)
go h.wsReadData()
return nil
}
func (h *HUOBI) wsDial(dialer *websocket.Dialer) error {
err := h.Websocket.Conn.Dial(dialer, http.Header{})
if err != nil {
return err
}
h.Websocket.Wg.Add(1)
go h.wsFunnelConnectionData(h.Websocket.Conn, wsMarketURL)
return nil
}
func (h *HUOBI) wsAuthenticatedDial(dialer *websocket.Dialer) error {
if !h.IsWebsocketAuthenticationSupported() {
return fmt.Errorf("%v AuthenticatedWebsocketAPISupport not enabled",
h.Name)
}
err := h.Websocket.AuthConn.Dial(dialer, http.Header{})
if err != nil {
return err
}
h.Websocket.Wg.Add(1)
go h.wsFunnelConnectionData(h.Websocket.AuthConn, wsAccountsOrdersURL)
return nil
}
// wsFunnelConnectionData manages data from multiple endpoints and passes it to
// a channel
func (h *HUOBI) wsFunnelConnectionData(ws stream.Connection, url string) {
defer h.Websocket.Wg.Done()
for {
resp := ws.ReadMessage()
if resp.Raw == nil {
return
}
comms <- WsMessage{Raw: resp.Raw, URL: url}
}
}
// wsReadData receives and passes on websocket messages for processing
func (h *HUOBI) wsReadData() {
defer h.Websocket.Wg.Done()
for {
select {
case <-h.Websocket.ShutdownC:
select {
case resp := <-comms:
err := h.wsHandleData(resp.Raw)
if err != nil {
select {
case h.Websocket.DataHandler <- err:
default:
log.Errorf(log.WebsocketMgr,
"%s websocket handle data error: %v",
h.Name,
err)
}
}
default:
}
return
case resp := <-comms:
err := h.wsHandleData(resp.Raw)
if err != nil {
h.Websocket.DataHandler <- err
}
}
}
}
func stringToOrderStatus(status string) (order.Status, error) {
switch status {
case "submitted":
return order.New, nil
case "canceled":
return order.Cancelled, nil
case "partial-filled":
return order.PartiallyFilled, nil
case "partial-canceled":
return order.PartiallyCancelled, nil
default:
return order.UnknownStatus,
errors.New(status + " not recognised as order status")
}
}
func stringToOrderSide(side string) (order.Side, error) {
switch {
case strings.Contains(side, "buy"):
return order.Buy, nil
case strings.Contains(side, "sell"):
return order.Sell, nil
}
return order.UnknownSide,
errors.New(side + " not recognised as order side")
}
func stringToOrderType(oType string) (order.Type, error) {
switch {
case strings.Contains(oType, "limit"):
return order.Limit, nil
case strings.Contains(oType, "market"):
return order.Market, nil
}
return order.UnknownType,
errors.New(oType + " not recognised as order type")
}
func (h *HUOBI) wsHandleData(respRaw []byte) error {
var init WsResponse
err := json.Unmarshal(respRaw, &init)
if err != nil {
return err
}
if init.Subscribed != "" ||
init.UnSubscribed != "" ||
init.Op == "sub" ||
init.Op == "unsub" {
// TODO handle subs
return nil
}
if init.Ping != 0 {
h.sendPingResponse(init.Ping)
return nil
}
if init.Op == "ping" {
authPing := authenticationPing{
OP: "pong",
TS: init.TS,
}
err := h.Websocket.AuthConn.SendJSONMessage(authPing)
if err != nil {
log.Errorln(log.ExchangeSys, err)
}
return nil
}
if init.ErrorMessage != "" {
if init.ErrorMessage == "api-signature-not-valid" {
h.Websocket.SetCanUseAuthenticatedEndpoints(false)
return errors.New(h.Name +
" - invalid credentials. Authenticated requests disabled")
}
codes, _ := init.ErrorCode.(string)
return errors.New(h.Name + " Code:" + codes + " Message:" + init.ErrorMessage)
}
if init.ClientID > 0 {
if h.Websocket.Match.IncomingWithData(init.ClientID, respRaw) {
return nil
}
}
switch {
case strings.EqualFold(init.Op, authOp):
h.Websocket.SetCanUseAuthenticatedEndpoints(true)
// Auth captured
return nil
case strings.EqualFold(init.Topic, "accounts"):
var response WsAuthenticatedAccountsResponse
err := json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
h.Websocket.DataHandler <- response
case strings.Contains(init.Topic, "orders") &&
strings.Contains(init.Topic, "update"):
var response WsAuthenticatedOrdersUpdateResponse
err := json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
data := strings.Split(response.Topic, ".")
if len(data) < 2 {
return errors.New(h.Name +
" - currency could not be extracted from response")
}
orderID := strconv.FormatInt(response.Data.OrderID, 10)
var oSide order.Side
oSide, err = stringToOrderSide(response.Data.OrderType)
if err != nil {
h.Websocket.DataHandler <- order.ClassificationError{
Exchange: h.Name,
OrderID: orderID,
Err: err,
}
}
var oType order.Type
oType, err = stringToOrderType(response.Data.OrderType)
if err != nil {
h.Websocket.DataHandler <- order.ClassificationError{
Exchange: h.Name,
OrderID: orderID,
Err: err,
}
}
var oStatus order.Status
oStatus, err = stringToOrderStatus(response.Data.OrderState)
if err != nil {
h.Websocket.DataHandler <- order.ClassificationError{
Exchange: h.Name,
OrderID: orderID,
Err: err,
}
}
var p currency.Pair
var a asset.Item
p, a, err = h.GetRequestFormattedPairAndAssetType(data[1])
if err != nil {
return err
}
h.Websocket.DataHandler <- &order.Detail{
Price: response.Data.Price,
Amount: response.Data.UnfilledAmount + response.Data.FilledAmount,
ExecutedAmount: response.Data.FilledAmount,
RemainingAmount: response.Data.UnfilledAmount,
Exchange: h.Name,
OrderID: orderID,
Type: oType,
Side: oSide,
Status: oStatus,
AssetType: a,
LastUpdated: time.Unix(response.TS*1000, 0),
Pair: p,
}
case strings.Contains(init.Topic, "orders"):
var response WsOldOrderUpdate
err := json.Unmarshal(respRaw, &response)
if err != nil {
return err
}
h.Websocket.DataHandler <- response
case strings.Contains(init.Channel, "depth"):
var depth WsDepth
err := json.Unmarshal(respRaw, &depth)
if err != nil {
return err
}
data := strings.Split(depth.Channel, ".")
err = h.WsProcessOrderbook(&depth, data[1])
if err != nil {
return err
}
case strings.Contains(init.Channel, "kline"):
var kline WsKline
err := json.Unmarshal(respRaw, &kline)
if err != nil {
return err
}
data := strings.Split(kline.Channel, ".")
var p currency.Pair
var a asset.Item
p, a, err = h.GetRequestFormattedPairAndAssetType(data[1])
if err != nil {
return err
}
h.Websocket.DataHandler <- stream.KlineData{
Timestamp: time.UnixMilli(kline.Timestamp),
Exchange: h.Name,
AssetType: a,
Pair: p,
OpenPrice: kline.Tick.Open,
ClosePrice: kline.Tick.Close,
HighPrice: kline.Tick.High,
LowPrice: kline.Tick.Low,
Volume: kline.Tick.Volume,
Interval: data[3],
}
case strings.Contains(init.Channel, "trade.detail"):
if !h.IsSaveTradeDataEnabled() {
return nil
}
var t WsTrade
err := json.Unmarshal(respRaw, &t)
if err != nil {
return err
}
data := strings.Split(t.Channel, ".")
var p currency.Pair
var a asset.Item
p, a, err = h.GetRequestFormattedPairAndAssetType(data[1])
if err != nil {
return err
}
var trades []trade.Data
for i := range t.Tick.Data {
side := order.Buy
if t.Tick.Data[i].Direction != "buy" {
side = order.Sell
}
trades = append(trades, trade.Data{
Exchange: h.Name,
AssetType: a,
CurrencyPair: p,
Timestamp: time.UnixMilli(t.Tick.Data[i].Timestamp),
Amount: t.Tick.Data[i].Amount,
Price: t.Tick.Data[i].Price,
Side: side,
TID: strconv.FormatFloat(t.Tick.Data[i].TradeID, 'f', -1, 64),
})
}
return trade.AddTradesToBuffer(h.Name, trades...)
case strings.Contains(init.Channel, "detail"),
strings.Contains(init.Rep, "detail"):
var wsTicker WsTick
err := json.Unmarshal(respRaw, &wsTicker)
if err != nil {
return err
}
var data []string
if wsTicker.Channel != "" {
data = strings.Split(wsTicker.Channel, ".")
}
if wsTicker.Rep != "" {
data = strings.Split(wsTicker.Rep, ".")
}
var p currency.Pair
var a asset.Item
p, a, err = h.GetRequestFormattedPairAndAssetType(data[1])
if err != nil {
return err
}
h.Websocket.DataHandler <- &ticker.Price{
ExchangeName: h.Name,
Open: wsTicker.Tick.Open,
Close: wsTicker.Tick.Close,
Volume: wsTicker.Tick.Amount,
QuoteVolume: wsTicker.Tick.Volume,
High: wsTicker.Tick.High,
Low: wsTicker.Tick.Low,
LastUpdated: time.UnixMilli(wsTicker.Timestamp),
AssetType: a,
Pair: p,
}
default:
h.Websocket.DataHandler <- stream.UnhandledMessageWarning{
Message: h.Name + stream.UnhandledMessage + string(respRaw),
}
return nil
}
return nil
}
func (h *HUOBI) sendPingResponse(pong int64) {
err := h.Websocket.Conn.SendJSONMessage(WsPong{Pong: pong})
if err != nil {
log.Errorln(log.ExchangeSys, err)
}
}
// WsProcessOrderbook processes new orderbook data
func (h *HUOBI) WsProcessOrderbook(update *WsDepth, symbol string) error {
pairs, err := h.GetEnabledPairs(asset.Spot)
if err != nil {
return err
}
format, err := h.GetPairFormat(asset.Spot, true)
if err != nil {
return err
}
p, err := currency.NewPairFromFormattedPairs(symbol,
pairs,
format)
if err != nil {
return err
}
bids := make(orderbook.Items, len(update.Tick.Bids))
for i := range update.Tick.Bids {
price, ok := update.Tick.Bids[i][0].(float64)
if !ok {
return errors.New("unable to type assert bid price")
}
amount, ok := update.Tick.Bids[i][1].(float64)
if !ok {
return errors.New("unable to type assert bid amount")
}
bids[i] = orderbook.Item{
Price: price,
Amount: amount,
}
}
asks := make(orderbook.Items, len(update.Tick.Asks))
for i := range update.Tick.Asks {
price, ok := update.Tick.Asks[i][0].(float64)
if !ok {
return errors.New("unable to type assert ask price")
}
amount, ok := update.Tick.Asks[i][1].(float64)
if !ok {
return errors.New("unable to type assert ask amount")
}
asks[i] = orderbook.Item{
Price: price,
Amount: amount,
}
}
var newOrderBook orderbook.Base
newOrderBook.Asks = asks
newOrderBook.Bids = bids
newOrderBook.Pair = p
newOrderBook.Asset = asset.Spot
newOrderBook.Exchange = h.Name
newOrderBook.VerifyOrderbook = h.CanVerifyOrderbook
newOrderBook.LastUpdated = time.UnixMilli(update.Timestamp)
return h.Websocket.Orderbook.LoadSnapshot(&newOrderBook)
}
// GenerateDefaultSubscriptions Adds default subscriptions to websocket to be handled by ManageSubscriptions()
func (h *HUOBI) GenerateDefaultSubscriptions() ([]stream.ChannelSubscription, error) {
var channels = []string{wsMarketKline,
wsMarketDepth,
wsMarketTrade,
wsMarketTicker}
var subscriptions []stream.ChannelSubscription
if h.Websocket.CanUseAuthenticatedEndpoints() {
channels = append(channels, "orders.%v", "orders.%v.update")
subscriptions = append(subscriptions, stream.ChannelSubscription{
Channel: "accounts",
})
}
enabledCurrencies, err := h.GetEnabledPairs(asset.Spot)
if err != nil {
return nil, err
}
for i := range channels {
for j := range enabledCurrencies {
enabledCurrencies[j].Delimiter = ""
channel := fmt.Sprintf(channels[i],
enabledCurrencies[j].Lower().String())
subscriptions = append(subscriptions, stream.ChannelSubscription{
Channel: channel,
Currency: enabledCurrencies[j],
})
}
}
return subscriptions, nil
}
// Subscribe sends a websocket message to receive data from the channel
func (h *HUOBI) Subscribe(channelsToSubscribe []stream.ChannelSubscription) error {
var creds *account.Credentials
if h.Websocket.CanUseAuthenticatedEndpoints() {
var err error
creds, err = h.GetCredentials(context.TODO())
if err != nil {
return err
}
}
var errs error
for i := range channelsToSubscribe {
if (strings.Contains(channelsToSubscribe[i].Channel, "orders.") ||
strings.Contains(channelsToSubscribe[i].Channel, "accounts")) && creds != nil {
err := h.wsAuthenticatedSubscribe(creds,
"sub",
wsAccountsOrdersEndPoint+channelsToSubscribe[i].Channel,
channelsToSubscribe[i].Channel)
if err != nil {
errs = common.AppendError(errs, err)
continue
}
h.Websocket.AddSuccessfulSubscriptions(channelsToSubscribe[i])
continue
}
err := h.Websocket.Conn.SendJSONMessage(WsRequest{
Subscribe: channelsToSubscribe[i].Channel,
})
if err != nil {
errs = common.AppendError(errs, err)
continue
}
h.Websocket.AddSuccessfulSubscriptions(channelsToSubscribe[i])
}
if errs != nil {
return errs
}
return nil
}
// Unsubscribe sends a websocket message to stop receiving data from the channel
func (h *HUOBI) Unsubscribe(channelsToUnsubscribe []stream.ChannelSubscription) error {
var creds *account.Credentials
if h.Websocket.CanUseAuthenticatedEndpoints() {
var err error
creds, err = h.GetCredentials(context.TODO())
if err != nil {
return err
}
}
var errs error
for i := range channelsToUnsubscribe {
if (strings.Contains(channelsToUnsubscribe[i].Channel, "orders.") ||
strings.Contains(channelsToUnsubscribe[i].Channel, "accounts")) && creds != nil {
err := h.wsAuthenticatedSubscribe(creds,
"unsub",
wsAccountsOrdersEndPoint+channelsToUnsubscribe[i].Channel,
channelsToUnsubscribe[i].Channel)
if err != nil {
errs = common.AppendError(errs, err)
continue
}
h.Websocket.RemoveSubscriptions(channelsToUnsubscribe[i])
continue
}
err := h.Websocket.Conn.SendJSONMessage(WsRequest{
Unsubscribe: channelsToUnsubscribe[i].Channel,
})
if err != nil {
errs = common.AppendError(errs, err)
continue
}
h.Websocket.RemoveSubscriptions(channelsToUnsubscribe[i])
}
if errs != nil {
return errs
}
return nil
}
func (h *HUOBI) wsGenerateSignature(creds *account.Credentials, timestamp, endpoint string) ([]byte, error) {
values := url.Values{}
values.Set("AccessKeyId", creds.Key)
values.Set("SignatureMethod", signatureMethod)
values.Set("SignatureVersion", signatureVersion)
values.Set("Timestamp", timestamp)
host := "api.huobi.pro"
payload := fmt.Sprintf("%s\n%s\n%s\n%s",
http.MethodGet, host, endpoint, values.Encode())
return crypto.GetHMAC(crypto.HashSHA256, []byte(payload), []byte(creds.Secret))
}
func (h *HUOBI) wsLogin(ctx context.Context) error {
if !h.IsWebsocketAuthenticationSupported() {
return fmt.Errorf("%v AuthenticatedWebsocketAPISupport not enabled", h.Name)
}
creds, err := h.GetCredentials(ctx)
if err != nil {
return err
}
h.Websocket.SetCanUseAuthenticatedEndpoints(true)
timestamp := time.Now().UTC().Format(wsDateTimeFormatting)
request := WsAuthenticationRequest{
Op: authOp,
AccessKeyID: creds.Key,
SignatureMethod: signatureMethod,
SignatureVersion: signatureVersion,
Timestamp: timestamp,
}
hmac, err := h.wsGenerateSignature(creds, timestamp, wsAccountsOrdersEndPoint)
if err != nil {
return err
}
request.Signature = crypto.Base64Encode(hmac)
err = h.Websocket.AuthConn.SendJSONMessage(request)
if err != nil {
h.Websocket.SetCanUseAuthenticatedEndpoints(false)
return err
}
time.Sleep(loginDelay)
return nil
}
func (h *HUOBI) wsAuthenticatedSubscribe(creds *account.Credentials, operation, endpoint, topic string) error {
timestamp := time.Now().UTC().Format(wsDateTimeFormatting)
request := WsAuthenticatedSubscriptionRequest{
Op: operation,
AccessKeyID: creds.Key,
SignatureMethod: signatureMethod,
SignatureVersion: signatureVersion,
Timestamp: timestamp,
Topic: topic,
}
hmac, err := h.wsGenerateSignature(creds, timestamp, endpoint)
if err != nil {
return err
}
request.Signature = crypto.Base64Encode(hmac)
return h.Websocket.AuthConn.SendJSONMessage(request)
}
func (h *HUOBI) wsGetAccountsList(ctx context.Context) (*WsAuthenticatedAccountsListResponse, error) {
if !h.Websocket.CanUseAuthenticatedEndpoints() {
return nil, fmt.Errorf("%v not authenticated cannot get accounts list", h.Name)
}
creds, err := h.GetCredentials(ctx)
if err != nil {
return nil, err
}
timestamp := time.Now().UTC().Format(wsDateTimeFormatting)
request := WsAuthenticatedAccountsListRequest{
Op: requestOp,
AccessKeyID: creds.Key,
SignatureMethod: signatureMethod,
SignatureVersion: signatureVersion,
Timestamp: timestamp,
Topic: wsAccountsList,
}
hmac, err := h.wsGenerateSignature(creds, timestamp, wsAccountListEndpoint)
if err != nil {
return nil, err
}
request.Signature = crypto.Base64Encode(hmac)
request.ClientID = h.Websocket.AuthConn.GenerateMessageID(true)
resp, err := h.Websocket.AuthConn.SendMessageReturnResponse(request.ClientID, request)
if err != nil {
return nil, err
}
var response WsAuthenticatedAccountsListResponse
err = json.Unmarshal(resp, &response)
if err != nil {
return nil, err
}
code, _ := response.ErrorCode.(int)
if code != 0 {
return nil, errors.New(response.ErrorMessage)
}
return &response, nil
}
func (h *HUOBI) wsGetOrdersList(ctx context.Context, accountID int64, pair currency.Pair) (*WsAuthenticatedOrdersResponse, error) {
if !h.Websocket.CanUseAuthenticatedEndpoints() {
return nil, fmt.Errorf("%v not authenticated cannot get orders list", h.Name)
}
creds, err := h.GetCredentials(ctx)
if err != nil {
return nil, err
}
fPair, err := h.FormatExchangeCurrency(pair, asset.Spot)
if err != nil {
return nil, err
}
timestamp := time.Now().UTC().Format(wsDateTimeFormatting)
request := WsAuthenticatedOrdersListRequest{
Op: requestOp,
AccessKeyID: creds.Key,
SignatureMethod: signatureMethod,
SignatureVersion: signatureVersion,
Timestamp: timestamp,
Topic: wsOrdersList,
AccountID: accountID,
Symbol: fPair.String(),
States: "submitted,partial-filled",
}
hmac, err := h.wsGenerateSignature(creds, timestamp, wsOrdersListEndpoint)
if err != nil {
return nil, err
}
request.Signature = crypto.Base64Encode(hmac)
request.ClientID = h.Websocket.AuthConn.GenerateMessageID(true)
resp, err := h.Websocket.AuthConn.SendMessageReturnResponse(request.ClientID, request)
if err != nil {
return nil, err
}
var response WsAuthenticatedOrdersResponse
err = json.Unmarshal(resp, &response)
if err != nil {
return nil, err
}
code, _ := response.ErrorCode.(int)
if code != 0 {
return nil, errors.New(response.ErrorMessage)
}
return &response, nil
}
func (h *HUOBI) wsGetOrderDetails(ctx context.Context, orderID string) (*WsAuthenticatedOrderDetailResponse, error) {
if !h.Websocket.CanUseAuthenticatedEndpoints() {
return nil, fmt.Errorf("%v not authenticated cannot get order details", h.Name)
}
creds, err := h.GetCredentials(ctx)
if err != nil {
return nil, err
}
timestamp := time.Now().UTC().Format(wsDateTimeFormatting)
request := WsAuthenticatedOrderDetailsRequest{
Op: requestOp,
AccessKeyID: creds.Key,
SignatureMethod: signatureMethod,
SignatureVersion: signatureVersion,
Timestamp: timestamp,
Topic: wsOrdersDetail,
OrderID: orderID,
}
hmac, err := h.wsGenerateSignature(creds, timestamp, wsOrdersDetailEndpoint)
if err != nil {
return nil, err
}
request.Signature = crypto.Base64Encode(hmac)
request.ClientID = h.Websocket.AuthConn.GenerateMessageID(true)
resp, err := h.Websocket.AuthConn.SendMessageReturnResponse(request.ClientID, request)
if err != nil {
return nil, err
}
var response WsAuthenticatedOrderDetailResponse
err = json.Unmarshal(resp, &response)
if err != nil {
return nil, err
}
code, _ := response.ErrorCode.(int)
if code != 0 {
return nil, errors.New(response.ErrorMessage)
}
return &response, nil
}