diff --git a/cmd/gen_enum.go b/cmd/gen_enum.go index 6898598..83ec1d5 100644 --- a/cmd/gen_enum.go +++ b/cmd/gen_enum.go @@ -8,18 +8,19 @@ import ( "path/filepath" "strings" - "go.ipao.vip/atomctl/pkg/utils" - "go.ipao.vip/atomctl/pkg/utils/generator" _ "github.com/lib/pq" "github.com/spf13/cobra" + "go.ipao.vip/atomctl/pkg/utils" + "go.ipao.vip/atomctl/pkg/utils/generator" ) func CommandGenEnum(root *cobra.Command) { cmd := &cobra.Command{ - Use: "enum", - Aliases: []string{"e"}, - Short: "Generate enums", - RunE: commandGenEnumE, + Use: "enum", + Aliases: []string{"e"}, + Short: "Generate enums", + RunE: commandGenEnumE, + PostRunE: commandGenProviderE, } cmd.Flags().BoolP("flag", "f", true, "Flag enum values") diff --git a/cmd/gen_route.go b/cmd/gen_route.go index 88d3e55..6b5e0a3 100644 --- a/cmd/gen_route.go +++ b/cmd/gen_route.go @@ -15,10 +15,10 @@ import ( func CommandGenRoute(root *cobra.Command) { cmd := &cobra.Command{ - Use: "route", - Short: "generate routes", - RunE: commandGenRouteE, - // PostRunE: commandGenProviderE, + Use: "route", + Short: "generate routes", + RunE: commandGenRouteE, + PostRunE: commandGenProviderE, } root.AddCommand(cmd)