You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

105 lines
2.6 KiB

  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. MaxIdleConns int `json:"maxIdleConns"`
  35. MaxOpenConns int `json:"maxOpenConns"`
  36. LogMode bool `json:"maxOpenConns"`
  37. }
  38. type RedisAdmin struct { // Redis admin 数据库配置
  39. Addr string `json:"addr"`
  40. Password string `json:"password"`
  41. DB int `json:"db"`
  42. }
  43. type Qiniu struct { // 七牛 密钥配置
  44. AccessKey string `json:"accessKey"`
  45. SecretKey string `json:"secretKey"`
  46. }
  47. type Captcha struct { // 验证码配置
  48. KeyLong int `json:"keyLong"`
  49. ImgWidth int `json:"imgWidth"`
  50. ImgHeight int `json:"imgHeight"`
  51. }
  52. /**
  53. Log Config
  54. "CRITICAL"
  55. "ERROR"
  56. "WARNING"
  57. "NOTICE"
  58. "INFO"
  59. "DEBUG"
  60. */
  61. type Log struct {
  62. // log 打印的前缀
  63. Prefix string `json:"prefix"`
  64. // 是否显示打印log的文件具体路径
  65. LogFile bool `json:"logFile"`
  66. // 在控制台打印log的级别, []默认不打印
  67. Stdout []string `json:"stdout"`
  68. // 在文件中打印log的级别 []默认不打印
  69. File []string `json:"file"`
  70. }
  71. var GinVueAdminconfig Config
  72. var VTool *viper.Viper
  73. func init() {
  74. v := viper.New()
  75. v.SetConfigName("config") // 设置配置文件名 (不带后缀)
  76. v.AddConfigPath("./static/config/") // 第一个搜索路径
  77. v.SetConfigType("json")
  78. err := v.ReadInConfig() // 搜索路径,并读取配置数据
  79. if err != nil {
  80. panic(fmt.Errorf("Fatal error config file: %s \n", err))
  81. }
  82. v.WatchConfig()
  83. v.OnConfigChange(func(e fsnotify.Event) {
  84. fmt.Println("config file changed:", e.Name)
  85. if err := v.Unmarshal(&GinVueAdminconfig); err != nil {
  86. fmt.Println(err)
  87. }
  88. })
  89. if err := v.Unmarshal(&GinVueAdminconfig); err != nil {
  90. fmt.Println(err)
  91. }
  92. VTool = v
  93. }