mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 23:16:45 +00:00
98a277a4c3
* bug fix for websocket orderbook processing * Fix more panics * fix linter issue * kick panic can down the road * temp fix for issue with a 404 returned error as chainz.cryptoid dropped eth support * Address nits and fixed orderbook updating * Fix trade data, rm'd event time from struct * fix time conversion for huobi * Actually process kline data and fix time stamps * btse time conversion fix and RM log, as it seems that the gain is reflecting transaction side. Drop ticker fetching support because there does not seem to be support on docs. And added trade fetching support. * revert huobi println * Adressed suggestion * rm unnecessary assignment * rm unnecessary check and assign * fix conversion mishap * fix currency conversion bug * update websocket logging * RM websocket type which stops conversion and copy * fix linter issue, add in unknown side type
94 lines
2.0 KiB
Go
94 lines
2.0 KiB
Go
package engine
|
|
|
|
import (
|
|
"errors"
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
log "github.com/thrasher-corp/gocryptotrader/logger"
|
|
"github.com/thrasher-corp/gocryptotrader/portfolio"
|
|
)
|
|
|
|
// vars for the fund manager package
|
|
var (
|
|
PortfolioSleepDelay = time.Minute
|
|
)
|
|
|
|
type portfolioManager struct {
|
|
started int32
|
|
stopped int32
|
|
shutdown chan struct{}
|
|
}
|
|
|
|
func (p *portfolioManager) Started() bool {
|
|
return atomic.LoadInt32(&p.started) == 1
|
|
}
|
|
|
|
func (p *portfolioManager) Start() error {
|
|
if atomic.AddInt32(&p.started, 1) != 1 {
|
|
return errors.New("portfolio manager already started")
|
|
}
|
|
|
|
log.Debugln(log.PortfolioMgr, "Portfolio manager starting...")
|
|
Bot.Portfolio = &portfolio.Portfolio
|
|
Bot.Portfolio.Seed(Bot.Config.Portfolio)
|
|
p.shutdown = make(chan struct{})
|
|
portfolio.Verbose = Bot.Settings.Verbose
|
|
|
|
go p.run()
|
|
return nil
|
|
}
|
|
func (p *portfolioManager) Stop() error {
|
|
if atomic.AddInt32(&p.stopped, 1) != 1 {
|
|
return errors.New("portfolio manager is already stopped")
|
|
}
|
|
|
|
log.Debugln(log.PortfolioMgr, "Portfolio manager shutting down...")
|
|
close(p.shutdown)
|
|
return nil
|
|
}
|
|
|
|
func (p *portfolioManager) run() {
|
|
log.Debugln(log.PortfolioMgr, "Portfolio manager started.")
|
|
Bot.ServicesWG.Add(1)
|
|
tick := time.NewTicker(PortfolioSleepDelay)
|
|
defer func() {
|
|
atomic.CompareAndSwapInt32(&p.stopped, 1, 0)
|
|
atomic.CompareAndSwapInt32(&p.started, 1, 0)
|
|
tick.Stop()
|
|
Bot.ServicesWG.Done()
|
|
log.Debugf(log.PortfolioMgr, "Portfolio manager shutdown.")
|
|
}()
|
|
|
|
for {
|
|
select {
|
|
case <-p.shutdown:
|
|
return
|
|
|
|
case <-tick.C:
|
|
p.processPortfolio()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (p *portfolioManager) processPortfolio() {
|
|
pf := portfolio.GetPortfolio()
|
|
data := pf.GetPortfolioGroupedCoin()
|
|
for key, value := range data {
|
|
err := pf.UpdatePortfolio(value, key)
|
|
if err != nil {
|
|
log.Errorf(log.PortfolioMgr,
|
|
"PortfolioWatcher error %s for currency %s\n",
|
|
err,
|
|
key)
|
|
continue
|
|
}
|
|
|
|
log.Debugf(log.PortfolioMgr,
|
|
"Portfolio manager: Successfully updated address balance for %s address(es) %s\n",
|
|
key,
|
|
value)
|
|
}
|
|
SeedExchangeAccountInfo(GetAllEnabledExchangeAccountInfo().Data)
|
|
}
|