You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
2.6 KiB
Go
101 lines
2.6 KiB
Go
/**
|
|
* @Author: jager
|
|
* @Email: lhj168os@gmail.com
|
|
* @File: cfg
|
|
* @Date: 2021/12/2 6:24 下午
|
|
* @package: cfg
|
|
* @Version: v1.0.0
|
|
*
|
|
* @Description:
|
|
*
|
|
*/
|
|
|
|
package cfg
|
|
|
|
import (
|
|
"github.com/jageros/hawox/logx"
|
|
"github.com/spf13/pflag"
|
|
"github.com/spf13/viper"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
QywxRobotUrl = ""
|
|
DingTalkSecret = ""
|
|
DingTalkRobotUrl = ""
|
|
TianApiKey = ""
|
|
UpdateDuration = time.Second * 60 // 默认一分钟
|
|
ThresholdValue float64 = 1 // 默认1%
|
|
StockCodes = []string{"600905", "600032", "002531", "600733", "000825", "002939"} // 三峡能源， 浙江新能， 天顺风能， 北汽蓝谷， 太钢不锈， 长城证券
|
|
FundCodes = []string{"006229", "162412", "008359", "161725", "012314", "162719", "501057"}
|
|
)
|
|
|
|
func load(v *viper.Viper) {
|
|
QywxRobotUrl = v.GetString("qywxRobotUrl")
|
|
DingTalkSecret = v.GetString("dingTalkSecret")
|
|
DingTalkRobotUrl = v.GetString("dingTalkRobotUrl")
|
|
TianApiKey = v.GetString("tianApiKey")
|
|
updates := v.GetInt("updateSecond")
|
|
if updates > 0 {
|
|
UpdateDuration = time.Duration(updates) * time.Second
|
|
}
|
|
ThresholdValue = v.GetFloat64("thresholdValue")
|
|
StockCodes = v.GetStringSlice("stock")
|
|
FundCodes = v.GetStringSlice("fund")
|
|
}
|
|
|
|
func Init() error {
|
|
pflag.String("c", "config.yaml", "config file")
|
|
pflag.Parse()
|
|
v := viper.New()
|
|
err := v.BindPFlags(pflag.CommandLine)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
path := v.GetString("c")
|
|
|
|
var dir, fileName, fileType string
|
|
// 配置文件路径不为空则读取配置文件， 即命令行有传入配置文件路径时
|
|
if path != "" {
|
|
// 获取配置文件的后缀名
|
|
strs := strings.Split(path, ".")
|
|
length := len(strs)
|
|
if length < 2 {
|
|
logx.Fatal("错误的配置文件路径")
|
|
}
|
|
|
|
fileType = strs[length-1]
|
|
switch fileType {
|
|
// 支持的文件类型yaml、json、 toml、hcl, ini
|
|
case "yaml", "yml", "json", "toml", "hcl", "ini":
|
|
fPath := strings.Replace(path, "."+fileType, "", -1)
|
|
strs = strings.Split(fPath, "/")
|
|
if len(strs) < 2 {
|
|
fileName = strs[0]
|
|
dir = "./"
|
|
} else {
|
|
fileName = strs[len(strs)-1]
|
|
dir = strings.Join(strs[:len(strs)-1], "/")
|
|
}
|
|
|
|
//设置读取的配置文件
|
|
v.SetConfigName(fileName)
|
|
//添加读取的配置文件路径
|
|
v.AddConfigPath(dir)
|
|
//设置配置文件类型
|
|
v.SetConfigType(fileType)
|
|
|
|
if err := v.ReadInConfig(); err != nil {
|
|
logx.Fatalf("v.ReadInConfig err: %v", err)
|
|
}
|
|
|
|
default:
|
|
// 其他类型抛出错误
|
|
logx.Fatal("错误的配置文件类型")
|
|
}
|
|
}
|
|
load(v)
|
|
return nil
|
|
}
|