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

fix channel type and add map guard

Browse Source
This commit is contained in:
c9s 2020-10-03 19:42:17 +08:00
parent 30a5193923
commit 9d26adb245
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
bbgo/trader.go
View File

@ -43,7 +43,7 @@ type ExchangeSession struct {
Trades map[string][]types.Trade
}
func (session *ExchangeSession) Subscribe(channel string, symbol string, options types.SubscribeOptions) *ExchangeSession {
func (session *ExchangeSession) Subscribe(channel types.Channel, symbol string, options types.SubscribeOptions) *ExchangeSession {
session.Symbols(symbol)
session.Subscriptions = append(session.Subscriptions, types.Subscription{
@ -56,6 +56,14 @@ func (session *ExchangeSession) Subscribe(channel string, symbol string, options
}
func (session *ExchangeSession) Symbols(symbols ...string) *ExchangeSession {
if session.loadedSymbols == nil {
session.loadedSymbols = make(map[string]struct{})
}
if session.Markets == nil {
session.Markets = make(map[string]types.Market)
}
for _, symbol := range symbols {
session.loadedSymbols[symbol] = struct{}{}
@ -117,6 +125,10 @@ func (trader *Trader) AddExchange(name string, exchange *binance.Exchange) (sess
Exchange: exchange,
}
if trader.ExchangeSessions == nil {
trader.ExchangeSessions = make(map[string]*ExchangeSession)
}
trader.ExchangeSessions[name] = session
return session
}
@ -146,6 +158,9 @@ func (trader *Trader) Connect(ctx context.Context) (err error) {
}
log.Infof("symbol %s: %d trades loaded", symbol, len(trades))
if session.Trades == nil {
session.Trades = make(map[string][]types.Trade)
}
session.Trades[symbol] = trades
stockManager := &StockManager{