add DailyDataTracker util

Author: narumiruna

pkg/strategy/common/fee_budget.go

@@ -2,10 +2,10 @@ package common
 
 import (
 	"sync"
-	"time"
 
 	"github.com/c9s/bbgo/pkg/fixedpoint"
 	"github.com/c9s/bbgo/pkg/types"
+	"github.com/c9s/bbgo/pkg/util"
 	log "github.com/sirupsen/logrus"
 )
 
@@ -73,20 +73,12 @@ func (f *FeeBudget) HandleTradeUpdate(trade types.Trade) {
 }
 
 type State struct {
-	AccumulatedFeeStartedAt time.Time                   `json:"accumulatedFeeStartedAt,omitempty"`
-	AccumulatedFees         map[string]fixedpoint.Value `json:"accumulatedFees,omitempty"`
-}
+	util.DailyDataTracker
 
-func (s *State) IsOver24Hours() bool {
-	return time.Since(s.AccumulatedFeeStartedAt) >= 24*time.Hour
+	AccumulatedFees map[string]fixedpoint.Value `json:"accumulatedFees,omitempty"`
 }
 
 func (s *State) Reset() {
-	t := time.Now()
-	dateTime := time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, t.Location())
-
-	log.Infof("[State] resetting accumulated started time to: %s", dateTime)
-
-	s.AccumulatedFeeStartedAt = dateTime
+	s.DailyDataTracker.Reset()
 	s.AccumulatedFees = make(map[string]fixedpoint.Value)
 }

pkg/strategy/common/fee_budget_test.go

@@ -50,7 +50,7 @@ func TestFeeBudget(t *testing.T) {
 		assert.Equal(t, c.expected, feeBudget.IsBudgetAllowed())
 
 		// test reset
-		feeBudget.State.AccumulatedFeeStartedAt = feeBudget.State.AccumulatedFeeStartedAt.Add(-24 * time.Hour)
+		feeBudget.State.StartedAt = feeBudget.State.StartedAt.Add(-24 * time.Hour)
 		assert.True(t, feeBudget.IsBudgetAllowed())
 	}
 }

pkg/util/countdown.go

@@ -0,0 +1,24 @@
+package util
+
+import (
+	"time"
+
+	log "github.com/sirupsen/logrus"
+)
+
+type DailyDataTracker struct {
+	StartedAt time.Time `json:"startedAt,omitempty"`
+}
+
+func (c *DailyDataTracker) IsOver24Hours() bool {
+	return time.Since(c.StartedAt) >= 24*time.Hour
+}
+
+func (s *DailyDataTracker) Reset() {
+	t := time.Now()
+	dateTime := time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, t.Location())
+
+	log.Infof("[Countdown] resetting accumulated started time to: %s", dateTime)
+
+	s.StartedAt = dateTime
+}