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.
54 lines
1.5 KiB
54 lines
1.5 KiB
package common
|
|
|
|
import (
|
|
"github.com/fsnotify/fsnotify"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/spf13/viper"
|
|
"os"
|
|
"sendmsg/global"
|
|
"sendmsg/model"
|
|
)
|
|
|
|
// InitConfig
|
|
//
|
|
// @Description: 初始化配置
|
|
// @return model.Config
|
|
// @return error
|
|
func InitConfig() (model.Config, error) {
|
|
workDir, _ := os.Getwd()
|
|
viper.SetConfigName("config")
|
|
viper.SetConfigType("yml")
|
|
viper.AddConfigPath("../config")
|
|
viper.AddConfigPath(workDir + "/config")
|
|
viper.OnConfigChange(func(e fsnotify.Event) {
|
|
log.Printf("[conf] config update: %v\n", e.Name)
|
|
global.GLO_CONF_CH <- updateConfig()
|
|
})
|
|
viper.WatchConfig()
|
|
err := viper.ReadInConfig()
|
|
if err != nil {
|
|
return model.Config{}, err
|
|
}
|
|
return updateConfig(), nil
|
|
}
|
|
|
|
// updateConfig
|
|
//
|
|
// @Description: 更新配置
|
|
// @return model.Config
|
|
func updateConfig() model.Config {
|
|
var config model.Config
|
|
config.MsgEnabled = viper.GetBool("message.enabled")
|
|
config.MsgType = viper.GetString("message.type")
|
|
config.BarkUrl = viper.GetString("message.bark.url")
|
|
config.BarkKey = viper.GetString("message.bark.key")
|
|
config.MailHost = viper.GetString("message.mail.host")
|
|
config.MailProtocol = viper.GetString("message.mail.protocol")
|
|
config.MailPort = viper.GetInt("message.mail.port")
|
|
config.MailUser = viper.GetString("message.mail.username")
|
|
config.MailPwd = viper.GetString("message.mail.password")
|
|
config.MailFromName = viper.GetString("message.mail.from_name")
|
|
config.MailTo = viper.GetStringSlice("message.mail.to")
|
|
return config
|
|
}
|