mirror of
https://github.com/keven1024/015.git
synced 2026-05-26 07:08:02 +00:00
30 lines
450 B
Go
30 lines
450 B
Go
package utils
|
|
|
|
import (
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var v = viper.New()
|
|
|
|
func init() {
|
|
v.SetConfigName(".env")
|
|
v.SetConfigType("env")
|
|
v.AddConfigPath(".")
|
|
v.AutomaticEnv()
|
|
if err := v.ReadInConfig(); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func GetEnv(key string) string {
|
|
return v.GetString(key)
|
|
}
|
|
|
|
func GetEnvWithDefault(key string, defaultValue string) string {
|
|
value := v.GetString(key)
|
|
if value == "" {
|
|
return defaultValue
|
|
}
|
|
return value
|
|
}
|