Files
gocryptotrader/exchanges/coinut/coinut_wrapper.go
Gareth Kirwan 52c6b3bf0b Websocket: Various refactors and test improvements (#1466)
* Websocket: Remove IsInit and simplify SetProxyAddress

IsInit was basically the same as IsConnected.
Any time Connect was called both would be set to true.
Any time we had a disconnect they'd both be set to false
Shutdown() incorrectly didn't setInit(false)

SetProxyAddress simplified to only reconnect a connected Websocket.
Any other state means it hasn't been Connected, or it's about to
reconnect anyway.
There's no handling for IsConnecting previously, either, so I've wrapped
that behind the main mutex.

* Websocket: Expand and Assertify tests

* Websocket: Simplify state transistions

* Websocket: Simplify Connecting/Connected state

* Websocket: Tests and errors for websocket

* Websocket: Make WebsocketNotEnabled a real error

This allows for testing and avoids the repetition.
If each returned error is a error.New() you can never use errors.Is()

* Websocket: Add more testable errors

* Websocket: Improve GenerateMessageID test

Testing just the last id doesn't feel very robust

* Websocket: Protect Setup() from races

* Websocket: Use atomics instead of mutex

This was spurred by looking at the setState call in trafficMonitor and
the effect on blocking and efficiency.
With the new atomic types in Go 1.19, and the small types in use here,
atomics should be safe for our usage. bools should be truly atomic,
and uint32 is atomic when the accepted value range is less than one byte/uint8 since
that can be written atomicly by concurrent processors.
Maybe that's not even a factor any more, however we don't even have to worry enough to check.

* Websocket: Fix and simplify traffic monitor

trafficMonitor had a check throttle at the end of the for loop to stop it just gobbling the (blocking) trafficAlert channel non-stop.
That makes sense, except that nothing is sent to the trafficAlert channel if there's no listener.
So that means that it's out by one second on the trafficAlert, because any traffic received during the pause is doesn't try to send a traffic alert.

The unstopped timer is deliberately leaked for later GC when shutdown.
It won't delay/block anything, and it's a trivial memory leak during an infrequent event.

Deliberately Choosing to recreate the timer each time instead of using Stop, drain and reset

* Websocket: Split traficMonitor test on behaviours

* Websocket: Remove trafficMonitor connected status

trafficMonitor does not need to set the connection to be connected.
Connect() does that. Anything after that should result in a full
shutdown and restart. It can't and shouldn't become connected
unexpectedly, and this is most likely a race anyway.

Also dropped trafficCheckInterval to 100ms to mitigate races of traffic
alerts being buffered for too long.

* Websocket: Set disconnected earlier in Shutdown

This caused a possible race where state is still connected, but we start
to trigger interested actors via ShutdownC and Wait.
They may check state and then call Shutdown again, such as
trafficMonitor

* Websocket: Wait 5s for slow tests to pass traffic draining

Keep getting failures upstream on test rigs.
Think they can be very contended, so this pushes the boundary right out
to 5s
2024-02-23 18:39:25 +11:00

1128 lines
32 KiB
Go

package coinut
import (
"context"
"errors"
"fmt"
"sort"
"strconv"
"strings"
"time"
"github.com/thrasher-corp/gocryptotrader/common"
"github.com/thrasher-corp/gocryptotrader/config"
"github.com/thrasher-corp/gocryptotrader/currency"
exchange "github.com/thrasher-corp/gocryptotrader/exchanges"
"github.com/thrasher-corp/gocryptotrader/exchanges/account"
"github.com/thrasher-corp/gocryptotrader/exchanges/asset"
"github.com/thrasher-corp/gocryptotrader/exchanges/deposit"
"github.com/thrasher-corp/gocryptotrader/exchanges/fundingrate"
"github.com/thrasher-corp/gocryptotrader/exchanges/futures"
"github.com/thrasher-corp/gocryptotrader/exchanges/kline"
"github.com/thrasher-corp/gocryptotrader/exchanges/order"
"github.com/thrasher-corp/gocryptotrader/exchanges/orderbook"
"github.com/thrasher-corp/gocryptotrader/exchanges/protocol"
"github.com/thrasher-corp/gocryptotrader/exchanges/request"
"github.com/thrasher-corp/gocryptotrader/exchanges/stream"
"github.com/thrasher-corp/gocryptotrader/exchanges/stream/buffer"
"github.com/thrasher-corp/gocryptotrader/exchanges/ticker"
"github.com/thrasher-corp/gocryptotrader/exchanges/trade"
"github.com/thrasher-corp/gocryptotrader/log"
"github.com/thrasher-corp/gocryptotrader/portfolio/withdraw"
)
// GetDefaultConfig returns a default exchange config
func (c *COINUT) GetDefaultConfig(ctx context.Context) (*config.Exchange, error) {
c.SetDefaults()
exchCfg, err := c.GetStandardConfig()
if err != nil {
return nil, err
}
err = c.SetupDefaults(exchCfg)
if err != nil {
return nil, err
}
if c.Features.Supports.RESTCapabilities.AutoPairUpdates {
err = c.UpdateTradablePairs(ctx, true)
if err != nil {
return nil, err
}
}
return exchCfg, nil
}
// SetDefaults sets current default values
func (c *COINUT) SetDefaults() {
c.Name = "COINUT"
c.Enabled = true
c.Verbose = true
c.API.CredentialsValidator.RequiresKey = true
c.API.CredentialsValidator.RequiresClientID = true
requestFmt := &currency.PairFormat{Uppercase: true}
configFmt := &currency.PairFormat{Uppercase: true, Delimiter: currency.DashDelimiter}
err := c.SetGlobalPairsManager(requestFmt, configFmt, asset.Spot)
if err != nil {
log.Errorln(log.ExchangeSys, err)
}
c.Features = exchange.Features{
Supports: exchange.FeaturesSupported{
REST: true,
Websocket: true,
RESTCapabilities: protocol.Features{
TickerFetching: true,
TradeFetching: true,
OrderbookFetching: true,
AutoPairUpdates: true,
AccountInfo: true,
GetOrders: true,
CancelOrders: true,
CancelOrder: true,
SubmitOrder: true,
SubmitOrders: true,
UserTradeHistory: true,
TradeFee: true,
FiatDepositFee: true,
FiatWithdrawalFee: true,
},
WebsocketCapabilities: protocol.Features{
AccountBalance: true,
GetOrders: true,
CancelOrders: true,
CancelOrder: true,
SubmitOrder: true,
SubmitOrders: true,
UserTradeHistory: true,
TickerFetching: true,
TradeFetching: true,
OrderbookFetching: true,
AccountInfo: true,
Subscribe: true,
Unsubscribe: true,
AuthenticatedEndpoints: true,
MessageCorrelation: true,
},
WithdrawPermissions: exchange.WithdrawCryptoViaWebsiteOnly |
exchange.WithdrawFiatViaWebsiteOnly,
},
Enabled: exchange.FeaturesEnabled{
AutoPairUpdates: true,
},
}
c.Requester, err = request.New(c.Name,
common.NewHTTPClientWithTimeout(exchange.DefaultHTTPTimeout))
if err != nil {
log.Errorln(log.ExchangeSys, err)
}
c.API.Endpoints = c.NewEndpoints()
err = c.API.Endpoints.SetDefaultEndpoints(map[exchange.URL]string{
exchange.RestSpot: coinutAPIURL,
exchange.WebsocketSpot: coinutWebsocketURL,
})
if err != nil {
log.Errorln(log.ExchangeSys, err)
}
c.Websocket = stream.NewWebsocket()
c.WebsocketResponseMaxLimit = exchange.DefaultWebsocketResponseMaxLimit
c.WebsocketResponseCheckTimeout = exchange.DefaultWebsocketResponseCheckTimeout
c.WebsocketOrderbookBufferLimit = exchange.DefaultWebsocketOrderbookBufferLimit
}
// Setup sets the current exchange configuration
func (c *COINUT) Setup(exch *config.Exchange) error {
err := exch.Validate()
if err != nil {
return err
}
if !exch.Enabled {
c.SetEnabled(false)
return nil
}
err = c.SetupDefaults(exch)
if err != nil {
return err
}
wsRunningURL, err := c.API.Endpoints.GetURL(exchange.WebsocketSpot)
if err != nil {
return err
}
err = c.Websocket.Setup(&stream.WebsocketSetup{
ExchangeConfig: exch,
DefaultURL: coinutWebsocketURL,
RunningURL: wsRunningURL,
Connector: c.WsConnect,
Subscriber: c.Subscribe,
Unsubscriber: c.Unsubscribe,
GenerateSubscriptions: c.GenerateDefaultSubscriptions,
Features: &c.Features.Supports.WebsocketCapabilities,
OrderbookBufferConfig: buffer.Config{
SortBuffer: true,
SortBufferByUpdateIDs: true,
},
})
if err != nil {
return err
}
return c.Websocket.SetupNewConnection(stream.ConnectionSetup{
ResponseCheckTimeout: exch.WebsocketResponseCheckTimeout,
ResponseMaxLimit: exch.WebsocketResponseMaxLimit,
RateLimit: wsRateLimitInMilliseconds,
})
}
// FetchTradablePairs returns a list of the exchanges tradable pairs
func (c *COINUT) FetchTradablePairs(ctx context.Context, _ asset.Item) (currency.Pairs, error) {
var resp Instruments
var err error
if c.Websocket.IsConnected() {
resp, err = c.WsGetInstruments()
} else {
resp, err = c.GetInstruments(ctx)
}
if err != nil {
return nil, err
}
pairs := make([]currency.Pair, 0, len(resp.Instruments))
var pair currency.Pair
for _, instrument := range resp.Instruments {
if len(instrument) == 0 {
return nil, errors.New("invalid data received")
}
c.instrumentMap.Seed(instrument[0].Base+instrument[0].Quote, instrument[0].InstrumentID)
pair, err = currency.NewPairFromStrings(instrument[0].Base, instrument[0].Quote)
if err != nil {
return nil, err
}
pairs = append(pairs, pair)
}
return pairs, nil
}
// UpdateTradablePairs updates the exchanges available pairs and stores
// them in the exchanges config
func (c *COINUT) UpdateTradablePairs(ctx context.Context, forceUpdate bool) error {
pairs, err := c.FetchTradablePairs(ctx, asset.Spot)
if err != nil {
return err
}
err = c.UpdatePairs(pairs, asset.Spot, false, forceUpdate)
if err != nil {
return err
}
return c.EnsureOnePairEnabled()
}
// UpdateAccountInfo retrieves balances for all enabled currencies for the
// COINUT exchange
func (c *COINUT) UpdateAccountInfo(ctx context.Context, assetType asset.Item) (account.Holdings, error) {
var info account.Holdings
var bal *UserBalance
var err error
if c.Websocket.CanUseAuthenticatedWebsocketForWrapper() {
var resp *UserBalance
resp, err = c.wsGetAccountBalance()
if err != nil {
return info, err
}
bal = resp
} else {
bal, err = c.GetUserBalance(ctx)
if err != nil {
return info, err
}
}
var balances = []account.Balance{
{
Currency: currency.BCH,
Total: bal.BCH,
},
{
Currency: currency.BTC,
Total: bal.BTC,
},
{
Currency: currency.BTG,
Total: bal.BTG,
},
{
Currency: currency.CAD,
Total: bal.CAD,
},
{
Currency: currency.ETC,
Total: bal.ETC,
},
{
Currency: currency.ETH,
Total: bal.ETH,
},
{
Currency: currency.LCH,
Total: bal.LCH,
},
{
Currency: currency.LTC,
Total: bal.LTC,
},
{
Currency: currency.MYR,
Total: bal.MYR,
},
{
Currency: currency.SGD,
Total: bal.SGD,
},
{
Currency: currency.USD,
Total: bal.USD,
},
{
Currency: currency.USDT,
Total: bal.USDT,
},
{
Currency: currency.XMR,
Total: bal.XMR,
},
{
Currency: currency.ZEC,
Total: bal.ZEC,
},
}
info.Exchange = c.Name
info.Accounts = append(info.Accounts, account.SubAccount{
AssetType: assetType,
Currencies: balances,
})
creds, err := c.GetCredentials(ctx)
if err != nil {
return account.Holdings{}, err
}
err = account.Process(&info, creds)
if err != nil {
return account.Holdings{}, err
}
return info, nil
}
// FetchAccountInfo retrieves balances for all enabled currencies
func (c *COINUT) FetchAccountInfo(ctx context.Context, assetType asset.Item) (account.Holdings, error) {
creds, err := c.GetCredentials(ctx)
if err != nil {
return account.Holdings{}, err
}
acc, err := account.GetHoldings(c.Name, creds, assetType)
if err != nil {
return c.UpdateAccountInfo(ctx, assetType)
}
return acc, nil
}
// UpdateTickers updates the ticker for all currency pairs of a given asset type
func (c *COINUT) UpdateTickers(_ context.Context, _ asset.Item) error {
return common.ErrFunctionNotSupported
}
// UpdateTicker updates and returns the ticker for a currency pair
func (c *COINUT) UpdateTicker(ctx context.Context, p currency.Pair, a asset.Item) (*ticker.Price, error) {
if p.IsEmpty() {
return nil, currency.ErrCurrencyPairEmpty
}
if !c.SupportsAsset(a) {
return nil, fmt.Errorf("%w %v", asset.ErrNotSupported, a)
}
err := c.loadInstrumentsIfNotLoaded()
if err != nil {
return nil, err
}
fPair, err := c.FormatExchangeCurrency(p, a)
if err != nil {
return nil, err
}
instID := c.instrumentMap.LookupID(fPair.String())
if instID == 0 {
return nil, errors.New("unable to lookup instrument ID")
}
var tick Ticker
tick, err = c.GetInstrumentTicker(ctx, instID)
if err != nil {
return nil, err
}
err = ticker.ProcessTicker(&ticker.Price{
Last: tick.Last,
High: tick.High24,
Low: tick.Low24,
Bid: tick.HighestBuy,
Ask: tick.LowestSell,
Volume: tick.Volume24,
Pair: p,
LastUpdated: time.Unix(0, tick.Timestamp),
ExchangeName: c.Name,
AssetType: a})
if err != nil {
return nil, err
}
return ticker.GetTicker(c.Name, p, a)
}
// FetchTicker returns the ticker for a currency pair
func (c *COINUT) FetchTicker(ctx context.Context, p currency.Pair, assetType asset.Item) (*ticker.Price, error) {
tickerNew, err := ticker.GetTicker(c.Name, p, assetType)
if err != nil {
return c.UpdateTicker(ctx, p, assetType)
}
return tickerNew, nil
}
// FetchOrderbook returns orderbook base on the currency pair
func (c *COINUT) FetchOrderbook(ctx context.Context, p currency.Pair, assetType asset.Item) (*orderbook.Base, error) {
ob, err := orderbook.Get(c.Name, p, assetType)
if err != nil {
return c.UpdateOrderbook(ctx, p, assetType)
}
return ob, nil
}
// UpdateOrderbook updates and returns the orderbook for a currency pair
func (c *COINUT) UpdateOrderbook(ctx context.Context, p currency.Pair, assetType asset.Item) (*orderbook.Base, error) {
if p.IsEmpty() {
return nil, currency.ErrCurrencyPairEmpty
}
if err := c.CurrencyPairs.IsAssetEnabled(assetType); err != nil {
return nil, err
}
book := &orderbook.Base{
Exchange: c.Name,
Pair: p,
Asset: assetType,
VerifyOrderbook: c.CanVerifyOrderbook,
}
err := c.loadInstrumentsIfNotLoaded()
if err != nil {
return book, err
}
fPair, err := c.FormatExchangeCurrency(p, assetType)
if err != nil {
return book, err
}
instID := c.instrumentMap.LookupID(fPair.String())
if instID == 0 {
return book, errLookupInstrumentID
}
orderbookNew, err := c.GetInstrumentOrderbook(ctx, instID, 200)
if err != nil {
return book, err
}
book.Bids = make(orderbook.Items, len(orderbookNew.Buy))
for x := range orderbookNew.Buy {
book.Bids[x] = orderbook.Item{
Amount: orderbookNew.Buy[x].Quantity,
Price: orderbookNew.Buy[x].Price,
}
}
book.Asks = make(orderbook.Items, len(orderbookNew.Sell))
for x := range orderbookNew.Sell {
book.Asks[x] = orderbook.Item{
Amount: orderbookNew.Sell[x].Quantity,
Price: orderbookNew.Sell[x].Price,
}
}
err = book.Process()
if err != nil {
return book, err
}
return orderbook.Get(c.Name, p, assetType)
}
// GetAccountFundingHistory returns funding history, deposits and
// withdrawals
func (c *COINUT) GetAccountFundingHistory(_ context.Context) ([]exchange.FundingHistory, error) {
return nil, common.ErrFunctionNotSupported
}
// GetWithdrawalsHistory returns previous withdrawals data
func (c *COINUT) GetWithdrawalsHistory(_ context.Context, _ currency.Code, _ asset.Item) ([]exchange.WithdrawalHistory, error) {
return nil, common.ErrFunctionNotSupported
}
// GetRecentTrades returns the most recent trades for a currency and asset
func (c *COINUT) GetRecentTrades(ctx context.Context, p currency.Pair, assetType asset.Item) ([]trade.Data, error) {
var err error
p, err = c.FormatExchangeCurrency(p, assetType)
if err != nil {
return nil, err
}
currencyID := c.instrumentMap.LookupID(p.String())
if currencyID == 0 {
return nil, errLookupInstrumentID
}
var tradeData Trades
tradeData, err = c.GetTrades(ctx, currencyID)
if err != nil {
return nil, err
}
resp := make([]trade.Data, len(tradeData.Trades))
for i := range tradeData.Trades {
var side order.Side
side, err = order.StringToOrderSide(tradeData.Trades[i].Side)
if err != nil {
return nil, err
}
resp[i] = trade.Data{
Exchange: c.Name,
TID: strconv.FormatInt(tradeData.Trades[i].TransactionID, 10),
CurrencyPair: p,
AssetType: assetType,
Side: side,
Price: tradeData.Trades[i].Price,
Amount: tradeData.Trades[i].Quantity,
Timestamp: time.Unix(0, tradeData.Trades[i].Timestamp*int64(time.Microsecond)),
}
}
err = c.AddTradesToBuffer(resp...)
if err != nil {
return nil, err
}
sort.Sort(trade.ByDate(resp))
return resp, nil
}
// GetHistoricTrades returns historic trade data within the timeframe provided
func (c *COINUT) GetHistoricTrades(_ context.Context, _ currency.Pair, _ asset.Item, _, _ time.Time) ([]trade.Data, error) {
return nil, common.ErrFunctionNotSupported
}
// SubmitOrder submits a new order
func (c *COINUT) SubmitOrder(ctx context.Context, o *order.Submit) (*order.SubmitResponse, error) {
err := o.Validate()
if err != nil {
return nil, err
}
if _, err = strconv.Atoi(o.ClientID); err != nil {
return nil, fmt.Errorf("%s - ClientID must be a number, received: %s",
c.Name, o.ClientID)
}
var orderID string
status := order.New
if c.Websocket.CanUseAuthenticatedWebsocketForWrapper() {
var response *order.Detail
response, err = c.wsSubmitOrder(&WsSubmitOrderParameters{
Currency: o.Pair,
Side: o.Side,
Amount: o.Amount,
Price: o.Price,
})
if err != nil {
return nil, err
}
orderID = response.OrderID
} else {
err = c.loadInstrumentsIfNotLoaded()
if err != nil {
return nil, err
}
var fPair currency.Pair
fPair, err = c.FormatExchangeCurrency(o.Pair, asset.Spot)
if err != nil {
return nil, err
}
currencyID := c.instrumentMap.LookupID(fPair.String())
if currencyID == 0 {
return nil, errLookupInstrumentID
}
var APIResponse interface{}
var clientIDInt uint64
clientIDInt, err = strconv.ParseUint(o.ClientID, 10, 32)
if err != nil {
return nil, err
}
APIResponse, err = c.NewOrder(ctx,
currencyID,
o.Amount,
o.Price,
o.Side.IsLong(),
uint32(clientIDInt))
if err != nil {
return nil, err
}
responseMap, ok := APIResponse.(map[string]interface{})
if !ok {
return nil, errors.New("unable to type assert responseMap")
}
orderType, ok := responseMap["reply"].(string)
if !ok {
return nil, errors.New("unable to type assert orderType")
}
switch orderType {
case "order_rejected":
return nil, fmt.Errorf("clientOrderID: %v was rejected: %v", o.ClientID, responseMap["reasons"])
case "order_filled":
orderIDResp, ok := responseMap["order_id"].(float64)
if !ok {
return nil, errors.New("unable to type assert orderID")
}
orderID = strconv.FormatFloat(orderIDResp, 'f', -1, 64)
status = order.Filled
case "order_accepted":
orderIDResp, ok := responseMap["order_id"].(float64)
if !ok {
return nil, errors.New("unable to type assert orderID")
}
orderID = strconv.FormatFloat(orderIDResp, 'f', -1, 64)
}
}
resp, err := o.DeriveSubmitResponse(orderID)
if err != nil {
return nil, err
}
resp.Status = status
return resp, nil
}
// ModifyOrder will allow of changing orderbook placement and limit to
// market conversion
func (c *COINUT) ModifyOrder(_ context.Context, _ *order.Modify) (*order.ModifyResponse, error) {
return nil, common.ErrFunctionNotSupported
}
// CancelOrder cancels an order by its corresponding ID number
func (c *COINUT) CancelOrder(ctx context.Context, o *order.Cancel) error {
if err := o.Validate(o.StandardCancel()); err != nil {
return err
}
err := c.loadInstrumentsIfNotLoaded()
if err != nil {
return err
}
orderIDInt, err := strconv.ParseInt(o.OrderID, 10, 64)
if err != nil {
return err
}
fPair, err := c.FormatExchangeCurrency(o.Pair, asset.Spot)
if err != nil {
return err
}
currencyID := c.instrumentMap.LookupID(fPair.String())
if c.Websocket.CanUseAuthenticatedWebsocketForWrapper() {
var resp *CancelOrdersResponse
resp, err = c.wsCancelOrder(&WsCancelOrderParameters{
Currency: o.Pair,
OrderID: orderIDInt,
})
if err != nil {
return err
}
if len(resp.Status) >= 1 && resp.Status[0] != "OK" {
return errors.New(c.Name + " - Failed to cancel order " + o.OrderID)
}
} else {
if currencyID == 0 {
return errLookupInstrumentID
}
_, err = c.CancelExistingOrder(ctx, currencyID, orderIDInt)
if err != nil {
return err
}
}
return nil
}
// CancelBatchOrders cancels an orders by their corresponding ID numbers
func (c *COINUT) CancelBatchOrders(ctx context.Context, o []order.Cancel) (*order.CancelBatchResponse, error) {
if len(o) == 0 {
return nil, order.ErrCancelOrderIsNil
}
req := make([]CancelOrders, 0, len(o))
for i := range o {
switch {
case o[i].ClientOrderID != "":
return nil, order.ErrClientOrderIDNotSupported
case o[i].OrderID != "":
currencyID := c.instrumentMap.LookupID(o[i].Pair.String())
oid, err := strconv.ParseInt(o[i].OrderID, 10, 64)
if err != nil {
return nil, err
}
req = append(req, CancelOrders{
InstrumentID: currencyID,
OrderID: oid,
})
default:
return nil, order.ErrOrderIDNotSet
}
}
results, err := c.CancelOrders(ctx, req)
if err != nil {
return nil, err
}
resp := &order.CancelBatchResponse{Status: make(map[string]string)}
for i := range results.Results {
resp.Status[strconv.FormatInt(results.Results[i].OrderID, 10)] = results.Results[i].Status
}
return resp, nil
}
// GetServerTime returns the current exchange server time.
func (c *COINUT) GetServerTime(_ context.Context, _ asset.Item) (time.Time, error) {
return time.Time{}, common.ErrFunctionNotSupported
}
// CancelAllOrders cancels all orders associated with a currency pair
func (c *COINUT) CancelAllOrders(ctx context.Context, details *order.Cancel) (order.CancelAllResponse, error) {
if err := details.Validate(); err != nil {
return order.CancelAllResponse{}, err
}
var cancelAllOrdersResponse order.CancelAllResponse
err := c.loadInstrumentsIfNotLoaded()
if err != nil {
return cancelAllOrdersResponse, err
}
cancelAllOrdersResponse.Status = make(map[string]string)
if c.Websocket.CanUseAuthenticatedWebsocketForWrapper() {
openOrders, err := c.wsGetOpenOrders(details.Pair.String())
if err != nil {
return cancelAllOrdersResponse, err
}
var ordersToCancel []WsCancelOrderParameters
for i := range openOrders.Orders {
var fPair currency.Pair
fPair, err = c.FormatExchangeCurrency(details.Pair, asset.Spot)
if err != nil {
return cancelAllOrdersResponse, err
}
if openOrders.Orders[i].InstrumentID == c.instrumentMap.LookupID(fPair.String()) {
ordersToCancel = append(ordersToCancel, WsCancelOrderParameters{
Currency: details.Pair,
OrderID: openOrders.Orders[i].OrderID,
})
}
}
resp, err := c.wsCancelOrders(ordersToCancel)
if err != nil {
return cancelAllOrdersResponse, err
}
for i := range resp.Results {
if openOrders.Orders[i].Status[0] != "OK" {
cancelAllOrdersResponse.Status[strconv.FormatInt(openOrders.Orders[i].OrderID, 10)] = strings.Join(openOrders.Orders[i].Status, ",")
}
}
} else {
var allTheOrders []OrderResponse
ids := c.instrumentMap.GetInstrumentIDs()
for x := range ids {
fPair, err := c.FormatExchangeCurrency(details.Pair, asset.Spot)
if err != nil {
return cancelAllOrdersResponse, err
}
if ids[x] == c.instrumentMap.LookupID(fPair.String()) {
openOrders, err := c.GetOpenOrders(ctx, ids[x])
if err != nil {
return cancelAllOrdersResponse, err
}
allTheOrders = append(allTheOrders, openOrders.Orders...)
}
}
var allTheOrdersToCancel []CancelOrders
for i := range allTheOrders {
cancelOrder := CancelOrders{
InstrumentID: allTheOrders[i].InstrumentID,
OrderID: allTheOrders[i].OrderID,
}
allTheOrdersToCancel = append(allTheOrdersToCancel, cancelOrder)
}
if len(allTheOrdersToCancel) > 0 {
resp, err := c.CancelOrders(ctx, allTheOrdersToCancel)
if err != nil {
return cancelAllOrdersResponse, err
}
for i := range resp.Results {
if resp.Results[i].Status != "OK" {
cancelAllOrdersResponse.Status[strconv.FormatInt(resp.Results[i].OrderID, 10)] = resp.Results[i].Status
}
}
}
}
return cancelAllOrdersResponse, nil
}
// GetOrderInfo returns order information based on order ID
func (c *COINUT) GetOrderInfo(_ context.Context, _ string, _ currency.Pair, _ asset.Item) (*order.Detail, error) {
return nil, common.ErrFunctionNotSupported
}
// GetDepositAddress returns a deposit address for a specified currency
func (c *COINUT) GetDepositAddress(_ context.Context, _ currency.Code, _, _ string) (*deposit.Address, error) {
return nil, common.ErrFunctionNotSupported
}
// WithdrawCryptocurrencyFunds returns a withdrawal ID when a withdrawal is
// submitted
func (c *COINUT) WithdrawCryptocurrencyFunds(_ context.Context, _ *withdraw.Request) (*withdraw.ExchangeResponse, error) {
return nil, common.ErrFunctionNotSupported
}
// WithdrawFiatFunds returns a withdrawal ID when a
// withdrawal is submitted
func (c *COINUT) WithdrawFiatFunds(_ context.Context, _ *withdraw.Request) (*withdraw.ExchangeResponse, error) {
return nil, common.ErrFunctionNotSupported
}
// WithdrawFiatFundsToInternationalBank returns a withdrawal ID when a
// withdrawal is submitted
func (c *COINUT) WithdrawFiatFundsToInternationalBank(_ context.Context, _ *withdraw.Request) (*withdraw.ExchangeResponse, error) {
return nil, common.ErrFunctionNotSupported
}
// GetFeeByType returns an estimate of fee based on type of transaction
func (c *COINUT) GetFeeByType(ctx context.Context, feeBuilder *exchange.FeeBuilder) (float64, error) {
if feeBuilder == nil {
return 0, fmt.Errorf("%T %w", feeBuilder, common.ErrNilPointer)
}
if !c.AreCredentialsValid(ctx) && // Todo check connection status
feeBuilder.FeeType == exchange.CryptocurrencyTradeFee {
feeBuilder.FeeType = exchange.OfflineTradeFee
}
return c.GetFee(feeBuilder)
}
// GetActiveOrders retrieves any orders that are active/open
func (c *COINUT) GetActiveOrders(ctx context.Context, req *order.MultiOrderRequest) (order.FilteredOrders, error) {
err := req.Validate()
if err != nil {
return nil, err
}
err = c.loadInstrumentsIfNotLoaded()
if err != nil {
return nil, err
}
var orders []order.Detail
var currenciesToCheck []string
if len(req.Pairs) == 0 {
for i := range req.Pairs {
var fPair currency.Pair
fPair, err = c.FormatExchangeCurrency(req.Pairs[i], asset.Spot)
if err != nil {
return nil, err
}
currenciesToCheck = append(currenciesToCheck, fPair.String())
}
} else {
for k := range c.instrumentMap.Instruments {
currenciesToCheck = append(currenciesToCheck, k)
}
}
if c.Websocket.CanUseAuthenticatedWebsocketForWrapper() {
for x := range currenciesToCheck {
var openOrders *WsUserOpenOrdersResponse
openOrders, err = c.wsGetOpenOrders(currenciesToCheck[x])
if err != nil {
return nil, err
}
for i := range openOrders.Orders {
var p currency.Pair
p, err = currency.NewPairFromString(currenciesToCheck[x])
if err != nil {
return nil, err
}
var fPair currency.Pair
fPair, err = c.FormatExchangeCurrency(p, asset.Spot)
if err != nil {
return nil, err
}
var side order.Side
side, err = order.StringToOrderSide(openOrders.Orders[i].Side)
if err != nil {
return nil, err
}
orders = append(orders, order.Detail{
Exchange: c.Name,
OrderID: strconv.FormatInt(openOrders.Orders[i].OrderID, 10),
Pair: fPair,
Side: side,
Date: time.Unix(0, openOrders.Orders[i].Timestamp),
Status: order.Active,
Price: openOrders.Orders[i].Price,
Amount: openOrders.Orders[i].Quantity,
ExecutedAmount: openOrders.Orders[i].Quantity - openOrders.Orders[i].OpenQuantity,
RemainingAmount: openOrders.Orders[i].OpenQuantity,
})
}
}
} else {
var instrumentsToUse []int64
for x := range req.Pairs {
var curr currency.Pair
curr, err = c.FormatExchangeCurrency(req.Pairs[x], asset.Spot)
if err != nil {
return nil, err
}
instrumentsToUse = append(instrumentsToUse,
c.instrumentMap.LookupID(curr.String()))
}
if len(instrumentsToUse) == 0 {
instrumentsToUse = c.instrumentMap.GetInstrumentIDs()
}
var pairs currency.Pairs
pairs, err = c.GetEnabledPairs(asset.Spot)
if err != nil {
return nil, err
}
var format currency.PairFormat
format, err = c.GetPairFormat(asset.Spot, true)
if err != nil {
return nil, err
}
for x := range instrumentsToUse {
var openOrders GetOpenOrdersResponse
openOrders, err = c.GetOpenOrders(ctx, instrumentsToUse[x])
if err != nil {
return nil, err
}
for y := range openOrders.Orders {
curr := c.instrumentMap.LookupInstrument(instrumentsToUse[x])
var p currency.Pair
p, err = currency.NewPairFromFormattedPairs(curr, pairs, format)
if err != nil {
return nil, err
}
var side order.Side
side, err = order.StringToOrderSide(openOrders.Orders[y].Side)
if err != nil {
return nil, err
}
orders = append(orders, order.Detail{
OrderID: strconv.FormatInt(openOrders.Orders[y].OrderID, 10),
Amount: openOrders.Orders[y].Quantity,
Price: openOrders.Orders[y].Price,
Exchange: c.Name,
Side: side,
Date: time.Unix(openOrders.Orders[y].Timestamp, 0),
Pair: p,
})
}
}
}
return req.Filter(c.Name, orders), nil
}
// GetOrderHistory retrieves account order information
// Can Limit response to specific order status
func (c *COINUT) GetOrderHistory(ctx context.Context, req *order.MultiOrderRequest) (order.FilteredOrders, error) {
err := req.Validate()
if err != nil {
return nil, err
}
err = c.loadInstrumentsIfNotLoaded()
if err != nil {
return nil, err
}
var allOrders []order.Detail
if c.Websocket.CanUseAuthenticatedWebsocketForWrapper() {
for i := range req.Pairs {
for j := int64(0); ; j += 100 {
var trades *WsTradeHistoryResponse
trades, err = c.wsGetTradeHistory(req.Pairs[i], j, 100)
if err != nil {
return allOrders, err
}
for x := range trades.Trades {
curr := c.instrumentMap.LookupInstrument(trades.Trades[x].InstrumentID)
var p currency.Pair
p, err = currency.NewPairFromString(curr)
if err != nil {
return nil, err
}
var side order.Side
side, err = order.StringToOrderSide(trades.Trades[x].Side)
if err != nil {
return nil, err
}
detail := order.Detail{
Exchange: c.Name,
OrderID: strconv.FormatInt(trades.Trades[x].OrderID, 10),
Pair: p,
Side: side,
Date: time.Unix(0, trades.Trades[x].Timestamp),
Status: order.Filled,
Price: trades.Trades[x].Price,
Amount: trades.Trades[x].Quantity,
ExecutedAmount: trades.Trades[x].Quantity - trades.Trades[x].OpenQuantity,
RemainingAmount: trades.Trades[x].OpenQuantity,
}
detail.InferCostsAndTimes()
allOrders = append(allOrders, detail)
}
if len(trades.Trades) < 100 {
break
}
}
}
} else {
var instrumentsToUse []int64
for x := range req.Pairs {
var curr currency.Pair
curr, err = c.FormatExchangeCurrency(req.Pairs[x], asset.Spot)
if err != nil {
return nil, err
}
instrumentID := c.instrumentMap.LookupID(curr.String())
if instrumentID > 0 {
instrumentsToUse = append(instrumentsToUse, instrumentID)
}
}
if len(instrumentsToUse) == 0 {
instrumentsToUse = c.instrumentMap.GetInstrumentIDs()
}
var pairs currency.Pairs
pairs, err = c.GetEnabledPairs(asset.Spot)
if err != nil {
return nil, err
}
var format currency.PairFormat
format, err = c.GetPairFormat(asset.Spot, true)
if err != nil {
return nil, err
}
for x := range instrumentsToUse {
var orders TradeHistory
orders, err = c.GetTradeHistory(ctx, instrumentsToUse[x], -1, -1)
if err != nil {
return nil, err
}
for y := range orders.Trades {
curr := c.instrumentMap.LookupInstrument(instrumentsToUse[x])
var p currency.Pair
p, err = currency.NewPairFromFormattedPairs(curr, pairs, format)
if err != nil {
return nil, err
}
var side order.Side
side, err = order.StringToOrderSide(orders.Trades[y].Order.Side)
if err != nil {
return nil, err
}
allOrders = append(allOrders, order.Detail{
OrderID: strconv.FormatInt(orders.Trades[y].Order.OrderID, 10),
Amount: orders.Trades[y].Order.Quantity,
Price: orders.Trades[y].Order.Price,
Exchange: c.Name,
Side: side,
Date: time.Unix(orders.Trades[y].Order.Timestamp, 0),
Pair: p,
})
}
}
}
return req.Filter(c.Name, allOrders), nil
}
// AuthenticateWebsocket sends an authentication message to the websocket
func (c *COINUT) AuthenticateWebsocket(ctx context.Context) error {
return c.wsAuthenticate(ctx)
}
func (c *COINUT) loadInstrumentsIfNotLoaded() error {
if !c.instrumentMap.IsLoaded() {
if c.Websocket.IsConnected() {
_, err := c.WsGetInstruments()
if err != nil {
return err
}
} else {
err := c.SeedInstruments(context.TODO())
if err != nil {
return err
}
}
}
return nil
}
// ValidateAPICredentials validates current credentials used for wrapper
// functionality
func (c *COINUT) ValidateAPICredentials(ctx context.Context, assetType asset.Item) error {
_, err := c.UpdateAccountInfo(ctx, assetType)
return c.CheckTransientError(err)
}
// GetHistoricCandles returns candles between a time period for a set time interval
func (c *COINUT) GetHistoricCandles(_ context.Context, _ currency.Pair, _ asset.Item, _ kline.Interval, _, _ time.Time) (*kline.Item, error) {
return nil, common.ErrFunctionNotSupported
}
// GetHistoricCandlesExtended returns candles between a time period for a set time interval
func (c *COINUT) GetHistoricCandlesExtended(_ context.Context, _ currency.Pair, _ asset.Item, _ kline.Interval, _, _ time.Time) (*kline.Item, error) {
return nil, common.ErrFunctionNotSupported
}
// GetFuturesContractDetails returns all contracts from the exchange by asset type
func (c *COINUT) GetFuturesContractDetails(context.Context, asset.Item) ([]futures.Contract, error) {
return nil, common.ErrFunctionNotSupported
}
// GetLatestFundingRates returns the latest funding rates data
func (c *COINUT) GetLatestFundingRates(context.Context, *fundingrate.LatestRateRequest) ([]fundingrate.LatestRateResponse, error) {
return nil, common.ErrFunctionNotSupported
}
// UpdateOrderExecutionLimits updates order execution limits
func (c *COINUT) UpdateOrderExecutionLimits(_ context.Context, _ asset.Item) error {
return common.ErrNotYetImplemented
}