我有以下代碼從yml讀取配置文件，其中也包括ENV變量：confContent, err := ioutil.ReadFile("config.yml") if err != nil { panic(err) } // expand environment variables confContent = []byte(os.ExpandEnv(string(confContent))) conf := &SysConfig{} if err := yaml.Unmarshal(confContent, conf); err != nil { panic(err) }config.ymldb: name: ${DB_NAME:qm} host: localhost它正在工作，但是如果沒有給出env，我怎么能讓它讀取默認值？DB_NAME
3 回答
呼如林
TA貢獻1798條經驗 獲得超3個贊
您可以使用時替換映射器，并考慮默認值，如下所示：ExpandEnvExpand
package main
import (
"fmt"
"os"
"strings"
)
func main() {
mapper := func(placeholderName string) string {
split := strings.Split(placeholderName, ":")
defValue := ""
if len(split) == 2 {
placeholderName = split[0]
defValue = split[1]
}
val, ok := os.LookupEnv(placeholderName)
if !ok {
return defValue
}
return val
}
os.Setenv("DAY_PART", "morning")
fmt.Println(os.Expand("Good ${DAY_PART:test}, ${NAME:Gopher}", mapper))
}
這將呈現
Good morning, Gopher
這是基于操作系統包文檔中的擴展中的示例。
Smart貓小萌
TA貢獻1911條經驗 獲得超7個贊
我建議你使用spf13）的真棒Viper軟件包來讀取conf文件，它可以優雅地解決你的問題，你可以用它來加載許多其他類型的配置文件。
解決您的問題
獲取軟件包
go get github.com/spf13/viper
配置文件
假設你有名為的配置文件：db.yaml
db: name: somedb host: localhost
代碼示例
正如我們所看到的，一旦Viper加載了conf文件，我們就可以通過鍵獲取值。Yaml文件將被解析為嵌套結構，您可以將其解封為Golang結構，我們應該使用它來獲取值，您可以參考此頁面以獲取更多用法信息。viper.GetString("db.name")
import (
"fmt"
"github.com/spf13/viper"
)
func InitConf() {
viper.AutomaticEnv() // read system env
viper.SetConfigName("db") // conf file name to be load
viper.AddConfigPath(".") // conf file path
viper.SetDefault("db.name", "mysqldb") // you can set default da name value here
// do read conf file and handle error
if err := viper.ReadInConfig(); err != nil {
if _, ok := err.(viper.ConfigFileNotFoundError); ok {
// Config file not found; ignore error if desired
} else {
// Config file was found but another error was produced
}
}
fmt.Printf("db.name: %s", viper.GetString("db.name"))
}
viper.AutomaticEnv()此方法將自動讀取系統 env，假設您有一個名為 env 的 env，然后您可以通過以下方式獲取它。如果 env 名稱與配置文件密鑰的名稱相同，則 viper 首先使用 envENV_XXXviper.GetString("ENV_XXX")
當我們不設置輸出時，輸出將是
namedb.yamldb.name: mysqldb當我們設置值時：在 db.yaml 中，輸出將是
namesomedbdb.name: somedb
希望你會發現它有用！有關用法的更多詳細信息，請參閱蝰蛇自述文件
守著一只汪
TA貢獻1872條經驗 獲得超4個贊
我猜你有一個這樣的會議結構：
Db *struct{
Name string
Host string
}因此，要檢索Dbname的默認變量，您必須像這樣檢查它：
if conf.Db == nil || conf.Db.Name == "" {
conf.Db.Name = "test"
}- 3 回答
- 0 關注
- 400 瀏覽
添加回答
舉報
0/150
提交
取消