详解Go使用Viper和YAML管理配置文件

go get github.com/spf13/viper

server:
  port: 8080
database:
  user: admin
  password: secret

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"))
}

server:
  port: 8080

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

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)
    }
}