84 lines
1.6 KiB
Go
84 lines
1.6 KiB
Go
package config
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/spf13/viper"
|
|
"os"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
const (
|
|
envConfigPrefix = "XH_G"
|
|
)
|
|
|
|
var (
|
|
currentEnv EnvType // 当前环境变量
|
|
currentEnvOnce sync.Once
|
|
currentCommit string // 当前提交的CommitID
|
|
currentCommitOnce sync.Once
|
|
)
|
|
|
|
type EnvType string
|
|
|
|
const (
|
|
EnvDev EnvType = "dev"
|
|
EnvTest EnvType = "test"
|
|
EnvProd EnvType = "prod"
|
|
)
|
|
|
|
// LoadConfig 加载并返回应用配置
|
|
func LoadConfig[T any](configDir string, configPtr *T) (*T, 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("failed to read config: %w", err)
|
|
}
|
|
|
|
if err := v.Unmarshal(&configPtr); err != nil {
|
|
return nil, fmt.Errorf("failed to unmarshal config: %w", err)
|
|
}
|
|
|
|
marshal, _ := json.Marshal(configPtr)
|
|
fmt.Printf("Configuration loading completed: %v\n", string(marshal))
|
|
return configPtr, nil
|
|
}
|
|
|
|
// Env 获取环境
|
|
func Env() EnvType {
|
|
currentEnvOnce.Do(func() {
|
|
if v, ok := os.LookupEnv(envConfigPrefix + "_ENV"); ok {
|
|
currentEnv = EnvType(v)
|
|
} else {
|
|
currentEnv = EnvDev
|
|
}
|
|
})
|
|
return currentEnv
|
|
}
|
|
|
|
// Commit 获取提交ID
|
|
func Commit() string {
|
|
currentCommitOnce.Do(func() {
|
|
if v, ok := os.LookupEnv(envConfigPrefix + "_COMMIT"); ok {
|
|
currentCommit = v
|
|
} else {
|
|
currentCommit = "unknown"
|
|
}
|
|
})
|
|
return currentCommit
|
|
}
|