-
Notifications
You must be signed in to change notification settings - Fork 53
/
duration.go
47 lines (39 loc) · 1.05 KB
/
duration.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package flagutil
import (
"strings"
"time"
"github.com/prometheus/common/model"
"github.com/spf13/pflag"
"google.golang.org/protobuf/types/known/durationpb"
)
type durationpbValue durationpb.Duration
func DurationpbValue(val time.Duration, p **durationpb.Duration) pflag.Value {
*p = durationpb.New(val)
return (*durationpbValue)(*p)
}
func (d *durationpbValue) Set(s string) error {
v, err := ParseDurationWithExtendedUnits(s)
*d = *(*durationpbValue)(durationpb.New(v))
return err
}
func (d *durationpbValue) Type() string {
return "duration"
}
func (d *durationpbValue) String() string {
return (*durationpb.Duration)(d).AsDuration().String()
}
func ParseDurationWithExtendedUnits(s string) (time.Duration, error) {
// try parsing with standard units
v, err := time.ParseDuration(s)
if err != nil {
if strings.Contains(err.Error(), "unknown unit") {
// prometheus duration accepts "w" and "y" units, for some reason
p, err := model.ParseDuration(s)
if err != nil {
return 0, err
}
v = time.Duration(p)
}
}
return v, nil
}