无须畏惧，Golang配置解析黑科技viper，让配置管理无所不能

2023-09-09 22:32:58

作为一名资深Go语言开发者，我经常需要处理复杂的应用程序配置。为了简化配置管理，我一直在寻找一个功能强大、易于使用的配置解析库。经过一番探索，我发现了一个名为viper的库，它完全满足了我的需求。

viper是一个用于Go语言的配置解析库，它可以从命令行参数、环境变量、配置文件等来源加载和解析配置。它支持多种配置文件格式，包括YAML、TOML、JSON等。viper还提供了一些高级特性，如配置加密、命令行参数覆盖配置文件、配置验证等。

在这篇文章中，我将详细介绍viper的基本用法以及一些高级技巧。

基本用法

安装

go get -u github.com/spf13/viper

初始化

import (
	"github.com/spf13/viper"
)

func main() {
	viper.SetConfigName("config")
	viper.AddConfigPath("/etc/app/")
	viper.AddConfigPath("$HOME/.app")
	viper.AutomaticEnv()
	err := viper.ReadInConfig()
	if err != nil {
		panic(fmt.Errorf("Fatal error config file: %s", err))
	}
}

获取配置值

value := viper.GetString("key")

设置配置值

viper.Set("key", value)

持久化配置值

viper.WriteConfig()

高级用法

命令行参数覆盖配置文件

viper.BindPFlag("key", flag.String("key", "", "Description"))

配置加密

viper.SetConfigType("yaml")
viper.AddConfigPath("/etc/app/")
viper.AddConfigPath("$HOME/.app")
viper.SetConfigFile("config")
viper.SetEncryptionKey([]byte("my-secret-key"))
err := viper.ReadInConfig()
if err != nil {
	panic(fmt.Errorf("Fatal error config file: %s", err))
}

配置验证

viper.SetDefault("key", "default value")
viper.RegisterAlias("alias", "key")
viper.SetConfigType("yaml")
viper.AddConfigPath("/etc/app/")
viper.AddConfigPath("$HOME/.app")
viper.SetConfigFile("config")
err := viper.ReadInConfig()
if err != nil {
	panic(fmt.Errorf("Fatal error config file: %s", err))
}

// Validate the configuration
if err := viper.ReadInConfig(); err != nil {
	panic(fmt.Errorf("Fatal error config file: %s", err))
}
if viper.GetBool("key") {
	fmt.Println("Key is enabled")
} else {
	fmt.Println("Key is disabled")
}