FEATURE: emit position when position updated and reset

c9s · Apr 11, 2024 · 0616c73 · 0616c73
1 parent e3a4887
commit 0616c73
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/pkg/strategy/dca2/state.go b/pkg/strategy/dca2/state.go
@@ -215,6 +215,9 @@ func (s *Strategy) runTakeProfitReady(ctx context.Context, next State) {
 	// reset position and open new round for profit stats before position opening
 	s.Position.Reset()
 
+	// emit position
+	s.EmitPosition(s.Position)
+
 	// store into redis
 	bbgo.Sync(ctx, s)
 

diff --git a/pkg/strategy/dca2/strategy.go b/pkg/strategy/dca2/strategy.go
@@ -237,6 +237,9 @@ func (s *Strategy) Run(ctx context.Context, _ bbgo.OrderExecutor, session *bbgo.
 
 		// update take profit price here
 		s.updateTakeProfitPrice()
+
+		// emit position update
+		s.EmitPosition(position)
 	})
 
 	s.OrderExecutor.ActiveMakerOrders().OnFilled(func(o types.Order) {
@@ -427,6 +430,14 @@ func (s *Strategy) ContinueNextRound() {
 	s.nextRoundPaused = false
 }
 
+func (s *Strategy) GetTakeProfitPrice() fixedpoint.Value {
+	if s.Position.Base == 0 {
+		return fixedpoint.Zero
+	}
+
+	return s.takeProfitPrice
+}
+
 func (s *Strategy) UpdateProfitStatsUntilSuccessful(ctx context.Context) error {
 	var op = func() error {
 		if updated, err := s.UpdateProfitStats(ctx); err != nil {