From 11d3c7ac689361e8ba1c69b38797732155abc12f Mon Sep 17 00:00:00 2001 From: Gordon <46924906+FGadvancer@users.noreply.github.com> Date: Fri, 12 Apr 2024 17:05:10 +0800 Subject: [PATCH] refactor: cmd update. --- pkg/common/cmd/root.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pkg/common/cmd/root.go b/pkg/common/cmd/root.go index 1ffa12a33..2a8b8515a 100644 --- a/pkg/common/cmd/root.go +++ b/pkg/common/cmd/root.go @@ -75,6 +75,9 @@ func NewRootCmd(processName string, opts ...func(*CmdOpts)) *RootCmd { SilenceUsage: true, SilenceErrors: true, } + cmd.Flags().StringP(FlagConf, "c", "", "path of config directory") + cmd.Flags().IntP(FlagTransferIndex, "i", 0, "process startup sequence number") + rootCmd.Command = cmd return rootCmd } @@ -147,12 +150,10 @@ func defaultCmdOpts() *CmdOpts { } func (r *RootCmd) getFlag(cmd *cobra.Command) (string, int, error) { - r.Command.Flags().StringP(FlagConf, "c", "", "path of config directory") configDirectory, err := cmd.Flags().GetString(FlagConf) if err != nil { return "", 0, errs.Wrap(err) } - r.Command.Flags().IntP(FlagTransferIndex, "i", 0, "process startup sequence number") index, err := cmd.Flags().GetInt(FlagTransferIndex) if err != nil { return "", 0, errs.Wrap(err)