lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-22 14:55:16 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

let strategy attach could be chained

Browse Source
This commit is contained in:
c9s 2020-10-16 13:52:18 +08:00
parent 4335cca0de
commit 25b4b22077
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/bbgo/trader.go
View File

@ -65,7 +65,7 @@ func NewTrader(environ *Environment) *Trader {
// AttachStrategy attaches the single exchange strategy on an exchange session.
// Single exchange strategy is the default behavior.
func (trader *Trader) AttachStrategy(session string, strategies ...SingleExchangeStrategy) {
func (trader *Trader) AttachStrategy(session string, strategies ...SingleExchangeStrategy) *Trader {
if _, ok := trader.environment.sessions[session]; !ok {
log.Panicf("session %s is not defined", session)
}
@ -73,12 +73,15 @@ func (trader *Trader) AttachStrategy(session string, strategies ...SingleExchang
for _, s := range strategies {
trader.exchangeStrategies[session] = append(trader.exchangeStrategies[session], s)
}
return trader
}
// AttachCrossExchangeStrategy attaches the cross exchange strategy
func (trader *Trader) AttachCrossExchangeStrategy(strategy CrossExchangeStrategy) error {
func (trader *Trader) AttachCrossExchangeStrategy(strategy CrossExchangeStrategy) *Trader {
trader.crossExchangeStrategies = append(trader.crossExchangeStrategies, strategy)
return nil
return trader
}
func (trader *Trader) Run(ctx context.Context) error {