46 lines
819 B
Go
46 lines
819 B
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/spf13/viper"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
defaultConfigName = "config.dev"
|
|
envConfigPrefix = "XH_G"
|
|
)
|
|
|
|
var cfg *Config
|
|
|
|
// LoadConfig 加载并返回应用配置
|
|
func LoadConfig(configDir string) (*Config, error) {
|
|
v := viper.New()
|
|
|
|
v.SetEnvPrefix(envConfigPrefix)
|
|
v.AutomaticEnv()
|
|
v.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
|
|
|
|
env := v.GetString("env")
|
|
if env == "" {
|
|
env = "dev"
|
|
}
|
|
|
|
v.SetConfigName(fmt.Sprintf("config.%s", strings.ToLower(env)))
|
|
v.AddConfigPath(configDir)
|
|
v.SetConfigType("yaml")
|
|
|
|
if err := v.ReadInConfig(); err != nil {
|
|
return nil, fmt.Errorf("读取配置失败: %w", err)
|
|
}
|
|
|
|
if err := v.Unmarshal(&cfg); err != nil {
|
|
return nil, fmt.Errorf("解析配置失败: %w", err)
|
|
}
|
|
return cfg, nil
|
|
}
|
|
|
|
func Get() *Config {
|
|
return cfg
|
|
}
|