config.go 518 B

123456789101112131415161718192021222324252627
  1. package config
  2. import (
  3. "os"
  4. "github.com/spf13/viper"
  5. )
  6. func init() {
  7. viper.SetConfigName("shopperthree_manage")
  8. viper.SetConfigType("yaml")
  9. viper.AddConfigPath(".")
  10. viper.AddConfigPath("..")
  11. viper.AddConfigPath("/etc/shopperthree_manage")
  12. if err := viper.ReadInConfig(); err != nil {
  13. panic(err)
  14. }
  15. if path := viper.GetString("log.path"); path != "" {
  16. f, err := os.OpenFile(path, os.O_CREATE|os.O_RDWR|os.O_APPEND, os.ModePerm)
  17. if err != nil {
  18. panic(err)
  19. }
  20. os.Stdout = f
  21. os.Stderr = f
  22. }
  23. }