将接口转换为定义的结构体（struct）Golang。

huangapple
go评论112阅读模式
英文:

convert interface to defined struct Golang

问题

我正在处理一个实用方法，假设叫做GetConfig()，它读取配置结构并将其返回给调用者。GetConfig() 不知道要读取的配置是什么，但调用者知道它将接收到的结构。

为此，我编写了下面的实用程序：

  1. =========== yaml 文件数据 ==========
  2. apiRouting:
  3.   enableThrottling: true
  4.   formFactor: 4
  5. leasing:
  6.   periodInSecs: 10
  7.   preemptionEnable: false
  8. =========== yaml 文件数据 ==========
  10. func GetConfig() (interface{}, error) {
  11. 	fmt.Println("正在读取通用服务配置")
  12. 	viper.SetConfigName("service_config")
  13. 	viper.AddConfigPath("config/default")
  14. 	if err := viper.ReadInConfig(); err != nil {
  15. 		return nil, err
  16. 	}
  17. 	var appConfig interface{}
  18. 	if err := viper.Unmarshal(&appConfig); err != nil {
  19. 		return nil, err
  20. 	}
  21. 	return appConfig, nil
  22. }

调用者使用 GetConfig() 如下：（我尝试了两种选项，但都没有成功）

  1. type ApiRouting struct {
  2. 	EnableThrottling bool  `json:"enableThrottling"`
  3. 	FormFactor       int32 `json:"formFactor"`
  4. }
  6. type Leasing struct {
  7. 	PeriodInSecs     int32 `json:"periodInSecs"`
  8. 	PreemptionEnable bool  `json:"preemptionEnable"`
  9. }
  11. type ServiceConfig struct {
  12. 	ApiRouting ApiRouting `json:"apiRouting"`
  13. 	Leasing    Leasing    `json:"leasing"`
  14. }
  16. // 代码片段 [选项 1]
  17. tmpinterface := GetConfig()
  18. myconfig, ok := tmpinterface.(ServiceConfig)
  19. if !ok {
  20. 	log.Fatal()
  21. } else {
  22. 	println(myconfig)
  23. }
  25. // 代码片段 [选项 2]
  26. tmpinterface := GetConfig()
  27. // 将 map 转换为 JSON 字符串
  28. jsonStr, err := json.Marshal(tmpinterface)
  29. if err != nil {
  30. 	fmt.Println(err)
  31. }
  33. // 将 JSON 字符串转换为结构体
  34. var sc ServiceConfig
  35. if err := json.Unmarshal(jsonStr, &sc); err != nil {
  36. 	fmt.Println(err)
  37. }

我已经验证了在这两种情况下 tmpinterface 正确地获取了值，但最终 myconfig{} 结构体是空的。

tmpinterface 的值是：

  1. map["apirouting"]:map["enablethrottling":true "formfactor":4] "leasing":map["periodinsecs":10 "preemptionenable":false]
英文:

I am working on a utility method lets say GetConfig(), which reads the config struct and return it to the caller. GetConfig() does not know what config it is going to read, but the caller knows the struct what it is going to receive.

In this regard, I have written a below utility program:

  1. =========== yaml file data ==========
  2. apiRouting:
  3.   enableThrottling: true
  4.   formFactor: 4
  5. leasing:
  6.   periodInSecs: 10
  7.   preemptionEnable: false
  8. =========== yaml file data ==========
  10. func GetConfig() (interface{}, error) {
  11. 	fmt.Println("reading generic service config")
  12. 	viper.SetConfigName("service_config")
  13. 	viper.AddConfigPath("config/default")
  14. 	if err := viper.ReadInConfig(); err != nil {
  15. 		return nil, err
  16. 	}
  17. 	var appConfig interface{}
  18. 	if err := viper.Unmarshal(&appConfig); err != nil {
  19. 		return nil, err
  20. 	}
  21. 	return appConfig, nil
  22. }

Caller uses GetConfig() like this: (I have tried 2 options, nothing is working)

  1. type ApiRouting struct {
  2. 	EnableThrottling bool  `json:"enableThrottling"`
  3. 	FormFactor       int32 `json:"formFactor"`
  4. }
  6. type Leasing struct {
  7. 	PeriodInSecs     int32 `json:"periodInSecs"`
  8. 	PreemptionEnable bool  `json:"preemptionEnable"`
  9. }
  11. type ServiceConfig struct {
  12. 	ApiRouting ApiRouting `json:"apiRouting"`
  13. 	Leasing    Leasing    `json:"leasing"`
  14. }
  16. // code snipped [option 1]
  17. 	tmpinterface := GetConfig()
  18. 	myconfig, ok := tmpinterface.(ServiceConfig)
  19. 	if !ok {
  20. 		log.Fatal()
  21. 	} else {
  22. 		println(myconfig)
  23. 	}
  25. // code snipped [option 2]
  26. 	tmpinterface := GetConfig()
  27. 	// Convert map to json string
  28. 	jsonStr, err := json.Marshal(tmpinterface)
  29. 	if err != nil {
  30. 		fmt.Println(err)
  31. 	}
  32. 	
  33. 	// Convert json string to struct
  34. 	var sc ServiceConfig
  35. 	if err := json.Unmarshal(jsonStr, &sc); err != nil {
  36. 		fmt.Println(err)
  37. 	}

I have verified that tmpinterface is getting the values correctly in both the cases, but finally myconfig{} struct is empty.

tmpinterface value is:

  1. map[%!f(string=apirouting):map[%!f(string=enablethrottling):%!f(bool=true) %!f(string=formfactor):%!f(int=4)] %!f(string=leasing):map[%!f(string=periodinsecs):%!f(int=10) %!f(string=preemptionenable):%!f(bool=false)]]

答案1

得分: 1

@mkopriva，感谢您提供了一个更简洁的解决方案。

  1. func GetConfig(appConfig any) error {
  2. 	fmt.Println("读取通用服务配置")
  3. 	viper.SetConfigName("service_config")
  4. 	viper.AddConfigPath("config/default")
  5. 	if err := viper.ReadInConfig(); err != nil {
  6. 		return err
  7. 	}
  8. 	if err := viper.Unmarshal(appConfig); err != nil {
  9. 		return err
  10. 	}
  11. 	return nil
  12. }
  14. func main() {
  15. 	var sc ServiceConfig
  16. 	if err := GetConfig(&sc); err != nil {
  17. 		panic(err)
  18. 	}
  19. 	fmt.Println(sc)
  20. }
英文:

@mkopriva, thanks for a cleaner solution.

  1. func GetConfig(appConfig any) error {
  2. 	fmt.Println("reading generic service config")
  3. 	viper.SetConfigName("service_config")
  4. 	viper.AddConfigPath("config/default")
  5. 	if err := viper.ReadInConfig(); err != nil {
  6. 		return err
  7. 	}
  8. 	if err := viper.Unmarshal(appConfig); err != nil {
  9. 		return err
  10. 	}
  11. 	return nil
  12. }
  14. func main() {
  15. 	var sc ServiceConfig
  16. 	if err := GetConfig(&sc); err != nil {
  17. 		panic(err)
  18. 	}
  19. 	fmt.Println(sc)
  20. }

huangapple
  • 本文由 发表于 2022年12月5日 14:07:10
  • 转载请务必保留本文链接：https://go.coder-hub.com/74684006.html
  • casting
  • go
  • interface
  • struct
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定