create project

2021-12-03 17:11:18 +08:00
commit 627c6582e3
15 changed files with 1713 additions and 0 deletions
--- a/cfg/cfg.go
+++ b/cfg/cfg.go
@@ -0,0 +1,100 @@
+/**
+ * @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
+}