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

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