Cobra 标志配置优先级解析

示例说明

配置来源一多，谁覆盖谁就成了玄学——运维改了配置文件不生效，排查半天发现被环境变量压住。把优先级写成显式的解析函数：默认值打底，配置文件覆盖，环境变量再覆盖，命令行标志最终拍板。每层只在有值时覆盖，空值不破坏下层配置，日志输出最终结果让生效组合一目了然。

技术要点

• 优先级链条显式编码不靠隐式合并
• 每层仅在有值时覆盖下层
• 解析结果统一打印便于核对

代码解读

resolveConfig 的 workers 走完整四层：默认四、文件八、环境变量与标志依次判断覆盖；环境与超时走类似但不完全相同的链。atoiSafe 手写安全解析并夹上限，防止环境变量注入异常值。Cobra 的三个标志声明短长名与默认值，未传时保持零值以便识别。

使用建议

配置文件解析可接 viper 自动化此模式。环境变量命名与标志名保持可推导的映射。最终配置落日志是排障刚需。

代码示例

package main

import (
	"fmt"
	"os"
	"time"

	"github.com/spf13/cobra"
)

type SyncConfig struct {
	Workers     int
	Timeout     time.Duration
	Environment string
}

func resolveConfig(flagWorkers int, flagTimeout time.Duration,
	flagEnv string, fileEnv string, fileWorkers int,
) SyncConfig {
	// 优先级：命令行标志 > 环境变量 > 配置文件 > 内置默认值
	workers := 4 // 默认值
	if fileWorkers > 0 {
		workers = fileWorkers
	}
	if env := os.Getenv("SYNC_WORKERS"); env != "" {
		if parsed, ok := atoiSafe(env); ok {
			workers = parsed
		}
	}
	if flagWorkers > 0 {
		workers = flagWorkers
	}

	environment := fileEnv
	if env := os.Getenv("SYNC_ENV"); env != "" {
		environment = env
	}
	if flagEnv != "" {
		environment = flagEnv
	}

	return SyncConfig{
		Workers:     workers,
		Timeout:     flagTimeout,
		Environment: environment,
	}
}

func atoiSafe(s string) (int, bool) {
	n := 0
	if s == "" {
		return 0, false
	}
	for _, ch := range s {
		if ch < '0' || ch > '9' {
			return 0, false
		}
		n = n*10 + int(ch-'0')
		if n > 10_000 {
			return 0, false
		}
	}
	return n, true
}

func main() {
	var (
		workers int
		timeout time.Duration
		env     string
	)

	root := &cobra.Command{
		Use:   "sync",
		Short: "Sync catalog data across regions",
		Run: func(cmd *cobra.Command, args []string) {
			cfg := resolveConfig(workers, timeout, env, "staging", 8)
			fmt.Printf("resolved: workers=%d timeout=%s env=%s\n",
				cfg.Workers, cfg.Timeout, cfg.Environment)
		},
	}

	root.Flags().IntVarP(&workers, "workers", "w", 0, "concurrent workers")
	root.Flags().DurationVarP(&timeout, "timeout", "t", 30*time.Second, "per-batch timeout")
	root.Flags().StringVarP(&env, "env", "e", "", "target environment")

	root.Execute()
}