diff --git a/utils/utils.go b/utils/utils.go index cba4891f..31e755b5 100644 --- a/utils/utils.go +++ b/utils/utils.go @@ -97,6 +97,27 @@ func GetCurDate() time.Time { return time.Date(year, month, day, 0, 0, 0, 0, now.Location()) } +func GetDate(dateTime time.Time) time.Time { + year, month, day := dateTime.Date() + return time.Date(year, month, day, 0, 0, 0, 0, dateTime.Location()) +} + +func GetTimeRange(baseTime time.Time, dayNum int, includeBaseTime bool) (beginTime, endTime time.Time) { + if dayNum <= 0 { + dayNum = 1 + } + baseTime = GetDate(baseTime) + if includeBaseTime { + dayNum = dayNum - 1 + endTime = baseTime.Add(time.Hour*24 - time.Second) + } else { + endTime = baseTime.Add(-time.Second) + } + beginTime = baseTime.Add(-time.Hour * 24 * time.Duration(dayNum)) + + return beginTime, endTime +} + // timestamp is in second func GetCurTimestamp() int64 { return time.Now().Unix()