Files
gocryptotrader/engine/sync_manager.go
Adrian Gallagher 4651af5767 modernise: Run new gopls modernise tool against the codebase and fix minor issues (#1826)
* modernise: Run new gopls modernise tool against codebase

* Address shazbert's nits

* apichecker, gctcli: Simplify HTML scraping functions and improve depth limit handling

* refactor: Create minSyncInterval const and update order book limit handling for binance and binanceUS

* refactor: Various slice usage improvements and rename TODO

* tranches: Revert deleteByID changes due to performance decrease

Shazbert was a F1 driver in a past lifetime 🏎️

* tranches: Simply retrieve copy

Thanks to shazbert

* documentation: Sort contributors list by contributions

* tranches: Remove deadcode in deleteByID
2025-03-21 09:17:10 +11:00

941 lines
26 KiB
Go

package engine
import (
"context"
"errors"
"fmt"
"strconv"
"strings"
"sync"
"sync/atomic"
"time"
"github.com/thrasher-corp/gocryptotrader/common"
"github.com/thrasher-corp/gocryptotrader/common/key"
"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/asset"
"github.com/thrasher-corp/gocryptotrader/exchanges/orderbook"
"github.com/thrasher-corp/gocryptotrader/exchanges/stats"
"github.com/thrasher-corp/gocryptotrader/exchanges/ticker"
"github.com/thrasher-corp/gocryptotrader/log"
)
type syncItemType int
// const holds the sync item types
const (
SyncItemTicker syncItemType = iota
SyncItemOrderbook
SyncItemTrade
SyncManagerName = "exchange_syncer"
minSyncInterval = time.Second
)
var (
createdCounter = 0
removedCounter = 0
errNoSyncItemsEnabled = errors.New("no sync items enabled")
errUnknownSyncItem = errors.New("unknown sync item")
errCouldNotSyncNewData = errors.New("could not sync new data")
)
// SetupSyncManager creates a new CurrencyPairSyncer
func SetupSyncManager(c *config.SyncManagerConfig, exchangeManager iExchangeManager, remoteConfig *config.RemoteControlConfig, websocketRoutineManagerEnabled bool) (*SyncManager, error) {
if c == nil {
return nil, fmt.Errorf("%T %w", c, common.ErrNilPointer)
}
if !c.SynchronizeOrderbook && !c.SynchronizeTicker && !c.SynchronizeTrades {
return nil, errNoSyncItemsEnabled
}
if exchangeManager == nil {
return nil, errNilExchangeManager
}
if remoteConfig == nil {
return nil, errNilConfig
}
if c.NumWorkers <= 0 {
c.NumWorkers = config.DefaultSyncerWorkers
}
if c.TimeoutREST <= time.Duration(0) {
c.TimeoutREST = config.DefaultSyncerTimeoutREST
}
if c.TimeoutWebsocket <= time.Duration(0) {
c.TimeoutWebsocket = config.DefaultSyncerTimeoutWebsocket
}
if c.FiatDisplayCurrency.IsEmpty() {
return nil, fmt.Errorf("FiatDisplayCurrency %w", currency.ErrCurrencyCodeEmpty)
}
if !c.FiatDisplayCurrency.IsFiatCurrency() {
return nil, fmt.Errorf("%s %w", c.FiatDisplayCurrency, currency.ErrFiatDisplayCurrencyIsNotFiat)
}
if c.PairFormatDisplay == nil {
return nil, fmt.Errorf("%T %w", c.PairFormatDisplay, common.ErrNilPointer)
}
s := &SyncManager{
config: *c,
remoteConfig: remoteConfig,
exchangeManager: exchangeManager,
websocketRoutineManagerEnabled: websocketRoutineManagerEnabled,
fiatDisplayCurrency: c.FiatDisplayCurrency,
format: *c.PairFormatDisplay,
tickerBatchLastRequested: make(map[key.ExchangeAsset]time.Time),
currencyPairs: make(map[key.ExchangePairAsset]*currencyPairSyncAgent),
}
log.Debugf(log.SyncMgr,
"Exchange currency pair syncer config: continuous: %v ticker: %v"+
" orderbook: %v trades: %v workers: %v verbose: %v timeout REST: %v"+
" timeout Websocket: %v",
s.config.SynchronizeContinuously, s.config.SynchronizeTicker, s.config.SynchronizeOrderbook,
s.config.SynchronizeTrades, s.config.NumWorkers, s.config.Verbose, s.config.TimeoutREST,
s.config.TimeoutWebsocket)
s.inService.Add(1)
return s, nil
}
// IsRunning safely checks whether the subsystem is running
func (m *SyncManager) IsRunning() bool {
return m != nil && atomic.LoadInt32(&m.started) == 1
}
// Start runs the subsystem
func (m *SyncManager) Start() error {
if m == nil {
return fmt.Errorf("exchange CurrencyPairSyncer %w", ErrNilSubsystem)
}
if !atomic.CompareAndSwapInt32(&m.started, 0, 1) {
return ErrSubSystemAlreadyStarted
}
if !m.config.SynchronizeTicker &&
!m.config.SynchronizeOrderbook &&
!m.config.SynchronizeTrades {
return errNoSyncItemsEnabled
}
m.shutdown = make(chan bool)
m.initSyncWG.Add(1)
m.inService.Done()
log.Debugln(log.SyncMgr, "Exchange CurrencyPairSyncer started.")
exchanges, err := m.exchangeManager.GetExchanges()
if err != nil {
return err
}
for x := range exchanges {
exchangeName := exchanges[x].GetName()
supportsWebsocket := exchanges[x].SupportsWebsocket()
supportsREST := exchanges[x].SupportsREST()
if !supportsREST && !supportsWebsocket {
log.Warnf(log.SyncMgr,
"Loaded exchange %s does not support REST or Websocket.",
exchangeName)
continue
}
var usingWebsocket bool
var usingREST bool
if m.websocketRoutineManagerEnabled &&
supportsWebsocket &&
exchanges[x].IsWebsocketEnabled() {
usingWebsocket = true
} else if supportsREST {
usingREST = true
}
assetTypes := exchanges[x].GetAssetTypes(false)
for y := range assetTypes {
if exchanges[x].GetBase().CurrencyPairs.IsAssetEnabled(assetTypes[y]) != nil {
log.Warnf(log.SyncMgr,
"%s asset type %s is disabled, fetching enabled pairs is paused",
exchangeName,
assetTypes[y])
continue
}
wsAssetSupported := exchanges[x].IsAssetWebsocketSupported(assetTypes[y])
if !wsAssetSupported {
log.Warnf(log.SyncMgr,
"%s asset type %s websocket functionality is unsupported, REST fetching only.",
exchangeName,
assetTypes[y])
}
enabledPairs, err := exchanges[x].GetEnabledPairs(assetTypes[y])
if err != nil {
log.Errorf(log.SyncMgr,
"%s failed to get enabled pairs. Err: %s",
exchangeName,
err)
continue
}
for i := range enabledPairs {
k := key.ExchangePairAsset{
Asset: assetTypes[y],
Exchange: exchangeName,
Base: enabledPairs[i].Base.Item,
Quote: enabledPairs[i].Quote.Item,
}
if e := m.get(k); e != nil {
continue
}
sBase := syncBase{
IsUsingREST: usingREST || !wsAssetSupported,
IsUsingWebsocket: usingWebsocket && wsAssetSupported,
}
m.add(k, sBase)
}
}
}
if atomic.CompareAndSwapInt32(&m.initSyncStarted, 0, 1) {
if m.config.LogInitialSyncEvents {
log.Debugf(log.SyncMgr,
"Exchange CurrencyPairSyncer initial sync started. %d items to process.",
createdCounter)
}
m.initSyncStartTime = time.Now()
}
go func() {
m.initSyncWG.Wait()
if atomic.CompareAndSwapInt32(&m.initSyncCompleted, 0, 1) {
if m.config.LogInitialSyncEvents {
log.Debugf(log.SyncMgr, "Exchange CurrencyPairSyncer initial sync is complete.")
log.Debugf(log.SyncMgr, "Exchange CurrencyPairSyncer initial sync took %v [%v sync items].",
time.Since(m.initSyncStartTime), createdCounter)
}
if !m.config.SynchronizeContinuously {
log.Debugln(log.SyncMgr, "Exchange CurrencyPairSyncer stopping.")
err := m.Stop()
if err != nil {
log.Errorln(log.SyncMgr, err)
}
return
}
}
}()
if atomic.LoadInt32(&m.initSyncCompleted) == 1 && !m.config.SynchronizeContinuously {
return nil
}
for range m.config.NumWorkers {
go m.worker()
}
m.initSyncWG.Done()
return nil
}
// Stop shuts down the exchange currency pair syncer
func (m *SyncManager) Stop() error {
if m == nil {
return fmt.Errorf("exchange CurrencyPairSyncer %w", ErrNilSubsystem)
}
if !atomic.CompareAndSwapInt32(&m.started, 1, 0) {
return fmt.Errorf("exchange CurrencyPairSyncer %w", ErrSubSystemNotStarted)
}
close(m.shutdown)
m.inService.Add(1)
log.Debugln(log.SyncMgr, "Exchange CurrencyPairSyncer stopped.")
return nil
}
func (m *SyncManager) get(k key.ExchangePairAsset) *currencyPairSyncAgent {
m.mux.Lock()
defer m.mux.Unlock()
return m.currencyPairs[k]
}
func newCurrencyPairSyncAgent(k key.ExchangePairAsset) *currencyPairSyncAgent {
return &currencyPairSyncAgent{
Key: k,
Pair: currency.NewPair(k.Base.Currency(), k.Quote.Currency()),
Created: time.Now(),
locks: make([]sync.Mutex, SyncItemTrade+1),
trackers: make([]*syncBase, SyncItemTrade+1),
}
}
func (m *SyncManager) add(k key.ExchangePairAsset, s syncBase) *currencyPairSyncAgent {
m.mux.Lock()
defer m.mux.Unlock()
c := newCurrencyPairSyncAgent(k)
if m.config.SynchronizeTicker {
s := s
c.trackers[SyncItemTicker] = &s
}
if m.config.SynchronizeOrderbook {
s := s
c.trackers[SyncItemOrderbook] = &s
}
if m.config.SynchronizeTrades {
s := s
c.trackers[SyncItemTrade] = &s
}
if m.config.SynchronizeTicker {
if m.config.Verbose {
log.Debugf(log.SyncMgr,
"%s: Added ticker sync item %v: using websocket: %v using REST: %v",
c.Key.Exchange,
m.FormatCurrency(c.Pair),
c.trackers[SyncItemTicker].IsUsingWebsocket,
c.trackers[SyncItemTicker].IsUsingREST)
}
if atomic.LoadInt32(&m.initSyncCompleted) != 1 {
m.initSyncWG.Add(1)
createdCounter++
}
}
if m.config.SynchronizeOrderbook {
if m.config.Verbose {
log.Debugf(log.SyncMgr,
"%s: Added orderbook sync item %v: using websocket: %v using REST: %v",
k.Exchange,
m.FormatCurrency(c.Pair),
c.trackers[SyncItemOrderbook].IsUsingWebsocket,
c.trackers[SyncItemOrderbook].IsUsingREST)
}
if atomic.LoadInt32(&m.initSyncCompleted) != 1 {
m.initSyncWG.Add(1)
createdCounter++
}
}
if m.config.SynchronizeTrades {
if m.config.Verbose {
log.Debugf(log.SyncMgr,
"%s: Added trade sync item %v: using websocket: %v using REST: %v",
k.Exchange,
m.FormatCurrency(c.Pair),
c.trackers[SyncItemTrade].IsUsingWebsocket,
c.trackers[SyncItemTrade].IsUsingREST)
}
if atomic.LoadInt32(&m.initSyncCompleted) != 1 {
m.initSyncWG.Add(1)
createdCounter++
}
}
if m.currencyPairs == nil {
m.currencyPairs = make(map[key.ExchangePairAsset]*currencyPairSyncAgent)
}
m.currencyPairs[k] = c
return c
}
// WebsocketUpdate notifies the SyncManager to change the last updated time for a exchange asset pair
// And set IsUsingWebsocket to true. It should be used externally only from websocket updaters
func (m *SyncManager) WebsocketUpdate(exchangeName string, p currency.Pair, a asset.Item, syncType syncItemType, err error) error {
if m == nil {
return fmt.Errorf("exchange CurrencyPairSyncer %w", ErrNilSubsystem)
}
if atomic.LoadInt32(&m.started) == 0 {
return fmt.Errorf("exchange CurrencyPairSyncer %w", ErrSubSystemNotStarted)
}
if atomic.LoadInt32(&m.initSyncStarted) != 1 {
return nil
}
switch syncType {
case SyncItemOrderbook:
if !m.config.SynchronizeOrderbook {
return nil
}
case SyncItemTicker:
if !m.config.SynchronizeTicker {
return nil
}
case SyncItemTrade:
if !m.config.SynchronizeTrades {
return nil
}
default:
return fmt.Errorf("%v %w", syncType, errUnknownSyncItem)
}
k := key.ExchangePairAsset{
Asset: a,
Exchange: exchangeName,
Base: p.Base.Item,
Quote: p.Quote.Item,
}
c, exists := m.currencyPairs[k]
if !exists {
return fmt.Errorf("%w for %s %s %s %s %s",
errCouldNotSyncNewData,
k.Exchange,
k.Base,
k.Quote,
k.Asset,
syncType)
}
c.locks[syncType].Lock()
defer c.locks[syncType].Unlock()
if c.trackers[syncType] == nil {
c.trackers[syncType] = &syncBase{}
}
s := c.trackers[syncType]
if !s.IsUsingWebsocket {
s.IsUsingWebsocket = true
s.IsUsingREST = false
if m.config.LogSwitchProtocolEvents {
log.Warnf(log.SyncMgr,
"%s %s %s: %s Websocket re-enabled, switching from rest to websocket",
k.Exchange,
m.FormatCurrency(c.Pair),
strings.ToUpper(k.Asset.String()),
syncType,
)
}
}
return m.update(c, syncType, err)
}
// update notifies the SyncManager to change the last updated time for a exchange asset pair
func (m *SyncManager) update(c *currencyPairSyncAgent, syncType syncItemType, err error) error {
if syncType < SyncItemTicker || syncType > SyncItemTrade {
return fmt.Errorf("%v %w", syncType, errUnknownSyncItem)
}
s := c.trackers[syncType]
origHadData := s.HaveData
s.LastUpdated = time.Now()
if err != nil {
s.NumErrors++
}
s.HaveData = true
if atomic.LoadInt32(&m.initSyncCompleted) != 1 && !origHadData {
removedCounter++
if m.config.LogInitialSyncEvents {
log.Debugf(log.SyncMgr, "%s %s sync complete %v [%d/%d].",
c.Key.Exchange,
syncType,
m.FormatCurrency(c.Pair),
removedCounter,
createdCounter)
}
m.initSyncWG.Done()
}
return nil
}
func (m *SyncManager) worker() {
cleanup := func() {
log.Debugln(log.SyncMgr,
"Exchange CurrencyPairSyncer worker shutting down.")
}
defer cleanup()
interval := min(greatestCommonDivisor(m.config.TimeoutWebsocket, m.config.TimeoutREST), minSyncInterval)
t := time.NewTicker(interval)
for {
select {
case <-m.shutdown:
return
case <-t.C:
exchanges, err := m.exchangeManager.GetExchanges()
if err != nil {
log.Errorf(log.SyncMgr, "Sync manager cannot get exchanges: %v", err)
continue
}
for _, e := range exchanges {
exchangeName := e.GetName()
supportsREST := e.SupportsREST()
// TODO: These vars are only used for enabling new pairs, deriving them every cycle is sub-optimal
var usingREST bool
var usingWebsocket bool
if e.SupportsWebsocket() && e.IsWebsocketEnabled() {
ws, err := e.GetWebsocket()
if err != nil {
log.Errorf(log.SyncMgr,
"%s unable to get websocket pointer. Err: %s",
exchangeName,
err)
usingREST = true
}
if ws.IsConnected() {
usingWebsocket = true
} else {
usingREST = true
}
} else if supportsREST {
usingREST = true
}
assetTypes := e.GetAssetTypes(true)
for y := range assetTypes {
wsAssetSupported := e.IsAssetWebsocketSupported(assetTypes[y])
enabledPairs, err := e.GetEnabledPairs(assetTypes[y])
if err != nil {
log.Errorf(log.SyncMgr,
"%s failed to get enabled pairs. Err: %s",
e.GetName(),
err)
continue
}
for i := range enabledPairs {
if atomic.LoadInt32(&m.started) == 0 {
return
}
k := key.ExchangePairAsset{
Asset: assetTypes[y],
Exchange: exchangeName,
Base: enabledPairs[i].Base.Item,
Quote: enabledPairs[i].Quote.Item,
}
c := m.get(k)
if c == nil {
c = m.add(k, syncBase{
IsUsingREST: usingREST || !wsAssetSupported,
IsUsingWebsocket: usingWebsocket && wsAssetSupported,
})
}
if m.config.SynchronizeTicker {
m.syncTicker(c, e)
}
if m.config.SynchronizeOrderbook {
m.syncOrderbook(c, e)
}
if m.config.SynchronizeTrades {
m.syncTrades(c)
}
}
}
}
}
}
}
func (m *SyncManager) syncTicker(c *currencyPairSyncAgent, e exchange.IBotExchange) {
if !c.locks[SyncItemTicker].TryLock() {
return
}
defer c.locks[SyncItemTicker].Unlock()
exchangeName := e.GetName()
s := c.trackers[SyncItemTicker]
if s.IsUsingWebsocket &&
e.SupportsREST() &&
time.Since(s.LastUpdated) > m.config.TimeoutWebsocket &&
time.Since(c.Created) > m.config.TimeoutWebsocket {
// Downgrade to REST
s.IsUsingWebsocket = false
s.IsUsingREST = true
if m.config.LogSwitchProtocolEvents {
log.Warnf(log.SyncMgr,
"%s %s %s: No ticker update after %s, switching from websocket to rest",
c.Key.Exchange,
m.FormatCurrency(c.Pair),
strings.ToUpper(c.Key.Asset.String()),
m.config.TimeoutWebsocket,
)
}
}
if s.IsUsingREST && time.Since(s.LastUpdated) > m.config.TimeoutREST {
var result *ticker.Price
var err error
if e.SupportsRESTTickerBatchUpdates() {
m.mux.Lock()
batchLastDone, ok := m.tickerBatchLastRequested[key.ExchangeAsset{
Exchange: c.Key.Exchange,
Asset: c.Key.Asset,
}]
if !ok {
m.tickerBatchLastRequested[key.ExchangeAsset{
Exchange: c.Key.Exchange,
Asset: c.Key.Asset,
}] = time.Time{}
}
m.mux.Unlock()
if batchLastDone.IsZero() || time.Since(batchLastDone) > m.config.TimeoutREST {
m.mux.Lock()
if m.config.Verbose {
log.Debugf(log.SyncMgr, "Initialising %s REST ticker batching", exchangeName)
}
err = e.UpdateTickers(context.TODO(), c.Key.Asset)
if err == nil {
result, err = e.GetCachedTicker(c.Pair, c.Key.Asset)
}
m.tickerBatchLastRequested[key.ExchangeAsset{
Exchange: c.Key.Exchange,
Asset: c.Key.Asset,
}] = time.Now()
m.mux.Unlock()
} else {
if m.config.Verbose {
log.Debugf(log.SyncMgr, "%s Using recent batching cache", exchangeName)
}
result, err = e.GetCachedTicker(c.Pair, c.Key.Asset)
}
} else {
result, err = e.UpdateTicker(context.TODO(),
c.Pair,
c.Key.Asset)
}
m.PrintTickerSummary(result, "REST", err)
if err == nil {
if m.remoteConfig.WebsocketRPC.Enabled {
relayWebsocketEvent(result, "ticker_update", c.Key.Asset.String(), exchangeName)
}
}
updateErr := m.update(c, SyncItemTicker, err)
if updateErr != nil {
log.Errorln(log.SyncMgr, updateErr)
}
}
}
func (m *SyncManager) syncOrderbook(c *currencyPairSyncAgent, e exchange.IBotExchange) {
if !c.locks[SyncItemOrderbook].TryLock() {
return
}
defer c.locks[SyncItemOrderbook].Unlock()
s := c.trackers[SyncItemOrderbook]
if s.IsUsingWebsocket &&
e.SupportsREST() &&
time.Since(s.LastUpdated) > m.config.TimeoutWebsocket &&
time.Since(c.Created) > m.config.TimeoutWebsocket {
if w, err := e.GetWebsocket(); err == nil && w.IsConnected() {
// With an active websocket connection, we can assume the orderbook
// is being updated via the websocket connection, It could be very
// illiquid. There is no need to fall over to rest.
return
}
// Downgrade to REST
s.IsUsingWebsocket = false
s.IsUsingREST = true
if m.config.LogSwitchProtocolEvents {
log.Warnf(log.SyncMgr,
"%s %s %s: No orderbook update after %s, switching from websocket to rest",
c.Key.Exchange,
m.FormatCurrency(c.Pair),
strings.ToUpper(c.Key.Asset.String()),
m.config.TimeoutWebsocket,
)
}
}
if s.IsUsingREST && time.Since(s.LastUpdated) > m.config.TimeoutREST {
result, err := e.UpdateOrderbook(context.TODO(),
c.Pair,
c.Key.Asset)
m.PrintOrderbookSummary(result, "REST", err)
if err == nil {
if m.remoteConfig.WebsocketRPC.Enabled {
relayWebsocketEvent(result, "orderbook_update", c.Key.Asset.String(), e.GetName())
}
}
updateErr := m.update(c, SyncItemOrderbook, err)
if updateErr != nil {
log.Errorln(log.SyncMgr, updateErr)
}
}
}
func (m *SyncManager) syncTrades(c *currencyPairSyncAgent) {
if !c.locks[SyncItemTrade].TryLock() {
return
}
defer c.locks[SyncItemTrade].Unlock()
if time.Since(c.trackers[SyncItemTrade].LastUpdated) > m.config.TimeoutREST {
err := m.update(c, SyncItemTrade, nil)
if err != nil {
log.Errorln(log.SyncMgr, err)
}
}
}
func printCurrencyFormat(price float64, displayCurrency currency.Code) string {
displaySymbol, err := currency.GetSymbolByCurrencyName(displayCurrency)
if err != nil {
log.Errorf(log.SyncMgr, "Failed to get display symbol: %s", err)
}
return fmt.Sprintf("%s%.8f", displaySymbol, price)
}
func printConvertCurrencyFormat(origPrice float64, origCurrency, displayCurrency currency.Code) string {
var conv string
if origPrice > 0 {
if convFloat, err := currency.ConvertFiat(origPrice, origCurrency, displayCurrency); err != nil {
conv = "?.??"
} else {
conv = fmt.Sprintf("%.2f", convFloat)
}
} else {
conv = "0.00"
}
displaySymbol, err := currency.GetSymbolByCurrencyName(displayCurrency)
if err != nil {
log.Errorf(log.SyncMgr, "Failed to get display symbol: %s", err)
}
origSymbol, err := currency.GetSymbolByCurrencyName(origCurrency)
if err != nil {
log.Errorf(log.SyncMgr, "Failed to get original currency symbol for %s: %s",
origCurrency,
err)
}
return fmt.Sprintf("%s%s %s (%s%.2f %s)",
displaySymbol,
conv,
displayCurrency,
origSymbol,
origPrice,
origCurrency,
)
}
// PrintTickerSummary outputs the ticker results
func (m *SyncManager) PrintTickerSummary(result *ticker.Price, protocol string, err error) {
if m == nil || atomic.LoadInt32(&m.started) == 0 {
return
}
if !m.config.SynchronizeTicker {
return
}
if err != nil {
if err == common.ErrNotYetImplemented {
log.Warnf(log.SyncMgr, "Failed to get %s ticker. Error: %s",
protocol,
err)
return
}
log.Errorf(log.SyncMgr, "Failed to get %s ticker. Error: %s",
protocol,
err)
return
}
// ignoring error as not all tickers have volume populated and error is not actionable
_ = stats.Add(result.ExchangeName, result.Pair, result.AssetType, result.Last, result.Volume)
if !m.config.LogSyncUpdateEvents {
return
}
if currency.ForexEnabled() &&
result.Pair.Quote.IsFiatCurrency() &&
!result.Pair.Quote.Equal(m.fiatDisplayCurrency) &&
!m.fiatDisplayCurrency.IsEmpty() {
origCurrency := result.Pair.Quote.Upper()
log.Infof(log.SyncMgr, "%s %s %s %s TICKER: Last %s Ask %s Bid %s High %s Low %s Volume %.8f",
result.ExchangeName,
protocol,
m.FormatCurrency(result.Pair),
strings.ToUpper(result.AssetType.String()),
printConvertCurrencyFormat(result.Last, origCurrency, m.fiatDisplayCurrency),
printConvertCurrencyFormat(result.Ask, origCurrency, m.fiatDisplayCurrency),
printConvertCurrencyFormat(result.Bid, origCurrency, m.fiatDisplayCurrency),
printConvertCurrencyFormat(result.High, origCurrency, m.fiatDisplayCurrency),
printConvertCurrencyFormat(result.Low, origCurrency, m.fiatDisplayCurrency),
result.Volume)
} else {
if result.Pair.Quote.IsFiatCurrency() &&
result.Pair.Quote.Equal(m.fiatDisplayCurrency) &&
!m.fiatDisplayCurrency.IsEmpty() {
log.Infof(log.SyncMgr, "%s %s %s %s TICKER: Last %s Ask %s Bid %s High %s Low %s Volume %.8f",
result.ExchangeName,
protocol,
m.FormatCurrency(result.Pair),
strings.ToUpper(result.AssetType.String()),
printCurrencyFormat(result.Last, m.fiatDisplayCurrency),
printCurrencyFormat(result.Ask, m.fiatDisplayCurrency),
printCurrencyFormat(result.Bid, m.fiatDisplayCurrency),
printCurrencyFormat(result.High, m.fiatDisplayCurrency),
printCurrencyFormat(result.Low, m.fiatDisplayCurrency),
result.Volume)
} else {
log.Infof(log.SyncMgr, "%s %s %s %s TICKER: Last %.8f Ask %.8f Bid %.8f High %.8f Low %.8f Volume %.8f",
result.ExchangeName,
protocol,
m.FormatCurrency(result.Pair),
strings.ToUpper(result.AssetType.String()),
result.Last,
result.Ask,
result.Bid,
result.High,
result.Low,
result.Volume)
}
}
}
// FormatCurrency is a method that formats and returns a currency pair
// based on the user currency display preferences
func (m *SyncManager) FormatCurrency(cp currency.Pair) string {
if m == nil || atomic.LoadInt32(&m.started) == 0 {
return ""
}
return m.format.Format(cp)
}
const (
book = "%s %s %s %s ORDERBOOK: Bids len: %d Amount: %f %s. Total value: %s Asks len: %d Amount: %f %s. Total value: %s"
)
// PrintOrderbookSummary outputs orderbook results
func (m *SyncManager) PrintOrderbookSummary(result *orderbook.Base, protocol string, err error) {
if m == nil || atomic.LoadInt32(&m.started) == 0 {
return
}
if !m.config.SynchronizeOrderbook {
return
}
if err != nil {
if result == nil {
log.Errorf(log.OrderBook, "Failed to get %s orderbook. Error: %s",
protocol,
err)
return
}
if err == common.ErrNotYetImplemented {
log.Warnf(log.OrderBook, "Failed to get %s orderbook for %s %s %s. Error: %s",
protocol,
result.Exchange,
result.Pair,
result.Asset,
err)
return
}
log.Errorf(log.OrderBook, "Failed to get %s orderbook for %s %s %s. Error: %s",
protocol,
result.Exchange,
result.Pair,
result.Asset,
err)
return
}
if !m.config.LogSyncUpdateEvents {
return
}
bidsAmount, bidsValue := result.TotalBidsAmount()
asksAmount, asksValue := result.TotalAsksAmount()
var bidValueResult, askValueResult string
switch {
case currency.ForexEnabled() && result.Pair.Quote.IsFiatCurrency() && !result.Pair.Quote.Equal(m.fiatDisplayCurrency) && !m.fiatDisplayCurrency.IsEmpty():
origCurrency := result.Pair.Quote.Upper()
if bidsValue > 0 {
bidValueResult = printConvertCurrencyFormat(bidsValue, origCurrency, m.fiatDisplayCurrency)
}
if asksValue > 0 {
askValueResult = printConvertCurrencyFormat(asksValue, origCurrency, m.fiatDisplayCurrency)
}
case result.Pair.Quote.IsFiatCurrency() && result.Pair.Quote.Equal(m.fiatDisplayCurrency) && !m.fiatDisplayCurrency.IsEmpty():
bidValueResult = printCurrencyFormat(bidsValue, m.fiatDisplayCurrency)
askValueResult = printCurrencyFormat(asksValue, m.fiatDisplayCurrency)
default:
bidValueResult = strconv.FormatFloat(bidsValue, 'f', -1, 64)
askValueResult = strconv.FormatFloat(asksValue, 'f', -1, 64)
}
log.Infof(log.SyncMgr, book,
result.Exchange,
protocol,
m.FormatCurrency(result.Pair),
strings.ToUpper(result.Asset.String()),
len(result.Bids),
bidsAmount,
result.Pair.Base,
bidValueResult,
len(result.Asks),
asksAmount,
result.Pair.Base,
askValueResult,
)
}
// WaitForInitialSync allows for a routine to wait for an initial sync to be
// completed without exposing the underlying type. This needs to be called in a
// separate routine.
func (m *SyncManager) WaitForInitialSync() error {
if m == nil {
return fmt.Errorf("sync manager %w", ErrNilSubsystem)
}
m.inService.Wait()
if atomic.LoadInt32(&m.started) == 0 {
return fmt.Errorf("sync manager %w", ErrSubSystemNotStarted)
}
m.initSyncWG.Wait()
return nil
}
func relayWebsocketEvent(result any, event, assetType, exchangeName string) {
evt := WebsocketEvent{
Data: result,
Event: event,
AssetType: assetType,
Exchange: exchangeName,
}
err := BroadcastWebsocketMessage(evt)
if !errors.Is(err, ErrWebsocketServiceNotRunning) {
log.Errorf(log.APIServerMgr, "Failed to broadcast websocket event %v. Error: %s",
event, err)
}
}
func greatestCommonDivisor(a, b time.Duration) time.Duration {
for b != 0 {
t := b
b = a % b
a = t
}
return a
}
func (s syncItemType) String() string {
switch s {
case SyncItemTicker:
return "Ticker"
case SyncItemOrderbook:
return "Orderbook"
case SyncItemTrade:
return "Trade"
default:
return fmt.Sprintf("Invalid syncItemType: %d", s)
}
}