config.go 990 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. package config
  2. import (
  3. "fmt"
  4. "gopkg.in/yaml.v3"
  5. "os"
  6. "reflect"
  7. )
  8. type IConfig interface {
  9. Invalidate() error
  10. }
  11. func Invalidate(config IConfig) error {
  12. if config == nil {
  13. return fmt.Errorf("config must not be a nil")
  14. }
  15. val := reflect.ValueOf(config)
  16. if val.Kind() != reflect.Struct {
  17. if val.IsZero() {
  18. return fmt.Errorf("config is empty")
  19. }
  20. val = reflect.ValueOf(config).Elem()
  21. }
  22. for i := 0; i < val.NumField(); i++ {
  23. if !val.Field(i).CanInterface() || val.Field(i).IsZero() {
  24. continue
  25. }
  26. if elm, ok := val.Field(i).Interface().(IConfig); ok {
  27. if er := elm.Invalidate(); er != nil {
  28. return er
  29. }
  30. }
  31. }
  32. return nil
  33. }
  34. func ReadConfig[T IConfig](path string) (*T, error) {
  35. var (
  36. cfg = new(T)
  37. fd []byte
  38. err error
  39. )
  40. if fd, err = os.ReadFile(path); err != nil {
  41. return nil, err
  42. }
  43. if err = yaml.Unmarshal(fd, cfg); err != nil {
  44. return nil, err
  45. }
  46. if err = Invalidate(*cfg); err != nil {
  47. return nil, err
  48. }
  49. return cfg, nil
  50. }