config.go 534 B

123456789101112131415161718192021222324252627282930
  1. package config
  2. import (
  3. "github.com/knadh/koanf"
  4. "github.com/knadh/koanf/parsers/yaml"
  5. "github.com/knadh/koanf/providers/file"
  6. "sync"
  7. )
  8. var k = koanf.New(".")
  9. var sy sync.Once
  10. //func init() {
  11. // Init()
  12. //}
  13. func init() {
  14. sy.Do(func() {
  15. err := k.Load(file.Provider("./config/config.yaml"), yaml.Parser())
  16. if err != nil {
  17. err = k.Load(file.Provider("../config/config.yaml"), yaml.Parser())
  18. if err != nil {
  19. panic("加载配置文件错误" + err.Error())
  20. }
  21. }
  22. })
  23. }
  24. func GetConfig() *koanf.Koanf {
  25. return k
  26. }