diff --git a/flagx/flagx.go b/flagx/flagx.go index e8af02b0..e4a865fa 100644 --- a/flagx/flagx.go +++ b/flagx/flagx.go @@ -38,3 +38,11 @@ func MustGetStringSlice(cmd *cobra.Command, name string) []string { } return ss } + +func MustGetInt(cmd *cobra.Command, name string) int { + ss, err := cmd.Flags().GetInt(name) + if err != nil { + cmdx.Fatalf(err.Error()) + } + return ss +}