1
package pflag
2

3
import (
4
	"time"
5
)
6

7
// -- time.Duration Value
8
type durationValue time.Duration
9

10
func newDurationValue(val time.Duration, p *time.Duration) *durationValue {
11
	*p = val
12
	return (*durationValue)(p)
13
}
14

15
func (d *durationValue) Set(s string) error {
16
	v, err := time.ParseDuration(s)
17
	*d = durationValue(v)
18
	return err
19
}
20

21
func (d *durationValue) Type() string {
22
	return "duration"
23
}
24

25
func (d *durationValue) String() string { return (*time.Duration)(d).String() }
26

27
func durationConv(sval string) (interface{}, error) {
28
	return time.ParseDuration(sval)
29
}
30

31
// GetDuration return the duration value of a flag with the given name
32
func (f *FlagSet) GetDuration(name string) (time.Duration, error) {
33
	val, err := f.getFlagType(name, "duration", durationConv)
34
	if err != nil {
35
		return 0, err
36
	}
37
	return val.(time.Duration), nil
38
}
39

40
// DurationVar defines a time.Duration flag with specified name, default value, and usage string.
41
// The argument p points to a time.Duration variable in which to store the value of the flag.
42
func (f *FlagSet) DurationVar(p *time.Duration, name string, value time.Duration, usage string) {
43
	f.VarP(newDurationValue(value, p), name, "", usage)
44
}
45

46
// DurationVarP is like DurationVar, but accepts a shorthand letter that can be used after a single dash.
47
func (f *FlagSet) DurationVarP(p *time.Duration, name, shorthand string, value time.Duration, usage string) {
48
	f.VarP(newDurationValue(value, p), name, shorthand, usage)
49
}
50

51
// DurationVar defines a time.Duration flag with specified name, default value, and usage string.
52
// The argument p points to a time.Duration variable in which to store the value of the flag.
53
func DurationVar(p *time.Duration, name string, value time.Duration, usage string) {
54
	CommandLine.VarP(newDurationValue(value, p), name, "", usage)
55
}
56

57
// DurationVarP is like DurationVar, but accepts a shorthand letter that can be used after a single dash.
58
func DurationVarP(p *time.Duration, name, shorthand string, value time.Duration, usage string) {
59
	CommandLine.VarP(newDurationValue(value, p), name, shorthand, usage)
60
}
61

62
// Duration defines a time.Duration flag with specified name, default value, and usage string.
63
// The return value is the address of a time.Duration variable that stores the value of the flag.
64
func (f *FlagSet) Duration(name string, value time.Duration, usage string) *time.Duration {
65
	p := new(time.Duration)
66
	f.DurationVarP(p, name, "", value, usage)
67
	return p
68
}
69

70
// DurationP is like Duration, but accepts a shorthand letter that can be used after a single dash.
71
func (f *FlagSet) DurationP(name, shorthand string, value time.Duration, usage string) *time.Duration {
72
	p := new(time.Duration)
73
	f.DurationVarP(p, name, shorthand, value, usage)
74
	return p
75
}
76

77
// Duration defines a time.Duration flag with specified name, default value, and usage string.
78
// The return value is the address of a time.Duration variable that stores the value of the flag.
79
func Duration(name string, value time.Duration, usage string) *time.Duration {
80
	return CommandLine.DurationP(name, "", value, usage)
81
}
82

83
// DurationP is like Duration, but accepts a shorthand letter that can be used after a single dash.
84
func DurationP(name, shorthand string, value time.Duration, usage string) *time.Duration {
85
	return CommandLine.DurationP(name, shorthand, value, usage)
86
}
87

88