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

grid2: implement CleanUp interface

Browse Source
This commit is contained in:
c9s 2023-02-20 16:52:39 +08:00
parent 6fd5d1c79b
commit a6047c4840
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
1 changed files with 29 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
pkg/strategy/grid2/strategy.go
View File

@ -145,6 +145,9 @@ type Strategy struct {
GridProfitStats *GridProfitStats `persistence:"grid_profit_stats"`
Position *types.Position `persistence:"position"`
// ExchangeSession is an injection field
ExchangeSession *bbgo.ExchangeSession
grid *Grid
session *bbgo.ExchangeSession
orderQueryService types.ExchangeOrderQueryService
@ -200,6 +203,17 @@ func (s *Strategy) Validate() error {
}
func (s *Strategy) Defaults() error {
if s.LogFields == nil {
s.LogFields = logrus.Fields{}
}
s.LogFields["symbol"] = s.Symbol
s.LogFields["strategy"] = ID
return nil
}
func (s *Strategy) Initialize() error {
s.logger = log.WithFields(s.LogFields)
return nil
}
@ -1392,6 +1406,20 @@ func (s *Strategy) newPrometheusLabels() prometheus.Labels {
return mergeLabels(s.PrometheusLabels, labels)
}
func (s *Strategy) CleanUp(ctx context.Context) error {
if s.ExchangeSession == nil {
return errors.New("ExchangeSession is nil, can not clean up")
}
openOrders, err := s.ExchangeSession.Exchange.QueryOpenOrders(ctx, s.Symbol)
if err != nil {
return err
}
err = s.ExchangeSession.Exchange.CancelOrders(ctx, openOrders...)
return errors.Wrapf(err, "can not cancel %s orders", s.Symbol)
}
func (s *Strategy) Run(ctx context.Context, _ bbgo.OrderExecutor, session *bbgo.ExchangeSession) error {
instanceID := s.InstanceID()
@ -1401,14 +1429,6 @@ func (s *Strategy) Run(ctx context.Context, _ bbgo.OrderExecutor, session *bbgo.
s.orderQueryService = service
}
if s.LogFields == nil {
s.LogFields = logrus.Fields{}
}
s.LogFields["symbol"] = s.Symbol
s.LogFields["strategy"] = ID
s.logger = log.WithFields(s.LogFields)
if s.OrderGroupID == 0 {
s.OrderGroupID = util.FNV32(instanceID)
}
@ -1592,4 +1612,4 @@ func (s *Strategy) openOrdersMismatches(ctx context.Context, session *bbgo.Excha
}
return false, nil
}
}