我是 Go 新手，想知道如何以慣用的方式解決以下問題：我正在使用 viper 將配置文件加載到程序中。我選擇 toml 格式是因為我想要一個配置文件，它可以指定幾種不同格式的所需輸入：例如，Alpha 提供程序需要一個 apikey，而 Beta 提供程序需要用戶名和密碼。[alpha]apikey = "123"domain = "example.com"# [beta]# username = ""# password = ""# domain = ""type ProviderService interface { PrintDomain() } type Provider struct { Alpha `mapstructure:"alpha"` Beta `mapstructure:"beta"` } type Alpha struct { Apikey string `mapstructure:"apikey"` Domain string `mapstructure:"domain"` } type Beta struct { Username string `mapstructure:"username"` Password string `mapstructure:"password"` Domain string `mapstructure:"domain"` } 上面的代碼導致{Alpha:{Apikey:123 Domain:example.com} Beta:{Username: Password: Domain:}}空/未使用的結構仍然存在。最終，我希望 ProviderService 接口與提供者無關，因此我可以簡單地調用provider.PrintDomain()而不是provider.Alpha.PrintDomain()讓代碼中散布著 if/else 語句。我也對構建代碼以實現這一結果的其他方式持開放態度。提前致謝！
2 回答
繁華開滿天機
TA貢獻1816條經驗 獲得超4個贊
這就是我最終得到的結果，它允許我根據 toml 文件中指定的內容有條件地加載結構，并且仍然使用接口將其視為結構不可知論。
如果有人有任何關于將其重構為更慣用的技巧，請告訴我！
type ProviderService interface {
DoAThing()
}
type Alpha struct {
Apikey string `mapstructure:"apikey"`
Domain string `mapstructure:"domain"`
}
type Beta struct {
Username string `mapstructure:"username"`
Password string `mapstructure:"password"`
Domain string `mapstructure:"domain"`
}
func main() {
provider := loadConfig()
if provider == nil {
log.Fatal("unable to parse config file")
}
provider.DoAThing()
}
func (a Alpha) DoAThing() {
fmt.Println("domain", a.Domain)
}
func (b Beta) DoAThing() {
fmt.Println("domain", b.Domain)
}
func loadConfig() ProviderService {
viper.AddConfigPath("./")
viper.SetConfigName("config")
viper.SetConfigType("toml")
err := viper.ReadInConfig()
if err != nil {
panic(fmt.Errorf("Fatal error config file: %w \n", err))
}
var a Alpha
_ = viper.UnmarshalKey("alpha", &a)
if a != (Alpha{}) {
return a
}
var b Beta
_ = viper.UnmarshalKey("beta", &b)
if b != (Beta{}) {
return b
}
return nil
}
- 2 回答
- 0 關注
- 128 瀏覽
添加回答
舉報
0/150
提交
取消