Cobra CLI中如何将命令执行状态传递给PostRun钩子

Cobra没有提供内置的状态传递API，你可以通过以下两种常用方案实现状态传递，满足上报需求：

方法1：自定义状态变量（最简单） #

直接在同一作用域下定义状态变量，在Run阶段赋值、PostRun阶段读取即可，示例代码如下：

package main

import (
	"fmt"
	"github.com/spf13/cobra"
)

// 自定义执行状态结构体，按需扩展字段
type ExecStatus struct {
	IsSuccess bool   // 执行是否成功
	BizCode   int    // 业务错误码
	ResultMsg string // 执行结果描述
}

func main() {
	var cmdStatus ExecStatus

	rootCmd := &cobra.Command{
		Use:   "status-demo",
		Short: "演示PostRun状态传递",
		Run: func(cmd *cobra.Command, args []string) {
			// 此处替换为你的实际业务执行逻辑
			cmdStatus.IsSuccess = true
			cmdStatus.BizCode = 0
			cmdStatus.ResultMsg = "业务逻辑执行完成"
			fmt.Println("主命令执行完毕")
		},
		PostRun: func(cmd *cobra.Command, args []string) {
			// 直接读取状态变量，执行上报逻辑
			fmt.Printf("待上报状态：成功=%v, 错误码=%d, 结果=%s\n", 
				cmdStatus.IsSuccess, cmdStatus.BizCode, cmdStatus.ResultMsg)
			// 此处添加你的上报服务器逻辑即可
		},
	}

	if err := rootCmd.Execute(); err != nil {
		fmt.Printf("命令执行出错：%v", err)
	}
}

方法2：Context传值（适合长链路场景） #

如果你的命令逻辑链路较长、跨多个函数调用，可以通过Context携带状态：

package main

import (
	"context"
	"fmt"
	"github.com/spf13/cobra"
)

// 自定义Context key，避免键名冲突
type statusCtxKey struct{}

func main() {
	rootCtx := context.Background()

	rootCmd := &cobra.Command{
		Use:   "ctx-demo",
		Run: func(cmd *cobra.Command, args []string) {
			// 业务逻辑执行完成后，把状态写入Context
			status := map[string]interface{}{
				"is_success": true,
				"biz_code":   0,
				"msg":        "执行成功",
			}
			newCtx := context.WithValue(cmd.Context(), statusCtxKey{}, status)
			cmd.SetContext(newCtx)
		},
		PostRun: func(cmd *cobra.Command, args []string) {
			// 从Context中取出状态执行上报
			status := cmd.Context().Value(statusCtxKey{}).(map[string]interface{})
			fmt.Printf("从Context获取到的执行状态：%v\n", status)
			// 添加上报逻辑即可
		},
	}

	rootCmd.SetContext(rootCtx)
	if err := rootCmd.Execute(); err != nil {
		fmt.Printf("命令执行出错：%v", err)
	}
}

注意事项 #

• 如果你用的是RunE返回执行错误，可以直接在RunE中把错误信息写入状态变量，不需要额外捕获全局错误
• 上报逻辑如果有网络IO耗时，建议添加超时控制，避免阻塞命令正常退出

内容的提问来源于stack exchange，提问作者gbvikas