23 lines
377 B
Go
23 lines
377 B
Go
package cmd
|
|
|
|
import "github.com/spf13/cobra"
|
|
|
|
func CommandGen(root *cobra.Command) {
|
|
cmd := &cobra.Command{
|
|
Use: "gen",
|
|
Short: "Generate code",
|
|
}
|
|
cmd.PersistentFlags().StringP("config", "c", "config.toml", "database config file")
|
|
|
|
cmds := []func(*cobra.Command){
|
|
CommandGenModel,
|
|
CommandGenEnum,
|
|
}
|
|
|
|
for _, c := range cmds {
|
|
c(cmd)
|
|
}
|
|
|
|
root.AddCommand(cmd)
|
|
}
|