diff options
| author | Eric Paris <[email protected]> | 2015-05-30 21:07:46 -0400 |
|---|---|---|
| committer | Eric Paris <[email protected]> | 2015-06-01 18:45:34 -0400 |
| commit | 1e0a23de9163cb0706137856f1d060f88e3f277c (patch) | |
| tree | de398763dd538847a8c32d4073ce07d5fe8bb210 /duration.go | |
| parent | 5644820622454e71517561946e3d94b9f9db6842 (diff) | |
Add new FlagSet.Get{Int,String,...} accessor functions
If I declared a bool flag named "hello" I can now call
b, err := f.GetBool("hello")
And b will hold the value of the flag
We can see this is already done in
https://github.com/codegangsta/cli/blob/bcec9b08c7e5564f7512ad7e7b03778fe1923116/context.go
If people use the codegangsta/cli
Other projects have done it themselves using pflags (what inspired this
patch)
https://github.com/GoogleCloudPlatform/kubernetes/blob/cd817aebd848facda29e0befbbd6e31bf22402e6/pkg/kubectl/cmd/util/helpers.go#L176
Lets just do it ourselves...
Diffstat (limited to 'duration.go')
| -rw-r--r-- | duration.go | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/duration.go b/duration.go index 66ed7ac..382ffd3 100644 --- a/duration.go +++ b/duration.go @@ -1,6 +1,8 @@ package pflag -import "time" +import ( + "time" +) // -- time.Duration Value type durationValue time.Duration @@ -22,6 +24,19 @@ func (d *durationValue) Type() string { func (d *durationValue) String() string { return (*time.Duration)(d).String() } +func durationConv(sval string) (interface{}, error) { + return time.ParseDuration(sval) +} + +// GetDuration return the duration value of a flag with the given name +func (f *FlagSet) GetDuration(name string) (time.Duration, error) { + val, err := f.getFlagType(name, "duration", durationConv) + if err != nil { + return 0, err + } + return val.(time.Duration), nil +} + // DurationVar defines a time.Duration flag with specified name, default value, and usage string. // The argument p points to a time.Duration variable in which to store the value of the flag. func (f *FlagSet) DurationVar(p *time.Duration, name string, value time.Duration, usage string) { |
