mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 23:16:45 +00:00
46747c13f4
* GetClosedOrder implemented for Kraken and Binance, fixed Binance MARKET order creaton, added rate, fee and cost fileds on SubmitOrder responce * return Trades on Binance SubmitOrder, new validation methods on Binance and kraken GetClosedOrderInfo * removed the Binance extra method GetClosedOrder * func description corrected * removed price, fee and cost from SimulateOrder response, as we get all necessary info in response to calculate them on client side * GetClosedOrder implementation moved to GetOrderInfo * changed GetOrderInfo params * removed Canceled order.Type used for Kraken * update QueryOrder in gctscript * add missed params to QueryOrder validator (gctscript) * fixed testing issues * GetClosedOrder implemented for Kraken and Binance, fixed Binance MARKET order creaton, added rate, fee and cost fileds on SubmitOrder responce * return Trades on Binance SubmitOrder, new validation methods on Binance and kraken GetClosedOrderInfo * removed the Binance extra method GetClosedOrder * func description corrected * removed price, fee and cost from SimulateOrder response, as we get all necessary info in response to calculate them on client side * GetClosedOrder implementation moved to GetOrderInfo * changed GetOrderInfo params * removed Canceled order.Type used for Kraken * update QueryOrder in gctscript * add missed params to QueryOrder validator (gctscript) * fixed testing issues * pull previous changes * linter issues fix * updated query_order exmple in gctscript, fixed params check * removed orderPair unnecessary conversion * added wsCancelAllOrders, fixed bugs * fixed Kraken wsAddOrder method * cleanup * CancelBatchOrders implementation * changed CancelBatchOrders signature * fixed tests and wrappers * btcmarkets_test fix * cleanup * cleanup * changed CancelBatchOrders signature * fmt * Update configtest.json * Update configtest.json * rollback configtest * refactored Kraken wsHandleData to allow tests * removed unnecessary error test in TestWsAddOrderJSON * dependencies updates * fixed issue with PortfolioSleepDelay set on startup Co-authored-by: Vazha Bezhanishvili <vazha.bezhanishvili@elegro.eu>
94 lines
2.1 KiB
Go
94 lines
2.1 KiB
Go
package engine
|
|
|
|
import (
|
|
"errors"
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/thrasher-corp/gocryptotrader/log"
|
|
"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(Bot.Settings.PortfolioManagerDelay)
|
|
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.")
|
|
}()
|
|
|
|
p.processPortfolio()
|
|
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(Bot.GetAllEnabledExchangeAccountInfo().Data)
|
|
}
|