mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 23:16:45 +00:00
3c72a199f2
* Updates starting and stopping routines to be a bit more parallel with less waiting required * Removes stop, removes debugging output * linting and test fixes * Add extra kill switch for exiting on exchange loading delay * Fixes fun math * breaks loop instead of switch. Moves param warns higher * Removes unceccary gos. passes in cfg to remove data race * Removes os signal processing. Fixes bad master merge
96 lines
2.2 KiB
Go
96 lines
2.2 KiB
Go
package engine
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"sync/atomic"
|
|
|
|
"github.com/thrasher-corp/gocryptotrader/communications"
|
|
"github.com/thrasher-corp/gocryptotrader/communications/base"
|
|
"github.com/thrasher-corp/gocryptotrader/engine/subsystem"
|
|
"github.com/thrasher-corp/gocryptotrader/log"
|
|
)
|
|
|
|
// commsManager starts the NTP manager
|
|
type commsManager struct {
|
|
started int32
|
|
shutdown chan struct{}
|
|
relayMsg chan base.Event
|
|
comms *communications.Communications
|
|
}
|
|
|
|
func (c *commsManager) Started() bool {
|
|
return atomic.LoadInt32(&c.started) == 1
|
|
}
|
|
|
|
func (c *commsManager) Start() (err error) {
|
|
if !atomic.CompareAndSwapInt32(&c.started, 0, 1) {
|
|
return fmt.Errorf("communications manager %w", subsystem.ErrSubSystemAlreadyStarted)
|
|
}
|
|
|
|
defer func() {
|
|
if err != nil {
|
|
atomic.CompareAndSwapInt32(&c.started, 1, 0)
|
|
}
|
|
}()
|
|
|
|
log.Debugln(log.CommunicationMgr, "Communications manager starting...")
|
|
commsCfg := Bot.Config.GetCommunicationsConfig()
|
|
c.comms, err = communications.NewComm(&commsCfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.shutdown = make(chan struct{})
|
|
c.relayMsg = make(chan base.Event)
|
|
go c.run()
|
|
log.Debugln(log.CommunicationMgr, "Communications manager started.")
|
|
return nil
|
|
}
|
|
|
|
func (c *commsManager) GetStatus() (map[string]base.CommsStatus, error) {
|
|
if !c.Started() {
|
|
return nil, errors.New("communications manager not started")
|
|
}
|
|
return c.comms.GetStatus(), nil
|
|
}
|
|
|
|
func (c *commsManager) Stop() error {
|
|
if atomic.LoadInt32(&c.started) == 0 {
|
|
return fmt.Errorf("communications manager %w", subsystem.ErrSubSystemNotStarted)
|
|
}
|
|
defer func() {
|
|
atomic.CompareAndSwapInt32(&c.started, 1, 0)
|
|
}()
|
|
close(c.shutdown)
|
|
log.Debugln(log.CommunicationMgr, "Communications manager shutting down...")
|
|
return nil
|
|
}
|
|
|
|
func (c *commsManager) PushEvent(evt base.Event) {
|
|
if !c.Started() {
|
|
return
|
|
}
|
|
select {
|
|
case c.relayMsg <- evt:
|
|
default:
|
|
log.Errorf(log.CommunicationMgr, "Failed to send, no receiver when pushing event [%v]", evt)
|
|
}
|
|
}
|
|
|
|
func (c *commsManager) run() {
|
|
defer func() {
|
|
// TO-DO shutdown comms connections for connected services (Slack etc)
|
|
log.Debugln(log.CommunicationMgr, "Communications manager shutdown.")
|
|
}()
|
|
|
|
for {
|
|
select {
|
|
case msg := <-c.relayMsg:
|
|
c.comms.PushEvent(msg)
|
|
case <-c.shutdown:
|
|
return
|
|
}
|
|
}
|
|
}
|