config.go 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  1. package config
  2. import (
  3. "fmt"
  4. "github.com/fsnotify/fsnotify"
  5. "github.com/spf13/viper"
  6. )
  7. type Config struct {
  8. MysqlAdmin MysqlAdmin `json:"mysqlAdmin"`
  9. Qiniu Qiniu `json:"qiniu"`
  10. CasbinConfig CasbinConfig `json:"casbinConfig"`
  11. RedisAdmin RedisAdmin `json:"redisAdmin"`
  12. System System `json:"system"`
  13. JWT JWT `json:"jwt"`
  14. Captcha Captcha `json:"captcha"`
  15. Log Log `json:"log"`
  16. }
  17. type System struct { // 系统配置
  18. UseMultipoint bool `json:"useMultipoint"`
  19. Env string `json:"env"`
  20. Addr int `json:"addr"`
  21. }
  22. type JWT struct { // jwt签名
  23. SigningKey string `json:"signingKey"`
  24. }
  25. type CasbinConfig struct { //casbin配置
  26. ModelPath string `json:"modelPath"` // casbin model地址配置
  27. }
  28. type MysqlAdmin struct { // mysql admin 数据库配置
  29. Username string `json:"username"`
  30. Password string `json:"password"`
  31. Path string `json:"path"`
  32. Dbname string `json:"dbname"`
  33. Config string `json:"config"`
  34. }
  35. type RedisAdmin struct { // Redis admin 数据库配置
  36. Addr string `json:"addr"`
  37. Password string `json:"password"`
  38. DB int `json:"db"`
  39. }
  40. type Qiniu struct { // 七牛 密钥配置
  41. AccessKey string `json:"accessKey"`
  42. SecretKey string `json:"secretKey"`
  43. }
  44. type Captcha struct { // 验证码配置
  45. KeyLong int `json:"keyLong"`
  46. ImgWidth int `json:"imgWidth"`
  47. ImgHeight int `json:"imgHeight"`
  48. }
  49. /**
  50. Log Config
  51. "CRITICAL"
  52. "ERROR"
  53. "WARNING"
  54. "NOTICE"
  55. "INFO"
  56. "DEBUG"
  57. */
  58. type Log struct {
  59. // log 打印的前缀
  60. Prefix string `json:"prefix"`
  61. // 是否显示打印log的文件具体路径
  62. LogFile bool `json:"logFile"`
  63. // 在控制台打印log的级别, []默认不打印
  64. Stdout []string `json:"stdout"`
  65. // 在文件中打印log的级别 []默认不打印
  66. File []string `json:"file"`
  67. }
  68. var GinVueAdminconfig Config
  69. var VTool *viper.Viper
  70. func init() {
  71. v := viper.New()
  72. v.SetConfigName("config") // 设置配置文件名 (不带后缀)
  73. v.AddConfigPath("./static/config/") // 第一个搜索路径
  74. v.SetConfigType("json")
  75. err := v.ReadInConfig() // 搜索路径,并读取配置数据
  76. if err != nil {
  77. panic(fmt.Errorf("Fatal error config file: %s \n", err))
  78. }
  79. v.WatchConfig()
  80. v.OnConfigChange(func(e fsnotify.Event) {
  81. fmt.Println("config file changed:", e.Name)
  82. if err := v.Unmarshal(&GinVueAdminconfig); err != nil {
  83. fmt.Println(err)
  84. }
  85. })
  86. if err := v.Unmarshal(&GinVueAdminconfig); err != nil {
  87. fmt.Println(err)
  88. }
  89. VTool = v
  90. }