Cold_DeviceEarlywarning/config/config.go

package config

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()))
	}
}