golang-cli-cobra-viper by bobmatnyc/claude-mpm-skills
npx skills add https://github.com/bobmatnyc/claude-mpm-skills --skill golang-cli-cobra-viperCobra 和 Viper 是 Go 语言中用于构建生产级 CLI 的行业标准库。Cobra 提供命令结构和参数解析,而 Viper 则通过清晰的优先级规则管理来自多个源的配置。
主要特性:
app verb noun --flag)使用者 :Kubernetes (kubectl)、Docker CLI、GitHub CLI (gh)、Hugo、Helm 以及 100 多个主要项目
在以下情况下激活此技能:
Cobra 遵循由 git 和 kubectl 推广的 APPNAME VERB NOUN --FLAG 模式。
// cmd/root.go
package cmd
import (
"fmt"
"os"
"github.com/spf13/cobra"
"github.com/spf13/viper"
)
var cfgFile string
var rootCmd = &cobra.Command{
Use: "myapp",
Short: "一个面向开发者的强大 CLI 工具",
Long: `MyApp 是一个 CLI 工具,展示了构建生产级命令行应用程序的最佳实践。
完整文档请访问 https://myapp.example.com`,
}
func Execute() {
if err := rootCmd.Execute(); err != nil {
fmt.Fprintln(os.Stderr, err)
os.Exit(1)
}
}
func init() {
cobra.OnInitialize(initConfig)
// 持久标志(对所有子命令可用)
rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "配置文件(默认为 $HOME/.myapp.yaml)")
rootCmd.PersistentFlags().Bool("verbose", false, "详细输出")
// 将持久标志绑定到 viper
viper.BindPFlag("config", rootCmd.PersistentFlags().Lookup("config"))
viper.BindPFlag("verbose", rootCmd.PersistentFlags().Lookup("verbose"))
}
func initConfig() {
if cfgFile != "" {
viper.SetConfigFile(cfgFile)
} else {
home, err := os.UserHomeDir()
if err != nil {
fmt.Fprintln(os.Stderr, err)
os.Exit(1)
}
viper.AddConfigPath(home)
viper.AddConfigPath(".")
viper.SetConfigType("yaml")
viper.SetConfigName(".myapp")
}
viper.SetEnvPrefix("MYAPP")
viper.AutomaticEnv()
if err := viper.ReadInConfig(); err == nil {
if viper.GetBool("verbose") {
fmt.Println("使用配置文件:", viper.ConfigFileUsed())
}
}
}
广告位招租
在这里展示您的产品或服务
触达数万 AI 开发者,精准高效
// cmd/deploy.go
package cmd
import (
"fmt"
"github.com/spf13/cobra"
"github.com/spf13/viper"
)
var deployCmd = &cobra.Command{
Use: "deploy [environment]",
Short: "将应用程序部署到指定环境",
Long: `将应用程序部署到指定环境。
支持:dev、staging、production`,
Args: cobra.ExactArgs(1),
ValidArgs: []string{"dev", "staging", "production"},
PreRunE: func(cmd *cobra.Command, args []string) error {
// 验证逻辑在 RunE 之前运行
env := args[0]
if env == "production" && !viper.GetBool("force") {
return fmt.Errorf("生产环境部署需要 --force 标志")
}
return nil
},
RunE: func(cmd *cobra.Command, args []string) error {
env := args[0]
region := viper.GetString("region")
force := viper.GetBool("force")
fmt.Printf("正在部署到 %s 区域 %s (force=%v)\n", env, region, force)
// 实际的部署逻辑
return deploy(env, region, force)
},
PostRunE: func(cmd *cobra.Command, args []string) error {
// 清理或通知
fmt.Println("部署完成")
return nil
},
}
func init() {
rootCmd.AddCommand(deployCmd)
// 本地标志(仅对此命令可用)
deployCmd.Flags().StringP("region", "r", "us-east-1", "AWS 区域")
deployCmd.Flags().BoolP("force", "f", false, "无需确认强制部署")
// 将标志绑定到 viper
viper.BindPFlag("region", deployCmd.Flags().Lookup("region"))
viper.BindPFlag("force", deployCmd.Flags().Lookup("force"))
}
func deploy(env, region string, force bool) error {
// 实现
return nil
}
// 持久标志:对命令及其所有子命令可用
rootCmd.PersistentFlags().String("config", "", "配置文件路径")
rootCmd.PersistentFlags().Bool("verbose", false, "详细输出")
// 本地标志:仅对此特定命令可用
deployCmd.Flags().String("region", "us-east-1", "部署区域")
deployCmd.Flags().Bool("force", false, "强制部署")
// 必需标志
deployCmd.MarkFlagRequired("region")
// 标志依赖关系
deployCmd.MarkFlagsRequiredTogether("username", "password")
deployCmd.MarkFlagsMutuallyExclusive("json", "yaml")
Cobra 为设置和清理提供执行钩子:
var serverCmd = &cobra.Command{
Use: "server",
Short: "启动 API 服务器",
// 执行顺序(均为可选):
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
// 在 PreRunE 之前运行,被子命令继承
return setupLogging()
},
PreRunE: func(cmd *cobra.Command, args []string) error {
// 在 RunE 之前进行验证和设置
return validateConfig()
},
RunE: func(cmd *cobra.Command, args []string) error {
// 主要命令逻辑
return startServer()
},
PostRunE: func(cmd *cobra.Command, args []string) error {
// 在 RunE 之后进行清理
return cleanup()
},
PersistentPostRunE: func(cmd *cobra.Command, args []string) error {
// 在 PostRunE 之后运行,被子命令继承
return flushLogs()
},
}
重要 :使用 RunE、PreRunE、PostRunE(返回错误的版本)而不是 Run、PreRun、PostRun。
Viper 遵循严格的优先级顺序(从高到低):
viper.Set("key", value))viper.BindPFlag 绑定)MYAPP_KEY=value)~/.myapp.yaml, ./config.yaml)viper.SetDefault("key", value))func initConfig() {
// 1. 设置默认值(最低优先级)
viper.SetDefault("port", 8080)
viper.SetDefault("database.host", "localhost")
viper.SetDefault("database.port", 5432)
// 2. 配置文件位置(按顺序检查)
viper.SetConfigName("config")
viper.SetConfigType("yaml")
viper.AddConfigPath("/etc/myapp/")
viper.AddConfigPath("$HOME/.myapp")
viper.AddConfigPath(".")
// 3. 环境变量(前缀 + 自动映射)
viper.SetEnvPrefix("MYAPP")
viper.AutomaticEnv() // MYAPP_PORT, MYAPP_DATABASE_HOST 等
// 4. 读取配置文件(可选)
if err := viper.ReadInConfig(); err != nil {
if _, ok := err.(viper.ConfigFileNotFoundError); ok {
// 未找到配置文件 - 使用默认值和环境变量
} else {
// 找到配置文件但读取时出错
return err
}
}
// 5. 标志将在 init() 函数中绑定(最高优先级)
}
Viper 自动映射带有前缀和点号表示法的环境变量:
viper.SetEnvPrefix("MYAPP") // 环境变量的前缀
viper.AutomaticEnv() // 启用自动映射
// 配置键 → 环境变量
// "port" → MYAPP_PORT
// "database.host" → MYAPP_DATABASE_HOST
// "database.port" → MYAPP_DATABASE_PORT
// "aws.s3.region" → MYAPP_AWS_S3_REGION
手动映射 非标准环境变量名称:
viper.BindEnv("database.host", "DB_HOST") // 自定义环境变量名
viper.BindEnv("database.password", "DB_PASSWORD") // 不同的命名约定
Viper 支持多种格式:YAML、JSON、TOML、HCL、INI、envfile、Java properties。
config.yaml :
port: 8080
log_level: info
database:
host: localhost
port: 5432
user: postgres
ssl_mode: require
aws:
region: us-east-1
s3:
bucket: my-app-bucket
访问配置值 :
port := viper.GetInt("port") // 8080
dbHost := viper.GetString("database.host") // "localhost"
s3Bucket := viper.GetString("aws.s3.bucket") // "my-app-bucket"
// 类型安全访问
if viper.IsSet("database.ssl_mode") {
sslMode := viper.GetString("database.ssl_mode")
}
// 解组到结构体
type Config struct {
Port int `mapstructure:"port"`
LogLevel string `mapstructure:"log_level"`
Database struct {
Host string `mapstructure:"host"`
Port int `mapstructure:"port"`
User string `mapstructure:"user"`
SSLMode string `mapstructure:"ssl_mode"`
} `mapstructure:"database"`
}
var config Config
if err := viper.Unmarshal(&config); err != nil {
return err
}
Cobra + Viper 集成的关键在于将标志绑定到 Viper 键:
// cmd/root.go
func init() {
cobra.OnInitialize(initConfig) // 在命令执行前加载配置
// 定义标志
rootCmd.PersistentFlags().String("config", "", "配置文件")
rootCmd.PersistentFlags().String("log-level", "info", "日志级别")
rootCmd.PersistentFlags().Int("port", 8080, "服务器端口")
// 将标志绑定到 Viper(关键步骤!)
viper.BindPFlag("config", rootCmd.PersistentFlags().Lookup("config"))
viper.BindPFlag("log_level", rootCmd.PersistentFlags().Lookup("log-level"))
viper.BindPFlag("port", rootCmd.PersistentFlags().Lookup("port"))
}
func initConfig() {
// 通过 cobra.OnInitialize 在命令执行前运行
if cfgFile := viper.GetString("config"); cfgFile != "" {
viper.SetConfigFile(cfgFile)
} else {
viper.AddConfigPath("$HOME/.myapp")
viper.AddConfigPath(".")
viper.SetConfigName("config")
}
viper.SetEnvPrefix("MYAPP")
viper.AutomaticEnv()
viper.ReadInConfig() // 忽略错误 - 配置文件是可选的
}
标志绑定策略 :
// 策略 1:单独绑定每个标志(显式)
viper.BindPFlag("log_level", rootCmd.Flags().Lookup("log-level"))
// 策略 2:自动绑定所有标志(方便)
viper.BindPFlags(rootCmd.Flags())
// 策略 3:混合方法(推荐)
// - 全局绑定持久标志
// - 在每个命令的 init() 中绑定本地标志
rootCmd.PersistentFlags().String("config", "", "配置文件")
viper.BindPFlags(rootCmd.PersistentFlags())
deployCmd.Flags().String("region", "us-east-1", "AWS 区域")
viper.BindPFlag("deploy.region", deployCmd.Flags().Lookup("region"))
使用 PersistentPreRunE 来加载和验证配置:
var rootCmd = &cobra.Command{
Use: "myapp",
Short: "我的应用程序",
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
// 在所有命令之前运行(被子命令继承)
// 1. 验证必需配置
if !viper.IsSet("api_key") {
return fmt.Errorf("未配置 API 密钥(设置 MYAPP_API_KEY 或添加到配置文件)")
}
// 2. 基于配置设置日志
logLevel := viper.GetString("log_level")
if err := setupLogging(logLevel); err != nil {
return fmt.Errorf("无效的日志级别:%w", err)
}
// 3. 初始化客户端/服务
apiKey := viper.GetString("api_key")
if err := initAPIClient(apiKey); err != nil {
return fmt.Errorf("初始化 API 客户端失败:%w", err)
}
return nil
},
}
Cobra 为 bash、zsh、fish 和 PowerShell 生成 shell 补全脚本。
// cmd/completion.go
package cmd
import (
"os"
"github.com/spf13/cobra"
)
var completionCmd = &cobra.Command{
Use: "completion [bash|zsh|fish|powershell]",
Short: "生成 shell 补全脚本",
Long: `为 myapp 生成 shell 补全脚本。
加载补全:
Bash:
$ source <(myapp completion bash)
# 要自动加载,请添加到 ~/.bashrc:
$ echo 'source <(myapp completion bash)' >> ~/.bashrc
Zsh:
$ source <(myapp completion zsh)
# 要自动加载,请添加到 ~/.zshrc:
$ echo 'source <(myapp completion zsh)' >> ~/.zshrc
Fish:
$ myapp completion fish | source
# 要自动加载:
$ myapp completion fish > ~/.config/fish/completions/myapp.fish
PowerShell:
PS> myapp completion powershell | Out-String | Invoke-Expression
# 要自动加载,请添加到 PowerShell 配置文件。
`,
DisableFlagsInUseLine: true,
ValidArgs: []string{"bash", "zsh", "fish", "powershell"},
Args: cobra.ExactValidArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
switch args[0] {
case "bash":
return cmd.Root().GenBashCompletion(os.Stdout)
case "zsh":
return cmd.Root().GenZshCompletion(os.Stdout)
case "fish":
return cmd.Root().GenFishCompletion(os.Stdout, true)
case "powershell":
return cmd.Root().GenPowerShellCompletionWithDesc(os.Stdout)
}
return nil
},
}
func init() {
rootCmd.AddCommand(completionCmd)
}
为命令参数提供动态补全:
var deployCmd = &cobra.Command{
Use: "deploy [environment]",
Short: "部署到环境",
Args: cobra.ExactArgs(1),
ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
// 返回可用环境
envs := []string{"dev", "staging", "production"}
return envs, cobra.ShellCompDirectiveNoFileComp
},
RunE: func(cmd *cobra.Command, args []string) error {
return deploy(args[0])
},
}
// 自定义标志补全
deployCmd.RegisterFlagCompletionFunc("region", func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
regions := []string{"us-east-1", "us-west-2", "eu-west-1"}
return regions, cobra.ShellCompDirectiveNoFileComp
})
// ❌ 不好:技术术语
return fmt.Errorf("数据库连接失败:EOF")
// ✅ 好:可操作的错误消息
return fmt.Errorf("无法连接到数据库 %s:%d\n请检查:\n - 数据库是否正在运行\n - 凭据是否正确 (MYAPP_DB_PASSWORD)\n - 网络连接性", host, port)
// ✅ 好:建议修复方法
if !viper.IsSet("api_key") {
return fmt.Errorf("未配置 API 密钥\n设置环境变量:export MYAPP_API_KEY=your-key\n或添加到配置文件:~/.myapp.yaml")
}
import "github.com/briandowns/spinner"
func deploy(env string) error {
s := spinner.New(spinner.CharSets[11], 100*time.Millisecond)
s.Suffix = " 正在部署到 " + env + "..."
s.Start()
defer s.Stop()
// 部署逻辑
if err := performDeployment(env); err != nil {
s.Stop()
return err
}
s.Stop()
fmt.Println("✓ 部署成功")
return nil
}
import (
"encoding/json"
"github.com/olekukonko/tablewriter"
)
func displayResults(items []Item, format string) error {
switch format {
case "json":
enc := json.NewEncoder(os.Stdout)
enc.SetIndent("", " ")
return enc.Encode(items)
case "table":
table := tablewriter.NewWriter(os.Stdout)
table.SetHeader([]string{"ID", "名称", "状态"})
for _, item := range items {
table.Append([]string{item.ID, item.Name, item.Status})
}
table.Render()
return nil
default:
return fmt.Errorf("未知格式:%s(使用 json 或 table)", format)
}
}
import (
"log"
"os"
)
var (
// 面向用户的输出(stdout)
out = os.Stdout
// 日志记录和错误(stderr)
logger = log.New(os.Stderr, "[myapp] ", log.LstdFlags)
)
func RunCommand() error {
// 用户输出:stdout
fmt.Fprintln(out, "正在处理文件...")
// 调试/详细日志记录:stderr
if viper.GetBool("verbose") {
logger.Println("从", viper.ConfigFileUsed(), "读取配置")
}
// 错误:stderr
if err := process(); err != nil {
fmt.Fprintf(os.Stderr, "错误:%v\n", err)
return err
}
// 成功消息:stdout
fmt.Fprintln(out, "✓ 完成")
return nil
}
// cmd/deploy_test.go
package cmd
import (
"bytes"
"testing"
"github.com/spf13/cobra"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func executeCommand(root *cobra.Command, args ...string) (output string, err error) {
buf := new(bytes.Buffer)
root.SetOut(buf)
root.SetErr(buf)
root.SetArgs(args)
err = root.Execute()
return buf.String(), err
}
func TestDeployCommand(t *testing.T) {
tests := []struct {
name string
args []string
wantErr bool
wantOut string
}{
{
name: "部署到 dev",
args: []string{"deploy", "dev"},
wantErr: false,
wantOut: "正在部署到 dev",
},
{
name: "部署到 production 但没有 force",
args: []string{"deploy", "production"},
wantErr: true,
wantOut: "生产环境部署需要 --force 标志",
},
{
name: "部署到 production 并带有 force",
args: []string{"deploy", "production", "--force"},
wantErr: false,
wantOut: "正在部署到 production",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
output, err := executeCommand(rootCmd, tt.args...)
if tt.wantErr {
require.Error(t, err)
} else {
require.NoError(t, err)
}
assert.Contains(t, output, tt.wantOut)
})
}
}
func TestCommandWithConfig(t *testing.T) {
// 在每个测试前重置 viper 状态
viper.Reset()
// 设置测试配置
viper.Set("region", "eu-west-1")
viper.Set("api_key", "test-key-123")
output, err := executeCommand(rootCmd, "deploy", "staging")
require.NoError(t, err)
assert.Contains(t, output, "eu-west-1")
}
func TestOutputFormat(t *testing.T) {
// 捕获 stdout
oldStdout := os.Stdout
r, w, _ := os.Pipe()
os.Stdout = w
defer func() { os.Stdout = oldStdout }()
// 执行命令
err := listCmd.RunE(listCmd, []string{})
require.NoError(t, err)
// 读取输出
w.Close()
var buf bytes.Buffer
io.Copy(&buf, r)
output := buf.String()
assert.Contains(t, output, "ID")
assert.Contains(t, output, "名称")
}
在以下情况下使用 Cobra:
git clone、docker run)--flag、-f)--help)在以下情况下不要使用 Cobra:
flag 包)在以下情况下使用 Viper:
在以下情况下不要使用 Viper:
在以下情况下添加 shell 补全:
在以下情况下跳过 shell 补全:
在以下情况下使用持久标志:
--verbose、--config)--dry-run、--output)在以下情况下使用本地标志:
--region 用于 deploy 命令)错误 :
var deployCmd = &cobra.Command{
Use: "deploy",
Run: func(cmd *cobra.Command, args []string) {
deploy(args[0]) // 忽略错误!
},
}
正确 :
var deployCmd = &cobra.Command{
Use: "deploy",
RunE: func(cmd *cobra.Command, args []string) error {
return deploy(args[0]) // 正确的错误处理
},
}
错误 :
// 混乱:哪个优先?
config.Port = viper.GetInt("port")
if os.Getenv("PORT") != "" {
config.Port = atoi(os.Getenv("PORT"))
}
if *flagPort != 0 {
config.Port = *flagPort
}
正确 :
// 清晰:Viper 自动处理优先级
viper.BindPFlag("port", rootCmd.Flags().Lookup("port"))
viper.SetEnvPrefix("MYAPP")
viper.AutomaticEnv()
viper.SetDefault("port", 8080)
config.Port = viper.GetInt("port") // 遵循:标志 > 环境变量 > 配置 > 默认值
错误 :
rootCmd.Flags().String("region", "us-east-1", "AWS 区域")
// 标志未绑定到 Viper - 不会遵循优先级!
func deploy() {
region := viper.GetString("region") // 始终返回配置文件的值
}
正确 :
rootCmd.Flags().String("region", "us-east-1", "AWS 区域")
viper.BindPFlag("region", rootCmd.Flags().Lookup("region")) // 绑定它!
func deploy() {
region := viper.GetString("region") // 遵循:标志 > 环境变量 > 配置
}
错误 :
// 没有 CLI 命令的测试 - 错误会悄悄溜过
正确 :
func TestDeployCommand(t *testing.T) {
output, err := executeCommand(rootCmd, "deploy", "staging", "--region", "eu-west-1")
require.NoError(t, err)
assert.Contains(t, output, "正在部署到 staging")
assert.Contains(t, output, "eu-west-1")
}
错误 :
return fmt.Errorf("连接失败") // 没有帮助
正确 :
return fmt.Errorf("无法连接到数据库 %s:%d\n检查:\n - 数据库是否正在运行\n - 凭据 (MYAPP_DB_PASSWORD)\n - 防火墙规则", host, port)
错误 :
var rootCmd = &cobra.Command{
Use: "myapp",
Run: func(cmd *cobra.Command, args []string) {
if err := execute(); err != nil {
fmt.Println(err) // 错误未传播
}
},
}
正确 :
var rootCmd = &cobra.Command{
Use: "myapp",
RunE: func(cmd *cobra.Command, args []string) error {
return execute() // Cobra 处理错误显示和退出代码
},
}
最小化的生产就绪 CLI 结构:
myapp/
├── cmd/
│ ├── root.go # 根命令 + 配置加载
│ ├── deploy.go # Deploy 子命令
│ ├── status.go # Status 子命令
│ └── completion.go # Shell 补全
├── main.go # 入口点
├── config.yaml # 示例配置文件
└── go.mod
main.go :
package main
import "myapp/cmd"
func main() {
cmd.Execute()
}
cmd/root.go :参见上面的“命令结构模式”部分
构建和安装 :
# 开发
go run main.go deploy staging --region us-west-2
# 生产构建
go build -o myapp
# 全局安装
go install
# 启用 shell 补全
myapp completion bash > /etc/bash_completion.d/myapp
官方文档 :
学习资源 :
生产示例 :
相关技能 :
golang-testing-strategies - 全面测试 CLI 命令golang-http-servers - 使用配置构建 API 服务器golang-concurrency-patterns - CLI 工具中的异步操作当您满足以下条件时,您就掌握了 Cobra + Viper:
VERB NOUN --FLAG)每周安装次数
152
仓库
GitHub 星标数
27
首次出现
2026 年 1 月 23 日
安全审计
安装于
opencode124
codex120
gemini-cli119
github-copilot117
claude-code116
cursor115
Cobra and Viper are the industry-standard libraries for building production-quality CLIs in Go. Cobra provides command structure and argument parsing, while Viper manages configuration from multiple sources with clear precedence rules.
Key Features:
app verb noun --flag)Used By : Kubernetes (kubectl), Docker CLI, GitHub CLI (gh), Hugo, Helm, and 100+ major projects
Activate this skill when:
Cobra follows the APPNAME VERB NOUN --FLAG pattern popularized by git and kubectl.
// cmd/root.go
package cmd
import (
"fmt"
"os"
"github.com/spf13/cobra"
"github.com/spf13/viper"
)
var cfgFile string
var rootCmd = &cobra.Command{
Use: "myapp",
Short: "A powerful CLI tool for developers",
Long: `MyApp is a CLI tool that demonstrates best practices
for building production-quality command-line applications.
Complete documentation is available at https://myapp.example.com`,
}
func Execute() {
if err := rootCmd.Execute(); err != nil {
fmt.Fprintln(os.Stderr, err)
os.Exit(1)
}
}
func init() {
cobra.OnInitialize(initConfig)
// Persistent flags (available to all subcommands)
rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.myapp.yaml)")
rootCmd.PersistentFlags().Bool("verbose", false, "verbose output")
// Bind persistent flags to viper
viper.BindPFlag("config", rootCmd.PersistentFlags().Lookup("config"))
viper.BindPFlag("verbose", rootCmd.PersistentFlags().Lookup("verbose"))
}
func initConfig() {
if cfgFile != "" {
viper.SetConfigFile(cfgFile)
} else {
home, err := os.UserHomeDir()
if err != nil {
fmt.Fprintln(os.Stderr, err)
os.Exit(1)
}
viper.AddConfigPath(home)
viper.AddConfigPath(".")
viper.SetConfigType("yaml")
viper.SetConfigName(".myapp")
}
viper.SetEnvPrefix("MYAPP")
viper.AutomaticEnv()
if err := viper.ReadInConfig(); err == nil {
if viper.GetBool("verbose") {
fmt.Println("Using config file:", viper.ConfigFileUsed())
}
}
}
// cmd/deploy.go
package cmd
import (
"fmt"
"github.com/spf13/cobra"
"github.com/spf13/viper"
)
var deployCmd = &cobra.Command{
Use: "deploy [environment]",
Short: "Deploy application to specified environment",
Long: `Deploy the application to the specified environment.
Supports: dev, staging, production`,
Args: cobra.ExactArgs(1),
ValidArgs: []string{"dev", "staging", "production"},
PreRunE: func(cmd *cobra.Command, args []string) error {
// Validation logic runs before RunE
env := args[0]
if env == "production" && !viper.GetBool("force") {
return fmt.Errorf("production deploys require --force flag")
}
return nil
},
RunE: func(cmd *cobra.Command, args []string) error {
env := args[0]
region := viper.GetString("region")
force := viper.GetBool("force")
fmt.Printf("Deploying to %s in region %s (force=%v)\n", env, region, force)
// Actual deployment logic
return deploy(env, region, force)
},
PostRunE: func(cmd *cobra.Command, args []string) error {
// Cleanup or notifications
fmt.Println("Deployment complete")
return nil
},
}
func init() {
rootCmd.AddCommand(deployCmd)
// Local flags (only for this command)
deployCmd.Flags().StringP("region", "r", "us-east-1", "AWS region")
deployCmd.Flags().BoolP("force", "f", false, "Force deployment without confirmation")
// Bind flags to viper
viper.BindPFlag("region", deployCmd.Flags().Lookup("region"))
viper.BindPFlag("force", deployCmd.Flags().Lookup("force"))
}
func deploy(env, region string, force bool) error {
// Implementation
return nil
}
// Persistent flags: Available to command and all subcommands
rootCmd.PersistentFlags().String("config", "", "config file path")
rootCmd.PersistentFlags().Bool("verbose", false, "verbose output")
// Local flags: Only available to this specific command
deployCmd.Flags().String("region", "us-east-1", "deployment region")
deployCmd.Flags().Bool("force", false, "force deployment")
// Required flags
deployCmd.MarkFlagRequired("region")
// Flag dependencies
deployCmd.MarkFlagsRequiredTogether("username", "password")
deployCmd.MarkFlagsMutuallyExclusive("json", "yaml")
Cobra provides execution hooks for setup and cleanup:
var serverCmd = &cobra.Command{
Use: "server",
Short: "Start API server",
// Execution order (all optional):
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
// Runs before PreRunE, inherited by subcommands
return setupLogging()
},
PreRunE: func(cmd *cobra.Command, args []string) error {
// Validation and setup before RunE
return validateConfig()
},
RunE: func(cmd *cobra.Command, args []string) error {
// Main command logic
return startServer()
},
PostRunE: func(cmd *cobra.Command, args []string) error {
// Cleanup after RunE
return cleanup()
},
PersistentPostRunE: func(cmd *cobra.Command, args []string) error {
// Runs after PostRunE, inherited by subcommands
return flushLogs()
},
}
Important : Use RunE, PreRunE, PostRunE (error-returning versions) instead of Run, PreRun, PostRun.
Viper follows a strict precedence order (highest to lowest):
viper.Set("key", value))viper.BindPFlag)MYAPP_KEY=value)~/.myapp.yaml, ./config.yaml)viper.SetDefault("key", value))func initConfig() {
// 1. Set defaults (lowest priority)
viper.SetDefault("port", 8080)
viper.SetDefault("database.host", "localhost")
viper.SetDefault("database.port", 5432)
// 2. Config file locations (checked in order)
viper.SetConfigName("config")
viper.SetConfigType("yaml")
viper.AddConfigPath("/etc/myapp/")
viper.AddConfigPath("$HOME/.myapp")
viper.AddConfigPath(".")
// 3. Environment variables (prefix + automatic mapping)
viper.SetEnvPrefix("MYAPP")
viper.AutomaticEnv() // MYAPP_PORT, MYAPP_DATABASE_HOST, etc.
// 4. Read config file (optional)
if err := viper.ReadInConfig(); err != nil {
if _, ok := err.(viper.ConfigFileNotFoundError); ok {
// Config file not found - use defaults and env vars
} else {
// Config file found but error reading it
return err
}
}
// 5. Flags will be bound in init() functions (highest priority)
}
Viper automatically maps environment variables with prefix and dot notation:
viper.SetEnvPrefix("MYAPP") // Prefix for env vars
viper.AutomaticEnv() // Enable automatic mapping
// Config key → Environment variable
// "port" → MYAPP_PORT
// "database.host" → MYAPP_DATABASE_HOST
// "database.port" → MYAPP_DATABASE_PORT
// "aws.s3.region" → MYAPP_AWS_S3_REGION
Manual mapping for non-standard env var names:
viper.BindEnv("database.host", "DB_HOST") // Custom env var name
viper.BindEnv("database.password", "DB_PASSWORD") // Different naming convention
Viper supports multiple formats: YAML, JSON, TOML, HCL, INI, envfile, Java properties.
config.yaml :
port: 8080
log_level: info
database:
host: localhost
port: 5432
user: postgres
ssl_mode: require
aws:
region: us-east-1
s3:
bucket: my-app-bucket
Accessing config values :
port := viper.GetInt("port") // 8080
dbHost := viper.GetString("database.host") // "localhost"
s3Bucket := viper.GetString("aws.s3.bucket") // "my-app-bucket"
// Type-safe access
if viper.IsSet("database.ssl_mode") {
sslMode := viper.GetString("database.ssl_mode")
}
// Unmarshal into struct
type Config struct {
Port int `mapstructure:"port"`
LogLevel string `mapstructure:"log_level"`
Database struct {
Host string `mapstructure:"host"`
Port int `mapstructure:"port"`
User string `mapstructure:"user"`
SSLMode string `mapstructure:"ssl_mode"`
} `mapstructure:"database"`
}
var config Config
if err := viper.Unmarshal(&config); err != nil {
return err
}
The key to Cobra + Viper integration is binding flags to Viper keys:
// cmd/root.go
func init() {
cobra.OnInitialize(initConfig) // Load config before command execution
// Define flags
rootCmd.PersistentFlags().String("config", "", "config file")
rootCmd.PersistentFlags().String("log-level", "info", "log level")
rootCmd.PersistentFlags().Int("port", 8080, "server port")
// Bind flags to Viper (critical step!)
viper.BindPFlag("config", rootCmd.PersistentFlags().Lookup("config"))
viper.BindPFlag("log_level", rootCmd.PersistentFlags().Lookup("log-level"))
viper.BindPFlag("port", rootCmd.PersistentFlags().Lookup("port"))
}
func initConfig() {
// This runs BEFORE command execution via cobra.OnInitialize
if cfgFile := viper.GetString("config"); cfgFile != "" {
viper.SetConfigFile(cfgFile)
} else {
viper.AddConfigPath("$HOME/.myapp")
viper.AddConfigPath(".")
viper.SetConfigName("config")
}
viper.SetEnvPrefix("MYAPP")
viper.AutomaticEnv()
viper.ReadInConfig() // Ignore errors - config file is optional
}
Flag binding strategies :
// Strategy 1: Bind each flag individually (explicit)
viper.BindPFlag("log_level", rootCmd.Flags().Lookup("log-level"))
// Strategy 2: Bind all flags automatically (convenient)
viper.BindPFlags(rootCmd.Flags())
// Strategy 3: Hybrid approach (recommended)
// - Bind persistent flags globally
// - Bind local flags in each command's init()
rootCmd.PersistentFlags().String("config", "", "config file")
viper.BindPFlags(rootCmd.PersistentFlags())
deployCmd.Flags().String("region", "us-east-1", "AWS region")
viper.BindPFlag("deploy.region", deployCmd.Flags().Lookup("region"))
Use PersistentPreRunE to load and validate configuration:
var rootCmd = &cobra.Command{
Use: "myapp",
Short: "My application",
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
// Runs before ALL commands (inherited by subcommands)
// 1. Validate required config
if !viper.IsSet("api_key") {
return fmt.Errorf("API key not configured (set MYAPP_API_KEY or add to config file)")
}
// 2. Setup logging based on config
logLevel := viper.GetString("log_level")
if err := setupLogging(logLevel); err != nil {
return fmt.Errorf("invalid log level: %w", err)
}
// 3. Initialize clients/services
apiKey := viper.GetString("api_key")
if err := initAPIClient(apiKey); err != nil {
return fmt.Errorf("failed to initialize API client: %w", err)
}
return nil
},
}
Cobra generates shell completion scripts for bash, zsh, fish, and PowerShell.
// cmd/completion.go
package cmd
import (
"os"
"github.com/spf13/cobra"
)
var completionCmd = &cobra.Command{
Use: "completion [bash|zsh|fish|powershell]",
Short: "Generate shell completion script",
Long: `Generate shell completion script for myapp.
To load completions:
Bash:
$ source <(myapp completion bash)
# To load automatically, add to ~/.bashrc:
$ echo 'source <(myapp completion bash)' >> ~/.bashrc
Zsh:
$ source <(myapp completion zsh)
# To load automatically, add to ~/.zshrc:
$ echo 'source <(myapp completion zsh)' >> ~/.zshrc
Fish:
$ myapp completion fish | source
# To load automatically:
$ myapp completion fish > ~/.config/fish/completions/myapp.fish
PowerShell:
PS> myapp completion powershell | Out-String | Invoke-Expression
# To load automatically, add to PowerShell profile.
`,
DisableFlagsInUseLine: true,
ValidArgs: []string{"bash", "zsh", "fish", "powershell"},
Args: cobra.ExactValidArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
switch args[0] {
case "bash":
return cmd.Root().GenBashCompletion(os.Stdout)
case "zsh":
return cmd.Root().GenZshCompletion(os.Stdout)
case "fish":
return cmd.Root().GenFishCompletion(os.Stdout, true)
case "powershell":
return cmd.Root().GenPowerShellCompletionWithDesc(os.Stdout)
}
return nil
},
}
func init() {
rootCmd.AddCommand(completionCmd)
}
Provide dynamic completions for command arguments:
var deployCmd = &cobra.Command{
Use: "deploy [environment]",
Short: "Deploy to environment",
Args: cobra.ExactArgs(1),
ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
// Return available environments
envs := []string{"dev", "staging", "production"}
return envs, cobra.ShellCompDirectiveNoFileComp
},
RunE: func(cmd *cobra.Command, args []string) error {
return deploy(args[0])
},
}
// Custom flag completion
deployCmd.RegisterFlagCompletionFunc("region", func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
regions := []string{"us-east-1", "us-west-2", "eu-west-1"}
return regions, cobra.ShellCompDirectiveNoFileComp
})
// ❌ BAD: Technical jargon
return fmt.Errorf("db connection failed: EOF")
// ✅ GOOD: Actionable error message
return fmt.Errorf("cannot connect to database at %s:%d\nPlease check:\n - Database is running\n - Credentials are correct (MYAPP_DB_PASSWORD)\n - Network connectivity", host, port)
// ✅ GOOD: Suggest remediation
if !viper.IsSet("api_key") {
return fmt.Errorf("API key not configured\nSet environment variable: export MYAPP_API_KEY=your-key\nOr add to config file: ~/.myapp.yaml")
}
import "github.com/briandowns/spinner"
func deploy(env string) error {
s := spinner.New(spinner.CharSets[11], 100*time.Millisecond)
s.Suffix = " Deploying to " + env + "..."
s.Start()
defer s.Stop()
// Deployment logic
if err := performDeployment(env); err != nil {
s.Stop()
return err
}
s.Stop()
fmt.Println("✓ Deployment successful")
return nil
}
import (
"encoding/json"
"github.com/olekukonko/tablewriter"
)
func displayResults(items []Item, format string) error {
switch format {
case "json":
enc := json.NewEncoder(os.Stdout)
enc.SetIndent("", " ")
return enc.Encode(items)
case "table":
table := tablewriter.NewWriter(os.Stdout)
table.SetHeader([]string{"ID", "Name", "Status"})
for _, item := range items {
table.Append([]string{item.ID, item.Name, item.Status})
}
table.Render()
return nil
default:
return fmt.Errorf("unknown format: %s (use json or table)", format)
}
}
import (
"log"
"os"
)
var (
// User-facing output (stdout)
out = os.Stdout
// Logging and errors (stderr)
logger = log.New(os.Stderr, "[myapp] ", log.LstdFlags)
)
func RunCommand() error {
// User output: stdout
fmt.Fprintln(out, "Processing files...")
// Debug/verbose logging: stderr
if viper.GetBool("verbose") {
logger.Println("Reading config from", viper.ConfigFileUsed())
}
// Errors: stderr
if err := process(); err != nil {
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
return err
}
// Success message: stdout
fmt.Fprintln(out, "✓ Complete")
return nil
}
// cmd/deploy_test.go
package cmd
import (
"bytes"
"testing"
"github.com/spf13/cobra"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func executeCommand(root *cobra.Command, args ...string) (output string, err error) {
buf := new(bytes.Buffer)
root.SetOut(buf)
root.SetErr(buf)
root.SetArgs(args)
err = root.Execute()
return buf.String(), err
}
func TestDeployCommand(t *testing.T) {
tests := []struct {
name string
args []string
wantErr bool
wantOut string
}{
{
name: "deploy to dev",
args: []string{"deploy", "dev"},
wantErr: false,
wantOut: "Deploying to dev",
},
{
name: "deploy to production without force",
args: []string{"deploy", "production"},
wantErr: true,
wantOut: "production deploys require --force flag",
},
{
name: "deploy to production with force",
args: []string{"deploy", "production", "--force"},
wantErr: false,
wantOut: "Deploying to production",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
output, err := executeCommand(rootCmd, tt.args...)
if tt.wantErr {
require.Error(t, err)
} else {
require.NoError(t, err)
}
assert.Contains(t, output, tt.wantOut)
})
}
}
func TestCommandWithConfig(t *testing.T) {
// Reset viper state before each test
viper.Reset()
// Set test configuration
viper.Set("region", "eu-west-1")
viper.Set("api_key", "test-key-123")
output, err := executeCommand(rootCmd, "deploy", "staging")
require.NoError(t, err)
assert.Contains(t, output, "eu-west-1")
}
func TestOutputFormat(t *testing.T) {
// Capture stdout
oldStdout := os.Stdout
r, w, _ := os.Pipe()
os.Stdout = w
defer func() { os.Stdout = oldStdout }()
// Execute command
err := listCmd.RunE(listCmd, []string{})
require.NoError(t, err)
// Read output
w.Close()
var buf bytes.Buffer
io.Copy(&buf, r)
output := buf.String()
assert.Contains(t, output, "ID")
assert.Contains(t, output, "Name")
}
Use Cobra when:
git clone, docker run)--flag, -f)--help)Don't use Cobra when:
flag package)Use Viper when:
Don't use Viper when:
Add shell completion when:
Skip shell completion when:
Use persistent flags when:
--verbose, --config)--dry-run, --output)Use local flags when:
--region for deploy command)Wrong :
var deployCmd = &cobra.Command{
Use: "deploy",
Run: func(cmd *cobra.Command, args []string) {
deploy(args[0]) // Ignores error!
},
}
Correct :
var deployCmd = &cobra.Command{
Use: "deploy",
RunE: func(cmd *cobra.Command, args []string) error {
return deploy(args[0]) // Proper error handling
},
}
Wrong :
// Confusing: Which takes precedence?
config.Port = viper.GetInt("port")
if os.Getenv("PORT") != "" {
config.Port = atoi(os.Getenv("PORT"))
}
if *flagPort != 0 {
config.Port = *flagPort
}
Correct :
// Clear: Viper handles precedence automatically
viper.BindPFlag("port", rootCmd.Flags().Lookup("port"))
viper.SetEnvPrefix("MYAPP")
viper.AutomaticEnv()
viper.SetDefault("port", 8080)
config.Port = viper.GetInt("port") // Respects: flag > env > config > default
Wrong :
rootCmd.Flags().String("region", "us-east-1", "AWS region")
// Flag not bound to Viper - won't respect precedence!
func deploy() {
region := viper.GetString("region") // Always returns config file value
}
Correct :
rootCmd.Flags().String("region", "us-east-1", "AWS region")
viper.BindPFlag("region", rootCmd.Flags().Lookup("region")) // Bind it!
func deploy() {
region := viper.GetString("region") // Respects flag > env > config
}
Wrong :
// No tests for CLI commands - bugs slip through
Correct :
func TestDeployCommand(t *testing.T) {
output, err := executeCommand(rootCmd, "deploy", "staging", "--region", "eu-west-1")
require.NoError(t, err)
assert.Contains(t, output, "Deploying to staging")
assert.Contains(t, output, "eu-west-1")
}
Wrong :
return fmt.Errorf("connection failed") // Unhelpful
Correct :
return fmt.Errorf("cannot connect to database at %s:%d\nCheck:\n - Database is running\n - Credentials (MYAPP_DB_PASSWORD)\n - Firewall rules", host, port)
Wrong :
var rootCmd = &cobra.Command{
Use: "myapp",
Run: func(cmd *cobra.Command, args []string) {
if err := execute(); err != nil {
fmt.Println(err) // Error not propagated
}
},
}
Correct :
var rootCmd = &cobra.Command{
Use: "myapp",
RunE: func(cmd *cobra.Command, args []string) error {
return execute() // Cobra handles error display and exit code
},
}
Minimal production-ready CLI structure:
myapp/
├── cmd/
│ ├── root.go # Root command + config loading
│ ├── deploy.go # Deploy subcommand
│ ├── status.go # Status subcommand
│ └── completion.go # Shell completion
├── main.go # Entry point
├── config.yaml # Example config file
└── go.mod
main.go :
package main
import "myapp/cmd"
func main() {
cmd.Execute()
}
cmd/root.go : See "Command Structure Pattern" section above
Building and installing :
# Development
go run main.go deploy staging --region us-west-2
# Production build
go build -o myapp
# Install globally
go install
# Enable shell completion
myapp completion bash > /etc/bash_completion.d/myapp
Official Documentation :
Learning Resources :
Production Examples :
Related Skills :
golang-testing-strategies - Testing CLI commands comprehensivelygolang-http-servers - Building API servers with configurationgolang-concurrency-patterns - Async operations in CLI toolsYou know you've mastered Cobra + Viper when:
VERB NOUN --FLAG)Weekly Installs
152
Repository
GitHub Stars
27
First Seen
Jan 23, 2026
Security Audits
Gen Agent Trust HubPassSocketPassSnykWarn
Installed on
opencode124
codex120
gemini-cli119
github-copilot117
claude-code116
cursor115
Azure 升级评估与自动化工具 - 轻松迁移 Functions 计划、托管层级和 SKU
90,800 周安装
浏览器自动化专家指南:Playwright、Puppeteer、Selenium实战技巧与最佳实践
317 周安装
AI文档生成与验证工具 - 自动检测项目类型、生成API文档、检查覆盖率
320 周安装
文档协同创作工作流:AI辅助结构化写作指南,提升团队文档质量与效率
319 周安装
using-superpowers技能使用指南:AI助手强制技能调用规则与工作流程优化
320 周安装
YouTube字幕下载器 - 一键下载视频字幕/说明文字,支持yt-dlp与Whisper转录
322 周安装
AiCoin Trading - 安全合规的加密货币交易下单工具,支持OKX、币安等主流交易所
327 周安装