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  }