配置文件读取

viperx.go

@@ -0,0 +1,58 @@
+package viperx
+
+// Author: yun
+// Version: 2023年6月12日18:54:48
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"os"
+
+	"github.com/fsnotify/fsnotify"
+	"github.com/spf13/viper"
+)
+
+// 使用viper初始化配置
+
+func InitConfig(path string, data interface{}) error {
+
+	if len(path) == 0 {
+		flag.StringVar(&path, "c", "", "choose config file.")
+		flag.Parse()
+		if path == "" { // 优先级: 命令行 > 环境变量 > 默认值
+			if configEnv := os.Getenv("config"); configEnv == "" {
+				path = "config.yaml"
+				log.Printf("您正在使用config的默认值,config的路径为%v\n", path)
+			} else {
+				path = configEnv
+				log.Printf("您正在使用GVA_CONFIG环境变量,config的路径为%v\n", path)
+			}
+		} else {
+			log.Printf("您正在使用命令行的-c参数传递的值,config的路径为%v\n", path)
+		}
+	}
+
+	v := viper.New()
+	v.SetConfigFile(path)
+	err := v.ReadInConfig()
+	if err != nil {
+		return err
+	}
+	// 监控配置文件的变化
+	v.WatchConfig()
+
+	// 监听配置的变化
+	v.OnConfigChange(func(e fsnotify.Event) {
+		fmt.Println("config file changed:", e.Name)
+		if err := v.Unmarshal(&data); err != nil {
+			log.Println(err)
+		}
+	})
+
+	if err := v.Unmarshal(&data); err != nil {
+		log.Println(err)
+		return err
+	}
+	return nil
+}