mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-23 23:35:14 +00:00
36 lines
730 B
Go
36 lines
730 B
Go
|
package util
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/c9s/bbgo/pkg/types"
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DailyDataTracker struct {
|
||
|
|
StartedAt time.Time `json:"startedAt,omitempty"`
|
||
|
|
Data types.ValueMap `json:"data,omitempty"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *DailyDataTracker) IsOver24Hours() bool {
|
||
|
|
return time.Since(d.StartedAt) >= 24*time.Hour
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *DailyDataTracker) ResetTime() {
|
||
|
|
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)
|
||
|
|
|
||
|
|
d.StartedAt = dateTime
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *DailyDataTracker) ResetData() {
|
||
|
|
d.Data = make(types.ValueMap)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *DailyDataTracker) Reset() {
|
||
|
|
d.ResetTime()
|
||
|
|
d.ResetData()
|
||
|
|
}
|