如何使用yaml文件根据环境设置golang应用程序配置

pxy2qtax 于 2023-01-28 发布在 Go

关注(0)|答案(1)|浏览(133)

我试图设置一个配置结构，通过我的应用程序使用。
目前，我加载了一个yaml文件，并在我的配置结构体中对其进行解码。

一个月一次

database_url: postgres://postgres:@localhost:5432/database_dev

一米一分一秒

import (
  "os"
  "gopkg.in/yaml.v2"
)

type AppConfig struct {
  DatabaseUrl      string `yaml:"database_url"`
}

func LoadConfig() *AppConfig {
  appConfig := &AppConfig{}
  file, _ := os.Open("config.yml")
  defer f.Close()
  decoder := yaml.NewDecoder(file)
  decoder.Decode(config)
  return appConfig
}

它工作得非常好，但是现在我需要根据环境（测试、本地、生产等）设置不同的配置。
我认为可以使用嵌套的yaml文件来声明环境变量。

`config.yml`

dev:
  database_url: postgres://postgres:@localhost:5432/database_dev
test:
  database_url: postgres://postgres:@localhost:5432/database_test

我想在LoadConfig函数中接收环境作为参数，并得到正确的配置，但是我不知道怎么做。

`config.go`

type configFile struct {
  Dev struct { AppConfig }`yaml:"dev"`
  Test struct { AppConfig }`yaml:"test"` 
}

func LoadConfig(env string) *AppConfig {
  appConfig := &AppConfig{}
  configFile := &configFile{}
  file, _ := os.Open("config.yml")
  defer f.Close()
  decoder := yaml.NewDecoder(file)
  decoder.Decode(configFile)
  
  // How to get the correct struct here ?
  // config = configFile["env"]
  // It doesn't works 
  // invalid operation: cannot index configFile (variable of type *configFile)

  return appConfig
}

欢迎提出任何建议。

go

来源：https://stackoverflow.com/questions/75263364/how-to-setup-a-golang-app-config-according-to-the-environment-using-yaml-file

1条答案

按热度按时间

shyt4zoc1#

如果环境列表是任意的，那么您不希望顶层使用struct;你需要一个map[string]AppConfig。它看起来像这样：

package main

import (
  "fmt"
  "os"

  "gopkg.in/yaml.v2"
)

type (
  AppConfig struct {
    DatabaseUrl string `yaml:"database_url"`
  }

  ConfigFile map[string]*AppConfig
)

func LoadConfig(env string) (*AppConfig, error) {
  configFile := ConfigFile{}
  file, _ := os.Open("config.yml")
  defer file.Close()
  decoder := yaml.NewDecoder(file)

  // Always check for errors!
  if err := decoder.Decode(&configFile); err != nil {
    return nil, err
  }

  appConfig, ok := configFile[env]
  if !ok {
    return nil, fmt.Errorf("no such environment: %s", env)
  }

  return appConfig, nil
}

func main() {
  appConfig, err := LoadConfig(os.Args[1])
  if err != nil {
    panic(err)
  }
  fmt.Printf("config: %+v\n", appConfig)
}

假设我们从您的问题中得到了config.yml，我们可以在不同的环境中运行上面的示例，并看到所需的输出：

$ ./example test
config: &{DatabaseUrl:postgres://postgres:@localhost:5432/database_test}
$ ./example dev
config: &{DatabaseUrl:postgres://postgres:@localhost:5432/database_dev}

赞(0）回复(0）举报 2023-01-28

我来回答

如何使用yaml文件根据环境设置golang应用程序配置

一个月一次

一米一分一秒

`config.yml`

`config.go`

1条答案

相关问题

热门标签

最新问答