mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-14 07:26:47 +00:00
d01e7bad72
* Added new base logger * updated example and test configs * updated exchange helpers restful router & server * logPath is now passed to the logger to remove dependency on common package * updated everything besides exchanges to use new logger * alphapoint to bitmex done * updated bitmex bitstamp bittrex btcc and also performance changes to logger * btcmarkets coinbase coinut exmo gateio wrappers updated * gateio and gemini logger updated * hitbtc huobi itbit & kraken updated * All exchanges updatd * return correct error for disabled websocket * don't disconnect client on invalid json * updated router internal logging * log.Fatal to t.Error for tests * Changed from fatal to error failure to set maxprocs * output ANSI codes for everything but windows for now due to lack of windows support * added error handling to logger and unit tests * clear wording on print -> log.print * added benchmark test * cleaned up import sections * Updated logger based on PR requests (added default config options on failure/setting errors) * ah this should fix travici enc config issue * Load entire config and clear out logging to hopefully fix travisci issue * wording & test error handling * fixed formatting issues based on feedback * fixed formatting issues based on feedback * changed CheckDir to use mkdirall instead of mkdir and other changes based on feedback
109 lines
3.1 KiB
Go
109 lines
3.1 KiB
Go
package base
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/thrasher-/gocryptotrader/config"
|
|
"github.com/thrasher-/gocryptotrader/exchanges/orderbook"
|
|
"github.com/thrasher-/gocryptotrader/exchanges/ticker"
|
|
log "github.com/thrasher-/gocryptotrader/logger"
|
|
)
|
|
|
|
// IComm is the main interface array across the communication packages
|
|
type IComm []ICommunicate
|
|
|
|
// ICommunicate enforces standard functions across communication packages
|
|
type ICommunicate interface {
|
|
Setup(config config.CommunicationsConfig)
|
|
Connect() error
|
|
PushEvent(Event) error
|
|
IsEnabled() bool
|
|
IsConnected() bool
|
|
GetName() string
|
|
}
|
|
|
|
// Setup sets up communication variables and intiates a connection to the
|
|
// communication mediums
|
|
func (c IComm) Setup() {
|
|
TickerStaged = make(map[string]map[string]map[string]ticker.Price)
|
|
OrderbookStaged = make(map[string]map[string]map[string]Orderbook)
|
|
ServiceStarted = time.Now()
|
|
|
|
for i := range c {
|
|
if c[i].IsEnabled() && !c[i].IsConnected() {
|
|
err := c[i].Connect()
|
|
if err != nil {
|
|
log.Errorf("Communications: %s failed to connect. Err: %s", c[i].GetName(), err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// PushEvent pushes triggered events to all enabled communication links
|
|
func (c IComm) PushEvent(event Event) {
|
|
for i := range c {
|
|
if c[i].IsEnabled() && c[i].IsConnected() {
|
|
err := c[i].PushEvent(event)
|
|
if err != nil {
|
|
log.Errorf("Communications error - PushEvent() in package %s with %v",
|
|
c[i].GetName(), event)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// GetEnabledCommunicationMediums prints out enabled and connected communication
|
|
// packages
|
|
func (c IComm) GetEnabledCommunicationMediums() {
|
|
var count int
|
|
for i := range c {
|
|
if c[i].IsEnabled() && c[i].IsConnected() {
|
|
log.Debugf("Communications: Medium %s is enabled.", c[i].GetName())
|
|
count++
|
|
}
|
|
}
|
|
if count == 0 {
|
|
log.Warnf("Communications: No communication mediums are enabled.")
|
|
}
|
|
}
|
|
|
|
// StageTickerData stages updated ticker data for the communications package
|
|
func (c IComm) StageTickerData(exchangeName, assetType string, tickerPrice ticker.Price) {
|
|
m.Lock()
|
|
defer m.Unlock()
|
|
|
|
if _, ok := TickerStaged[exchangeName]; !ok {
|
|
TickerStaged[exchangeName] = make(map[string]map[string]ticker.Price)
|
|
}
|
|
|
|
if _, ok := TickerStaged[exchangeName][assetType]; !ok {
|
|
TickerStaged[exchangeName][assetType] = make(map[string]ticker.Price)
|
|
}
|
|
|
|
TickerStaged[exchangeName][assetType][tickerPrice.CurrencyPair] = tickerPrice
|
|
}
|
|
|
|
// StageOrderbookData stages updated orderbook data for the communications
|
|
// package
|
|
func (c IComm) StageOrderbookData(exchangeName, assetType string, orderbook orderbook.Base) {
|
|
m.Lock()
|
|
defer m.Unlock()
|
|
|
|
if _, ok := OrderbookStaged[exchangeName]; !ok {
|
|
OrderbookStaged[exchangeName] = make(map[string]map[string]Orderbook)
|
|
}
|
|
|
|
if _, ok := OrderbookStaged[exchangeName][assetType]; !ok {
|
|
OrderbookStaged[exchangeName][assetType] = make(map[string]Orderbook)
|
|
}
|
|
|
|
_, totalAsks := orderbook.CalculateTotalAsks()
|
|
_, totalBids := orderbook.CalculateTotalBids()
|
|
|
|
OrderbookStaged[exchangeName][assetType][orderbook.CurrencyPair] = Orderbook{
|
|
CurrencyPair: orderbook.CurrencyPair,
|
|
TotalAsks: totalAsks,
|
|
TotalBids: totalBids,
|
|
LastUpdated: orderbook.LastUpdated.String()}
|
|
}
|