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

fix graceful shutdown

Browse Source
This commit is contained in:
c9s 2020-07-13 13:31:40 +08:00
parent 9b042b5e8f
commit 63dc2e8d55
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
bbgo/trader.go
View File

@ -26,16 +26,16 @@ type Strategy interface {
Init(trader *Trader, stream *binance.PrivateStream) error
}
func (t *Trader) RunStrategy(ctx context.Context, strategy Strategy) error {
func (t *Trader) RunStrategy(ctx context.Context, strategy Strategy) (chan struct{}, error) {
symbol := t.Context.Symbol
stream, err := t.Exchange.NewPrivateStream()
if err != nil {
return err
return nil, err
}
if err := strategy.Init(t, stream); err != nil {
return err
return nil, err
}
t.reportTimer = time.AfterFunc(1*time.Second, func() {
@ -65,10 +65,13 @@ func (t *Trader) RunStrategy(ctx context.Context, strategy Strategy) error {
var eventC = make(chan interface{}, 20)
if err := stream.Connect(ctx, eventC); err != nil {
return err
return nil, err
}
done := make(chan struct{})
go func() {
defer close(done)
defer stream.Close()
for {
@ -84,7 +87,7 @@ func (t *Trader) RunStrategy(ctx context.Context, strategy Strategy) error {
}
}()
return nil
return done, nil
}
func (t *Trader) Infof(format string, args ...interface{}) {