Go：使用Viper和YAML管理配置文件，2024年最新腾讯架构师首发

最新推荐文章于 2025-04-01 16:49:52 发布

2401_84252890

最新推荐文章于 2025-04-01 16:49:52 发布

阅读量984

点赞数 5

分类专栏：程序员文章标签： golang 数据库开发语言

本文链接：https://blog.csdn.net/2401_84252890/article/details/137705818

版权

程序员专栏收录该内容

17 篇文章

订阅专栏

package main

import (
“fmt”
“github.com/spf13/viper”
)

func main() {
viper.SetConfigName(“config”) // 配置文件名称(无扩展名)
viper.SetConfigType(“yaml”) // 或viper.SetConfigType(“YAML”)
viper.AddConfigPath(“.”) // 配置文件路径
err := viper.ReadInConfig() // 查找并读取配置文件
if err != nil { // 处理读取配置文件的错误
panic(fmt.Errorf(“Fatal error config file: %w \n”, err))
}

fmt.Println(“Server Port:”, viper.GetInt(“server.port”))
fmt.Println(“Database User:”, viper.GetString(“database.user”))
}

通过上述步骤，我们的Go应用程序就可以轻松地读取并使用YAML配置文件中的配置了。Viper库使得管理配置变得更加简单，无论是在小型项目还是在复杂的微服务架构中，都能提高开发和维护的效率。

如果我们的YAML配置文件中包含了列表（数组）类型的数据，Viper同样提供了简单的方法来读取这些数据。这对于配置文件中有复杂数据结构时尤为重要。下面我们将通过一个例子来详细说明如何读取YAML配置文件中的列表数据。

假设我们有一个config.yaml文件，内容如下：

server:
port: 8080

databases:

user: admin
password: secret
host: localhost
user: guest
password: guest123
host: localhost

在这个配置文件中，databases是一个列表，每个元素包含三个属性：user、password和host。

读取列表配置

要在Go程序中读取一个列表，我们可以使用Viper的Get函数配合类型断言，或直接使用专门的函数如GetStringMapString（适用于键值对列表）。不过，由于我们的列表中包含复杂对象，我们更倾向于使用Unmarshal功能来将配置直接绑定到一个结构体切片中。

首先，定义一个与YAML结构相匹配的Go结构体：

package main

import (
“fmt”
“github.com/spf13/viper”
)

type Config struct {
Server struct {
Port int mapstructure:"port"
} mapstructure:"server"
Databases []struct {
User string mapstructure:"user"
Password string mapstructure:"password"
Host string mapstructure:"host"
} mapstructure:"databases"
}

func main() {
viper.SetConfigName(“config”)
viper.SetConfigType(“yaml”)
viper.AddConfigPath(“.”)

if err := viper.ReadInConfig(); err != nil {
fmt.Printf(“Error reading config file, %s”, err)
}

var config Config
if err := viper.Unmarshal(&config); err != nil {
fmt.Printf(“Unable to decode into struct, %v”, err)
}

fmt.Println(“Server Port:”, config.Server.Port)
for _, db := range config.Databases {
fmt.Printf(“Database User: %s, Password: %s, Host: %s\n”, db.User, db.Password, db.Host)
}
}

在这个例子中，我们首先定义了一个Config结构体，它反映了YAML配置文件的结构。然后，我们使用viper.Unmarshal方法将配置文件的内容自动绑定到Config结构体实例上。最后，通过遍历Databases切片，我们可以轻松访问列表中的每个数据库配置。

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Go语言工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Go语言全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。