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

add defaulter interface

Browse Source
This commit is contained in:
c9s 2022-07-19 16:59:56 +08:00
parent 808ba2fc02
commit 8af2f2f83f
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pkg/bbgo/trader.go
View File

@ -24,10 +24,15 @@ type SingleExchangeStrategy interface {
Run(ctx context.Context, orderExecutor OrderExecutor, session *ExchangeSession) error Run(ctx context.Context, orderExecutor OrderExecutor, session *ExchangeSession) error
} }
// StrategyInitializer's Initialize method is called before the Subscribe method call.
type StrategyInitializer interface { type StrategyInitializer interface {
Initialize() error Initialize() error
} }
type StrategyDefaulter interface {
Defaults() error
}
// ExchangeSessionSubscriber provides an interface for collecting subscriptions from different strategies // ExchangeSessionSubscriber provides an interface for collecting subscriptions from different strategies
// Subscribe method will be called before the user data stream connection is created. // Subscribe method will be called before the user data stream connection is created.
type ExchangeSessionSubscriber interface { type ExchangeSessionSubscriber interface {
@ -153,6 +158,12 @@ func (trader *Trader) Subscribe() {
for sessionName, strategies := range trader.exchangeStrategies { for sessionName, strategies := range trader.exchangeStrategies {
session := trader.environment.sessions[sessionName] session := trader.environment.sessions[sessionName]
for _, strategy := range strategies { for _, strategy := range strategies {
if defaulter, ok := strategy.(StrategyDefaulter) ; ok {
if err := defaulter.Defaults(); err != nil {
panic(err)
}
}
if initializer, ok := strategy.(StrategyInitializer); ok { if initializer, ok := strategy.(StrategyInitializer); ok {
if err := initializer.Initialize(); err != nil { if err := initializer.Initialize(); err != nil {
panic(err) panic(err)