61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
package getopt
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
type intVal int
|
|
|
|
func (i *intVal) String() string {
|
|
return fmt.Sprint(*i)
|
|
}
|
|
|
|
func (i *intVal) Set(val string) error {
|
|
_, err := fmt.Sscanf(val, "%d", i)
|
|
return err
|
|
}
|
|
|
|
// IntVar defines a int flag with specified name, default value,
|
|
// and usage string. The argument p points to a int variable in
|
|
// which to store the value of the flag.
|
|
func (set *FlagSet) IntVar(p *int, name string, value int, usage string) {
|
|
*p = value
|
|
set.Var((*intVal)(p), name, usage)
|
|
}
|
|
|
|
// Int defines an int flag with specified name, default value, and usage
|
|
// string. The return value is the address of an int variable that
|
|
// stores the value of the flag.
|
|
func (set *FlagSet) Int(name string, value int, usage string) *int {
|
|
p := new(int)
|
|
set.IntVar(p, name, value, usage)
|
|
return p
|
|
}
|
|
|
|
type int64Val int64
|
|
|
|
func (i *int64Val) String() string {
|
|
return fmt.Sprint(int64(*i))
|
|
}
|
|
|
|
func (i *int64Val) Set(val string) error {
|
|
_, err := fmt.Sscanf(val, "%d", i)
|
|
return err
|
|
}
|
|
|
|
// Int64Var defines an int64 flag with specified name, default value,
|
|
// and usage string. The argument p points to an int64 variable in which
|
|
// to store the value of the flag.
|
|
func (set *FlagSet) Int64Var(p *int64, name string, value int64, usage string) {
|
|
*p = value
|
|
set.Var((*int64Val)(p), name, usage)
|
|
}
|
|
|
|
// Int64 defines an int64 flag with specified name, default value, and
|
|
// usage string. The return value is the address of an int64 variable
|
|
// that stores the value of the flag.
|
|
func (set *FlagSet) Int64(name string, value int64, usage string) *int64 {
|
|
var i int64
|
|
set.Int64Var(&i, name, value, usage)
|
|
return &i
|
|
}
|