From e4ad0152d3a2973e4be906669ba619fd4dfd7e7e Mon Sep 17 00:00:00 2001 From: Rogee Date: Thu, 13 Feb 2025 18:45:03 +0800 Subject: [PATCH] feat: update gen model --- cmd/gen_enum.go | 13 +++++++------ cmd/gen_route.go | 8 ++++---- 2 files changed, 11 insertions(+), 10 deletions(-) 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)