forked from lxh/go-wxhelper
48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
|
package initialize
|
||
|
|
|
||
|
|
import (
|
||
|
|
"gitee.ltd/lxh/logger/log"
|
||
|
|
"github.com/fsnotify/fsnotify"
|
||
|
|
"github.com/spf13/viper"
|
||
|
|
"wechat-robot/config"
|
||
|
|
"wechat-robot/internal/database"
|
||
|
|
"wechat-robot/internal/redis"
|
||
|
|
)
|
||
|
|
|
||
|
|
// 配置管理工具
|
||
|
|
var vp *viper.Viper
|
||
|
|
|
||
|
|
// Config
|
||
|
|
// @description: 初始化配置
|
||
|
|
func initConfig() {
|
||
|
|
vp = viper.New()
|
||
|
|
vp.AddConfigPath(".") // 设置配置文件路径
|
||
|
|
vp.SetConfigName("config") // 设置配置文件名
|
||
|
|
vp.SetConfigType("yaml") // 设置配置文件类型
|
||
|
|
// 读取配置文件
|
||
|
|
if err := vp.ReadInConfig(); err != nil {
|
||
|
|
log.Panicf("读取配置文件失败: %v", err)
|
||
|
|
}
|
||
|
|
// 绑定配置文件
|
||
|
|
if err := vp.Unmarshal(&config.Conf); err != nil {
|
||
|
|
log.Panicf("配置文件解析失败: %v", err)
|
||
|
|
}
|
||
|
|
log.Debugf("配置文件解析完成: %+v", config.Conf)
|
||
|
|
// 初始化数据库连接
|
||
|
|
database.Init()
|
||
|
|
redis.Init()
|
||
|
|
|
||
|
|
// 下面的代码是配置变动之后自动刷新的
|
||
|
|
vp.WatchConfig()
|
||
|
|
vp.OnConfigChange(func(e fsnotify.Event) {
|
||
|
|
// 绑定配置文件
|
||
|
|
if err := vp.Unmarshal(&config.Conf); err != nil {
|
||
|
|
log.Errorf("配置文件更新失败: %v", err)
|
||
|
|
} else {
|
||
|
|
// 初始化数据库连接
|
||
|
|
database.Init()
|
||
|
|
redis.Init()
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|