mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-23 00:46:23 +01:00
d6415b2572
'any' type is supported starting from Go1.18. Let's consistently use it instead of 'interface{}' type across the code base, since `any` is easier to read than 'interface{}'.
56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
package promutils
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/VictoriaMetrics/metricsql"
|
|
)
|
|
|
|
// Duration is duration, which must be used in Prometheus-compatible yaml configs.
|
|
type Duration struct {
|
|
D time.Duration
|
|
}
|
|
|
|
// NewDuration returns Duration for given d.
|
|
func NewDuration(d time.Duration) *Duration {
|
|
return &Duration{
|
|
D: d,
|
|
}
|
|
}
|
|
|
|
// MarshalYAML implements yaml.Marshaler interface.
|
|
func (pd Duration) MarshalYAML() (any, error) {
|
|
return pd.D.String(), nil
|
|
}
|
|
|
|
// UnmarshalYAML implements yaml.Unmarshaler interface.
|
|
func (pd *Duration) UnmarshalYAML(unmarshal func(any) error) error {
|
|
var s string
|
|
if err := unmarshal(&s); err != nil {
|
|
return err
|
|
}
|
|
ms, err := metricsql.DurationValue(s, 0)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
pd.D = time.Duration(ms) * time.Millisecond
|
|
return nil
|
|
}
|
|
|
|
// Duration returns duration for pd.
|
|
func (pd *Duration) Duration() time.Duration {
|
|
if pd == nil {
|
|
return 0
|
|
}
|
|
return pd.D
|
|
}
|
|
|
|
// ParseDuration parses duration string in Prometheus format
|
|
func ParseDuration(s string) (time.Duration, error) {
|
|
ms, err := metricsql.DurationValue(s, 0)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return time.Duration(ms) * time.Millisecond, nil
|
|
}
|