add http config loader

This commit is contained in:
yanghao05
2023-04-27 16:27:09 +08:00
parent da7cde1149
commit f1fc4cf7fa
3 changed files with 65 additions and 13 deletions

44
providers/http/env.go Normal file
View File

@@ -0,0 +1,44 @@
package http
import (
"log"
"github.com/rogeecn/atom/utils/fs"
"github.com/spf13/viper"
)
const DefaultPrefix = "HTTP"
func AutoLoadConfig() *Config {
return LoadConfig("", DefaultPrefix)
}
func LoadConfig(file, envPrefix string) *Config {
if file == "" {
file = "config.toml"
}
if envPrefix == "" {
envPrefix = DefaultPrefix
}
viper.SetEnvPrefix(envPrefix)
viper.AutomaticEnv()
if !fs.FileExist(file) {
return &Config{}
}
// load file
viper.SetConfigFile(file)
if err := viper.ReadInConfig(); err != nil {
log.Fatal(err)
}
var config Config
if err := viper.Unmarshal(&config); err != nil {
log.Fatal(err)
}
return &config
}