|
@@ -0,0 +1,25 @@
|
|
|
+package configs
|
|
|
+
|
|
|
+import (
|
|
|
+ "github.com/spf13/viper"
|
|
|
+ "os"
|
|
|
+)
|
|
|
+
|
|
|
+var Config *viper.Viper
|
|
|
+
|
|
|
+func init() {
|
|
|
+ Config = viper.New()
|
|
|
+ dir, _ := os.Getwd()
|
|
|
+
|
|
|
+ _, err := os.Stat(dir + "/config.yaml")
|
|
|
+ if err != nil {
|
|
|
+ panic(any("配置文件不存在,请检查配置文件!"))
|
|
|
+ }
|
|
|
+
|
|
|
+ Config.SetConfigName("config")
|
|
|
+ Config.SetConfigType("yaml")
|
|
|
+ Config.AddConfigPath(dir)
|
|
|
+ if err := Config.ReadInConfig(); err != nil {
|
|
|
+ panic(any(err.Error()))
|
|
|
+ }
|
|
|
+}
|