From 30e9b44054269109e5cfadb6fc9a0db9f36ed15f Mon Sep 17 00:00:00 2001 From: wangchuxiao Date: Tue, 7 Mar 2023 20:41:09 +0800 Subject: [PATCH] test cobra --- pkg/common/cmd/root.go | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/pkg/common/cmd/root.go b/pkg/common/cmd/root.go index dd0ffec46..37eee7cdf 100644 --- a/pkg/common/cmd/root.go +++ b/pkg/common/cmd/root.go @@ -15,21 +15,21 @@ type RootCmd struct { prometheusPortFlag bool } -func NewRootCmd() *RootCmd { +func NewRootCmd() (rootCmd *RootCmd) { + rootCmd = &RootCmd{} c := cobra.Command{ Use: "start", Short: "Start the server", Long: `Start the server`, - } - rootCmd := &RootCmd{} - c.PersistentPreRunE = func(cmd *cobra.Command, args []string) error { - if rootCmd.portFlag { - rootCmd.port = rootCmd.getPortFlag(cmd) - } - if rootCmd.prometheusPortFlag { - rootCmd.prometheusPort = rootCmd.getPrometheusPortFlag(cmd) - } - return rootCmd.getConfFromCmdAndInit(cmd) + PersistentPreRunE: func(cmd *cobra.Command, args []string) error { + if rootCmd.portFlag { + rootCmd.port = rootCmd.getPortFlag(cmd) + } + if rootCmd.prometheusPortFlag { + rootCmd.prometheusPort = rootCmd.getPrometheusPortFlag(cmd) + } + return rootCmd.getConfFromCmdAndInit(cmd) + }, } rootCmd.Command = c rootCmd.init()