mirror of
https://github.com/kubernetes/sample-controller.git
synced 2025-01-21 01:12:51 +08:00
4a1ec65955
Automatic merge from submit-queue. If you want to cherry-pick this change to another branch, please follow the instructions <a href="https://github.com/kubernetes/community/blob/master/contributors/devel/cherry-picks.md">here</a>. update vendor spf13/cobra to enforce required flags **What this PR does / why we need it**: spf13/cobra#502 has enforced checking flags that marked as required, an error will be raised if unset. **Which issue this PR fixes** *(optional, in `fixes #<issue number>(, fixes #<issue_number>, ...)` format, will close that issue when PR gets merged)*:fixes #54855 xref #48400 fixes kubernetes/kubectl#121 **Special notes for your reviewer**: /assign @liggitt @eparis **Release note**: ```release-note kubectl now enforces required flags at a more fundamental level ``` Kubernetes-commit: 048757b8a51333f59d3112d2b228d2f0102a4afc
97 lines
2.9 KiB
Go
97 lines
2.9 KiB
Go
package pflag
|
|
|
|
import "strconv"
|
|
|
|
// -- count Value
|
|
type countValue int
|
|
|
|
func newCountValue(val int, p *int) *countValue {
|
|
*p = val
|
|
return (*countValue)(p)
|
|
}
|
|
|
|
func (i *countValue) Set(s string) error {
|
|
// "+1" means that no specific value was passed, so increment
|
|
if s == "+1" {
|
|
*i = countValue(*i + 1)
|
|
return nil
|
|
}
|
|
v, err := strconv.ParseInt(s, 0, 0)
|
|
*i = countValue(v)
|
|
return err
|
|
}
|
|
|
|
func (i *countValue) Type() string {
|
|
return "count"
|
|
}
|
|
|
|
func (i *countValue) String() string { return strconv.Itoa(int(*i)) }
|
|
|
|
func countConv(sval string) (interface{}, error) {
|
|
i, err := strconv.Atoi(sval)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return i, nil
|
|
}
|
|
|
|
// GetCount return the int value of a flag with the given name
|
|
func (f *FlagSet) GetCount(name string) (int, error) {
|
|
val, err := f.getFlagType(name, "count", countConv)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return val.(int), nil
|
|
}
|
|
|
|
// CountVar defines a count flag with specified name, default value, and usage string.
|
|
// The argument p points to an int variable in which to store the value of the flag.
|
|
// A count flag will add 1 to its value evey time it is found on the command line
|
|
func (f *FlagSet) CountVar(p *int, name string, usage string) {
|
|
f.CountVarP(p, name, "", usage)
|
|
}
|
|
|
|
// CountVarP is like CountVar only take a shorthand for the flag name.
|
|
func (f *FlagSet) CountVarP(p *int, name, shorthand string, usage string) {
|
|
flag := f.VarPF(newCountValue(0, p), name, shorthand, usage)
|
|
flag.NoOptDefVal = "+1"
|
|
}
|
|
|
|
// CountVar like CountVar only the flag is placed on the CommandLine instead of a given flag set
|
|
func CountVar(p *int, name string, usage string) {
|
|
CommandLine.CountVar(p, name, usage)
|
|
}
|
|
|
|
// CountVarP is like CountVar only take a shorthand for the flag name.
|
|
func CountVarP(p *int, name, shorthand string, usage string) {
|
|
CommandLine.CountVarP(p, name, shorthand, usage)
|
|
}
|
|
|
|
// Count defines a count 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.
|
|
// A count flag will add 1 to its value evey time it is found on the command line
|
|
func (f *FlagSet) Count(name string, usage string) *int {
|
|
p := new(int)
|
|
f.CountVarP(p, name, "", usage)
|
|
return p
|
|
}
|
|
|
|
// CountP is like Count only takes a shorthand for the flag name.
|
|
func (f *FlagSet) CountP(name, shorthand string, usage string) *int {
|
|
p := new(int)
|
|
f.CountVarP(p, name, shorthand, usage)
|
|
return p
|
|
}
|
|
|
|
// Count defines a count 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.
|
|
// A count flag will add 1 to its value evey time it is found on the command line
|
|
func Count(name string, usage string) *int {
|
|
return CommandLine.CountP(name, "", usage)
|
|
}
|
|
|
|
// CountP is like Count only takes a shorthand for the flag name.
|
|
func CountP(name, shorthand string, usage string) *int {
|
|
return CommandLine.CountP(name, shorthand, usage)
|
|
}
|