如何在眼镜蛇 cli 中将命令状态传递给 Postrun

Question

Cobra CLI 支持在执行命令后调用 PostRun。

https://github.com/spf13/cobra#prerun-and-postrun-hooks

如何将命令状态传递给 PostRun 调用？我需要在执行命令后将命令状态发布到服务器

Answer 1

更简洁的方法是利用 cobra 命令中提供的注释package mainimport (    "fmt"    "github.com/spf13/cobra")func main() {    var rootCmd = &cobra.Command{        Use:   "root [sub]",        Short: "My root command",        Run: func(cmd *cobra.Command, args []string) {            // Do your processing here            // Set the command annotations            cmd.Annotations = make(map[string]string)            cmd.Annotations["status"] = "status_goes_here"            cmd.Annotations["error"] = "error_goes_here"        },        PostRun: func(cmd *cobra.Command, args []string) {            // Retrieve the annotations            fmt.Println(cmd.Annotations["status"])            fmt.Println(cmd.Annotations["error"])        },    }    rootCmd.SetArgs([]string{"sub", "arg1", "arg2"})    rootCmd.Execute()}真的很喜欢@Bracken在这里采取的方法，尽管有一些调整可以使其工作package mainimport (    "fmt"    "errors"    "github.com/spf13/cobra")type wrapper struct {    err error}// RunE fails to proceed further in case of error resulting in not executing PostRun actionsfunc (w *wrapper) Run(f func(cmd *cobra.Command, args []string) error) func(cmd *cobra.Command, args []string) {    return func(cmd *cobra.Command, args []string) {        err := f(cmd, args)        w.err = err    }}func (w *wrapper) PostRun(f func(cmd *cobra.Command, args []string, cmdErr error)) func(cmd *cobra.Command, args []string) {    return func(cmd *cobra.Command, args []string) {        f(cmd, args, w.err)    }}func main() {    cmdWrap := wrapper{}    var rootCmd = &cobra.Command{        Use:   "root [sub]",        Short: "My root command",        Run: cmdWrap.Run(func(cmd *cobra.Command, args []string) error {            return errors.New("i'm not in the book, you know")        }),        PostRun: cmdWrap.PostRun(func(cmd *cobra.Command, args []string, cmdErr error) {            fmt.Printf("error was %v\n", cmdErr)        }),    }    rootCmd.SetArgs([]string{"sub", "arg1", "arg2"})    rootCmd.Execute()}----- 旧答案-----如果我理解正确，有一些状态需要从cmd.Execute传递到PostRun。您可以使用方法代替并设置需要在上下文中设置的任何键值。ExecuteContext(ctx context.Context)Execute()ctx := context.WithValue(context.Background(), "status", "statusValue")rootCmd.ExecuteContext(ctx)可以使用在 PostRun 中检索相同的值cmd.Context()PostRun: func(cmd *cobra.Command, args []string) {   ctx := cmd.Context()   status := ctx.Value("status")}

Answer 2

我会使用高阶函数来包装你的（或）函数和你的函数来捕获错误或恐慌，然后将它们传递到你的：RunRunEPostRunPostRunpackage mainimport (    "fmt"    "github.com/spf13/cobra")type wrapper struct {    err error}func (w *wrapper) RunE(f func(cmd *cobra.Command, args []string) error) func(cmd *cobra.Command, args []string) error {    return func(cmd *cobra.Command, args []string) (err error) {        defer func() {            if r := recover(); r != nil {                err = fmt.Errorf("panic: %v", r)                w.err = err            }        }()        err = f(cmd, args)        w.err = err        return    }}func (w *wrapper) PostRun(f func(cmd *cobra.Command, args []string, cmdErr error)) func(cmd *cobra.Command, args []string) {    return func(cmd *cobra.Command, args []string) {        f(cmd, args, w.err)    }}func main() {    cmdWrap := wrapper{}    var cmdFail = &cobra.Command{        Use:   "fail",        Short: "Doesn't work",        RunE: cmdWrap.RunE(func(cmd *cobra.Command, args []string) error {            panic("i'm not in the book, you know")        }),        PostRun: cmdWrap.PostRun(func(cmd *cobra.Command, args []string, cmdErr error) {            fmt.Printf("error was %v\n", cmdErr)        }),    }    var rootCmd = &cobra.Command{}    rootCmd.AddCommand(cmdFail)    rootCmd.Execute()}