本文介绍在 Cobra 命令行框架中，通过提取核心逻辑为独立函数、统一管理标志位，安全高效地复用已有命令（如 zip 和 upload）来构建新命令（如 zipup），避免代码重复和进程重启开销。
本文介绍在 cobra 命令行框架中，通过提取核心逻辑为独立函数、统一管理标志位，安全高效地复用已有命令（如 `zip` 和 `upload`）来构建新命令（如 `zipup`），避免代码重复和进程重启开销。
在 Cobra 中，直接“调用另一个命令”并非推荐做法——例如通过 exec.Command 启动子进程执行 myapp zip，不仅性能低、难以调试，还会丢失上下文（如配置、日志、信号处理等）。真正的最佳实践是：解耦命令的“定义”与“执行逻辑”，将业务逻辑封装为纯函数，再由各 Command.Run 函数按需调用。
✅ 正确做法：函数化 + 标志共享
首先，将所有跨命令共用的标志（如 -directory, -filename, -output）声明为全局变量，并通过 PersistentFlags() 统一挂载到根命令上，确保所有子命令均可访问：
var (
cfgDirectory string
cfgFilename string
cfgOutput string
)
func init() {
rootCmd.PersistentFlags().StringVar(&cfgDirectory, "directory", "", "source directory to compress")
rootCmd.PersistentFlags().StringVar(&cfgFilename, "filename", "", "target file to upload")
rootCmd.PersistentFlags().StringVar(&cfgOutput, "output", "", "output ZIP filename (default: auto-generated)")
}接着，将每个命令的核心逻辑抽象为无副作用（或可控副作用）的 Go 函数：
func Zip() error {
if cfgDirectory == "" {
return fmt.Errorf("missing required flag: --directory")
}
output := cfgOutput
if output == "" {
output = fmt.Sprintf("%s.zip", filepath.Base(cfgDirectory))
}
// 执行压缩逻辑（如 archive/zip）
fmt.Printf("? Zipping %s → %s\n", cfgDirectory, output)
return os.WriteFile(output, []byte("fake-zip-content"), 0644)
}
func Upload() error {
if cfgFilename == "" {
return fmt.Errorf("missing required flag: --filename")
}
// 执行上传逻辑（如 HTTP POST）
fmt.Printf("? Uploading %s...\n", cfgFilename)
return nil
}最后，在各 Command.Run 中直接调用这些函数，并在组合命令中顺序编排：
var zipCmd = &cobra.Command{
Use: "zip",
Short: "Compress a directory into ZIP",
RunE: func(cmd *cobra.Command, args []string) error {
return Zip()
},
}
var uploadCmd = &cobra.Command{
Use: "upload",
Short: "Upload a file to remote storage",
RunE: func(cmd *cobra.Command, args []string) error {
return Upload()
},
}
var zipupCmd = &cobra.Command{
Use: "zipup",
Short: "Zip a directory and upload the result",
RunE: func(cmd *cobra.Command, args []string) error {
// 复用 zip 逻辑，自动设置上传文件名
if cfgOutput == "" {
cfgOutput = fmt.Sprintf("%s.zip", filepath.Base(cfgDirectory))
}
cfgFilename = cfgOutput // 关键：将 zip 输出作为 upload 输入
if err := Zip(); err != nil {
return err
}
return Upload()
},
}⚠️ 注意事项：
- 使用
RunE（返回error）替代Run，便于错误传播与统一处理；- 避免在
Zip()中硬编码cfgFilename赋值，而应在zipupCmd.RunE中显式桥接参数，保持函数职责单一；- 若需更灵活的参数传递（如动态覆盖标志值），可改用结构体传参（如
Zip(Options{Dir: ..., Output: ...})），提升可测试性；- 组合命令中应添加必要校验（如
cfgDirectory是否为空），防止静默失败。
通过该模式，你不仅实现了 zipup 的功能复用，还让 zip 和 upload 本身更易单元测试、复用至其他场景（如 API handler 或定时任务），真正践行了“关注点分离”与“DRY（Don’t Repeat Yourself）”原则。
