package cmd import ( "fmt" verifyCode "archgrid.xyz/ag/toolsbox/random/verify_code" "github.com/spf13/cobra" ) var ( verifyCodeLength int ) var verifyCodeCmd = &cobra.Command{ Use: "verify_code", Short: "生成指定长度的随机验证码", Run: verifyCodeExecute, } func verifyCodeExecute(cmd *cobra.Command, args []string) { code := verifyCode.RandStr(verifyCodeLength) fmt.Printf("生成的验证码为:%s\n", code) } func init() { verifyCodeCmd.PersistentFlags().IntVarP(&verifyCodeLength, "length", "l", 10, "要生成的验证码长度,默认生成长度为10个字符的验证码") rootCmd.AddCommand(verifyCodeCmd) }