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

backtest: assign user data stream to backtest.Exchange before we call EmitStart

Browse Source
This commit is contained in:
c9s 2022-07-04 02:21:14 +08:00
parent 449b2d8220
commit ecd4df86f9
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
3 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pkg/backtest/matching.go
View File

@ -99,7 +99,6 @@ func (m *SimplePriceMatching) CancelOrder(o types.Order) (types.Order, error) {
}
m.askOrders = orders
m.mu.Unlock()
}
if !found {
@ -191,6 +190,8 @@ func (m *SimplePriceMatching) PlaceOrder(o types.SubmitOrder) (*types.Order, *ty
order2.ExecutedQuantity = order2.Quantity
order2.IsWorking = false
m.EmitOrderUpdate(order2)
// let the exchange emit the "FILLED" order update (we need the closed order)
// m.EmitOrderUpdate(order2)
return &order2, &trade, nil

2
pkg/bbgo/activeorderbook.go
View File

@ -131,8 +131,6 @@ func (b *ActiveOrderBook) orderUpdateHandler(order types.Order) {
return
}
log.Debugf("[ActiveOrderBook] received order update: %+v", order)
switch order.Status {
case types.OrderStatusFilled:
// make sure we have the order and we remove it

16
pkg/cmd/backtest.go
View File

@ -267,6 +267,12 @@ var BacktestCmd = &cobra.Command{
return err
}
for _, session := range environ.Sessions() {
backtestEx := session.Exchange.(*backtest.Exchange)
backtestEx.UserDataStream = session.UserDataStream.(types.StandardStreamEmitter)
backtestEx.MarketDataStream = session.MarketDataStream.(types.StandardStreamEmitter)
}
trader := bbgo.NewTrader(environ)
if verboseCnt == 0 {
trader.DisableLogging()
@ -642,12 +648,10 @@ func confirmation(s string) bool {
func toExchangeSources(sessions map[string]*bbgo.ExchangeSession, extraIntervals ...types.Interval) (exchangeSources []backtest.ExchangeDataSource, err error) {
for _, session := range sessions {
exchange := session.Exchange.(*backtest.Exchange)
exchange.UserDataStream = session.UserDataStream.(types.StandardStreamEmitter)
exchange.MarketDataStream = session.MarketDataStream.(types.StandardStreamEmitter)
exchange.InitMarketData()
backtestEx := session.Exchange.(*backtest.Exchange)
backtestEx.InitMarketData()
c, err := exchange.SubscribeMarketData(extraIntervals...)
c, err := backtestEx.SubscribeMarketData(extraIntervals...)
if err != nil {
return exchangeSources, err
}
@ -655,7 +659,7 @@ func toExchangeSources(sessions map[string]*bbgo.ExchangeSession, extraIntervals
sessionCopy := session
exchangeSources = append(exchangeSources, backtest.ExchangeDataSource{
C: c,
Exchange: exchange,
Exchange: backtestEx,
Session: sessionCopy,
})
}