github.com/artisanhe/tools@v1.0.1-0.20210607022958-19a8fef2eb04/cmd/cmd_gen_method.go (about) 1 package cmd 2 3 import ( 4 "github.com/spf13/cobra" 5 6 "github.com/artisanhe/tools/gen_method" 7 ) 8 9 var cmdGenMethodFlagNoTableName bool 10 11 var cmdGenMethod = &cobra.Command{ 12 Use: "method", 13 Short: "generate db model method", 14 Run: func(cmd *cobra.Command, args []string) { 15 eg := gen_method.NewScanner(packageName) 16 if args[0] != "" { 17 eg.Output(args[0], cmdGenMethodFlagNoTableName) 18 } 19 }, 20 } 21 22 func init() { 23 cmdGenMethod.Flags(). 24 BoolVarP(&cmdGenMethodFlagNoTableName, "no-table-name", "", false, "skip tableName") 25 26 cmdGen.AddCommand(cmdGenMethod) 27 }