vendor/github.com/spf13/pflag/uint.go

   1 package pflag
   2
   3 import "strconv"
   4
   5 // -- uint Value
   6 type uintValue uint
   7
   8 func newUintValue(val uint, p *uint) *uintValue {
   9         *p = val
  10         return (*uintValue)(p)
  11 }
  12
  13 func (i *uintValue) Set(s string) error {
  14         v, err := strconv.ParseUint(s, 0, 64)
  15         *i = uintValue(v)
  16         return err
  17 }
  18
  19 func (i *uintValue) Type() string {
  20         return "uint"
  21 }
  22
  23 func (i *uintValue) String() string { return strconv.FormatUint(uint64(*i), 10) }
  24
  25 func uintConv(sval string) (interface{}, error) {
  26         v, err := strconv.ParseUint(sval, 0, 0)
  27         if err != nil {
  28                 return 0, err
  29         }
  30         return uint(v), nil
  31 }
  32
  33 // GetUint return the uint value of a flag with the given name
  34 func (f *FlagSet) GetUint(name string) (uint, error) {
  35         val, err := f.getFlagType(name, "uint", uintConv)
  36         if err != nil {
  37                 return 0, err
  38         }
  39         return val.(uint), nil
  40 }
  41
  42 // UintVar defines a uint flag with specified name, default value, and usage string.
  43 // The argument p points to a uint variable in which to store the value of the flag.
  44 func (f *FlagSet) UintVar(p *uint, name string, value uint, usage string) {
  45         f.VarP(newUintValue(value, p), name, "", usage)
  46 }
  47
  48 // UintVarP is like UintVar, but accepts a shorthand letter that can be used after a single dash.
  49 func (f *FlagSet) UintVarP(p *uint, name, shorthand string, value uint, usage string) {
  50         f.VarP(newUintValue(value, p), name, shorthand, usage)
  51 }
  52
  53 // UintVar defines a uint flag with specified name, default value, and usage string.
  54 // The argument p points to a uint  variable in which to store the value of the flag.
  55 func UintVar(p *uint, name string, value uint, usage string) {
  56         CommandLine.VarP(newUintValue(value, p), name, "", usage)
  57 }
  58
  59 // UintVarP is like UintVar, but accepts a shorthand letter that can be used after a single dash.
  60 func UintVarP(p *uint, name, shorthand string, value uint, usage string) {
  61         CommandLine.VarP(newUintValue(value, p), name, shorthand, usage)
  62 }
  63
  64 // Uint defines a uint flag with specified name, default value, and usage string.
  65 // The return value is the address of a uint  variable that stores the value of the flag.
  66 func (f *FlagSet) Uint(name string, value uint, usage string) *uint {
  67         p := new(uint)
  68         f.UintVarP(p, name, "", value, usage)
  69         return p
  70 }
  71
  72 // UintP is like Uint, but accepts a shorthand letter that can be used after a single dash.
  73 func (f *FlagSet) UintP(name, shorthand string, value uint, usage string) *uint {
  74         p := new(uint)
  75         f.UintVarP(p, name, shorthand, value, usage)
  76         return p
  77 }
  78
  79 // Uint defines a uint flag with specified name, default value, and usage string.
  80 // The return value is the address of a uint  variable that stores the value of the flag.
  81 func Uint(name string, value uint, usage string) *uint {
  82         return CommandLine.UintP(name, "", value, usage)
  83 }
  84
  85 // UintP is like Uint, but accepts a shorthand letter that can be used after a single dash.
  86 func UintP(name, shorthand string, value uint, usage string) *uint {
  87         return CommandLine.UintP(name, shorthand, value, usage)
  88 }