mirror of
https://github.com/grafana/grafana.git
synced 2024-11-26 02:40:26 -06:00
e79b2a3f66
* Core: add location option to parse timerange * Extensions: add go-datemath to not break Enterprise
133 lines
2.8 KiB
Go
133 lines
2.8 KiB
Go
package tsdb
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/timberio/go-datemath"
|
|
)
|
|
|
|
func NewTimeRange(from, to string) *TimeRange {
|
|
return &TimeRange{
|
|
From: from,
|
|
To: to,
|
|
now: time.Now(),
|
|
}
|
|
}
|
|
|
|
func NewFakeTimeRange(from, to string, now time.Time) *TimeRange {
|
|
return &TimeRange{
|
|
From: from,
|
|
To: to,
|
|
now: now,
|
|
}
|
|
}
|
|
|
|
type TimeRange struct {
|
|
From string
|
|
To string
|
|
now time.Time
|
|
}
|
|
|
|
func (tr *TimeRange) GetFromAsMsEpoch() int64 {
|
|
return tr.MustGetFrom().UnixNano() / int64(time.Millisecond)
|
|
}
|
|
|
|
func (tr *TimeRange) GetFromAsSecondsEpoch() int64 {
|
|
return tr.GetFromAsMsEpoch() / 1000
|
|
}
|
|
|
|
func (tr *TimeRange) GetFromAsTimeUTC() time.Time {
|
|
return tr.MustGetFrom().UTC()
|
|
}
|
|
|
|
func (tr *TimeRange) GetToAsMsEpoch() int64 {
|
|
return tr.MustGetTo().UnixNano() / int64(time.Millisecond)
|
|
}
|
|
|
|
func (tr *TimeRange) GetToAsSecondsEpoch() int64 {
|
|
return tr.GetToAsMsEpoch() / 1000
|
|
}
|
|
|
|
func (tr *TimeRange) GetToAsTimeUTC() time.Time {
|
|
return tr.MustGetTo().UTC()
|
|
}
|
|
|
|
func (tr *TimeRange) MustGetFrom() time.Time {
|
|
res, err := tr.ParseFrom()
|
|
if err != nil {
|
|
return time.Unix(0, 0)
|
|
}
|
|
return res
|
|
}
|
|
|
|
func (tr *TimeRange) MustGetTo() time.Time {
|
|
res, err := tr.ParseTo()
|
|
if err != nil {
|
|
return time.Unix(0, 0)
|
|
}
|
|
return res
|
|
}
|
|
|
|
func tryParseUnixMsEpoch(val string) (time.Time, bool) {
|
|
if val, err := strconv.ParseInt(val, 10, 64); err == nil {
|
|
seconds := val / 1000
|
|
nano := (val - seconds*1000) * 1000000
|
|
return time.Unix(seconds, nano), true
|
|
}
|
|
return time.Time{}, false
|
|
}
|
|
|
|
func (tr *TimeRange) ParseFrom() (time.Time, error) {
|
|
return parse(tr.From, tr.now, false, nil)
|
|
}
|
|
|
|
func (tr *TimeRange) ParseTo() (time.Time, error) {
|
|
return parse(tr.To, tr.now, true, nil)
|
|
}
|
|
|
|
func (tr *TimeRange) ParseFromWithLocation(location *time.Location) (time.Time, error) {
|
|
return parse(tr.From, tr.now, false, location)
|
|
}
|
|
|
|
func (tr *TimeRange) ParseToWithLocation(location *time.Location) (time.Time, error) {
|
|
return parse(tr.To, tr.now, true, location)
|
|
}
|
|
|
|
func parse(s string, now time.Time, withRoundUp bool, location *time.Location) (time.Time, error) {
|
|
if res, ok := tryParseUnixMsEpoch(s); ok {
|
|
return res, nil
|
|
}
|
|
|
|
diff, err := time.ParseDuration("-" + s)
|
|
if err != nil {
|
|
options := []func(*datemath.Options){
|
|
datemath.WithNow(now),
|
|
datemath.WithRoundUp(withRoundUp),
|
|
}
|
|
if location != nil {
|
|
options = append(options, datemath.WithLocation(location))
|
|
}
|
|
|
|
return datemath.ParseAndEvaluate(s, options...)
|
|
}
|
|
|
|
return now.Add(diff), nil
|
|
}
|
|
|
|
// EpochPrecisionToMs converts epoch precision to millisecond, if needed.
|
|
// Only seconds to milliseconds supported right now
|
|
func EpochPrecisionToMs(value float64) float64 {
|
|
s := strconv.FormatFloat(value, 'e', -1, 64)
|
|
if strings.HasSuffix(s, "e+09") {
|
|
return value * float64(1e3)
|
|
}
|
|
|
|
if strings.HasSuffix(s, "e+18") {
|
|
return value / float64(time.Millisecond)
|
|
}
|
|
|
|
return value
|
|
}
|