Go:使用Viper和YAML管理配置文件,2024年最新腾讯架构师首发
package main
import (
"fmt"
"log"
"github.com/spf13/viper"
)
func main() {
viper.SetConfigFile("config.yaml") // 指定配置文件路径
if err := viper.ReadInConfig(); err != nil { // 读取配置文件
log.Fatalf("Error reading config file, %s", err)
}
// 获取配置参数
username := viper.GetString("database.username")
password := viper.GetString("database.password")
host := viper.GetString("database.host")
port := viper.GetInt("database.port")
fmt.Printf("Database username: %s\n", username)
fmt.Printf("Database password: %s\n", password)
fmt.Printf("Database host: %s\n", host)
fmt.Printf("Database port: %d\n", port)
}
这段代码演示了如何使用Viper库读取YAML格式的配置文件,并获取相应的配置参数。首先,我们通过viper.SetConfigFile
设置配置文件的路径,然后调用viper.ReadInConfig
读取配置文件。如果读取成功,我们可以通过viper.GetString
和viper.GetInt
获取字符串和整数类型的配置参数。如果读取配置文件失败,程序会记录错误信息并退出。这是一个简单的实践,展示了如何在Go语言中管理配置文件。
评论已关闭