如何用Golang Viper实现CLI工具从配置映射数据库别名参数？

1. 确定配置文件路径 #

默认将配置放在用户主目录的隐藏文件夹中，比如 ~/.toolbox/config.yaml，同时支持通过--config参数指定自定义配置路径，适配不同场景需求。

2. 读取解析YAML配置 #

使用对应开发语言的YAML解析库，把配置文件中的databases节点转换成键值对结构，方便后续别名匹配。

3. 解析CLI参数并匹配别名 #

捕获-host参数传入的别名，从解析后的配置中取出对应的实际数据库地址；如果传入的不是配置中的别名，则直接将其作为连接地址使用，提升工具灵活性。

Python 示例（基于Click CLI框架）

import yaml
import click
import os

@click.group()
def toolbox():
    pass

@toolbox.group()
def db():
    pass

@db.command(name="connect")
@click.option("--host", required=True, help="数据库别名或地址")
def connect(host):
    # 读取配置文件
    config_path = os.path.expanduser("~/.toolbox/config.yaml")
    if not os.path.exists(config_path):
        click.echo(f"错误：配置文件不存在，请创建 {config_path}", err=True)
        return

    with open(config_path, "r") as f:
        try:
            config = yaml.safe_load(f)
        except yaml.YAMLError as e:
            click.echo(f"错误：解析配置文件失败 - {str(e)}", err=True)
            return

    # 匹配别名，无匹配则直接使用传入值
    databases = config.get("databases", {})
    actual_host = databases.get(host, host)

    # 此处替换为实际数据库连接逻辑
    click.echo(f"正在连接到实际地址：{actual_host}")

if __name__ == "__main__":
    toolbox()

Go 示例（基于Cobra CLI框架）

package main

import (
	"fmt"
	"os"
	"path/filepath"

	"gopkg.in/yaml.v3"

	"github.com/spf13/cobra"
)

type Config struct {
	Databases map[string]string `yaml:"databases"`
}

var (
	hostFlag string
	config   Config
)

func main() {
	var rootCmd = &cobra.Command{Use: "toolbox"}
	var dbCmd = &cobra.Command{Use: "db"}
	var connectCmd = &cobra.Command{
		Use:   "connect",
		Short: "连接到数据库",
		Run: func(cmd *cobra.Command, args []string) {
			// 加载配置文件
			configPath := filepath.Join(os.Getenv("HOME"), ".toolbox", "config.yaml")
			file, err := os.Open(configPath)
			if err != nil {
				fmt.Printf("错误：配置文件不存在，请创建 %s\n", configPath)
				return
			}
			defer file.Close()

			if err := yaml.NewDecoder(file).Decode(&config); err != nil {
				fmt.Printf("错误：解析配置文件失败 - %v\n", err)
				return
			}

			// 匹配别名，无匹配则直接使用传入值
			actualHost, exists := config.Databases[hostFlag]
			if !exists {
				actualHost = hostFlag
			}

			// 此处替换为实际数据库连接逻辑
			fmt.Printf("正在连接到实际地址：%s\n", actualHost)
		},
	}

	connectCmd.Flags().StringVarP(&hostFlag, "host", "", "", "数据库别名或地址")
	connectCmd.MarkFlagRequired("host")

	dbCmd.AddCommand(connectCmd)
	rootCmd.AddCommand(dbCmd)

	if err := rootCmd.Execute(); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

关键容错处理 #

• 检查配置文件是否存在，给出明确的创建提示
• 捕获YAML格式错误，输出具体错误信息
• 兼容直接输入数据库地址的场景，避免强制依赖配置

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