New communications package

Support for Slack, SMSGlobal, SMTP and Telegram

Supersedes: https://github.com/thrasher-/gocryptotrader/pull/126

communications/base/base_interface.go

@@ -0,0 +1,108 @@
+package base
+
+import (
+	"log"
+	"time"
+
+	"github.com/thrasher-/gocryptotrader/config"
+	"github.com/thrasher-/gocryptotrader/exchanges/orderbook"
+	"github.com/thrasher-/gocryptotrader/exchanges/ticker"
+)
+
+// 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.Printf("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.Printf("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.Printf("Communications: Medium %s is enabled.", c[i].GetName())
+			count++
+		}
+	}
+	if count == 0 {
+		log.Println("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()}
+}