config.go 547 B

1234567891011121314151617181920212223242526272829
  1. package config
  2. import (
  3. "log"
  4. "github.com/spf13/viper"
  5. )
  6. // Config 全局配置
  7. type Config struct {
  8. ServerPort string
  9. DBPath string
  10. }
  11. // AppConfig 全局配置实例
  12. var AppConfig Config
  13. // InitConfig 初始化配置
  14. func InitConfig() {
  15. viper.SetConfigFile("config.yaml") // 配置文件路径
  16. err := viper.ReadInConfig()
  17. if err != nil {
  18. log.Fatal("Failed to read configuration file:", err)
  19. }
  20. // 读取配置信息
  21. AppConfig.ServerPort = viper.GetString("server.port")
  22. AppConfig.DBPath = viper.GetString("database.path")
  23. }