108 lines
2.7 KiB
Go
108 lines
2.7 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
// Config 结构体定义了应用程序的所有配置项
|
|
type Config struct {
|
|
// 数据库配置
|
|
DB struct {
|
|
Host string `yaml:"host"` // 数据库主机地址
|
|
Port string `yaml:"port"` // 数据库端口
|
|
User string `yaml:"user"` // 数据库用户名
|
|
Password string `yaml:"password"` // 数据库密码
|
|
Name string `yaml:"name"` // 数据库名称
|
|
} `yaml:"db"`
|
|
|
|
Redis struct {
|
|
Host string `yaml:"host"` // Redis主机地址
|
|
Port string `yaml:"port"` // Redis端口
|
|
Password string `yaml:"password"` // Redis密码
|
|
DB string `yaml:"db"` // Redis数据库索引
|
|
} `yaml:"redis"`
|
|
|
|
// 服务器配置
|
|
Server struct {
|
|
Port string `yaml:"port"` // 服务器监听端口
|
|
Mode string `yaml:"mode"` // 运行模式
|
|
} `yaml:"server"`
|
|
|
|
Jwt struct {
|
|
Secret string `yaml:"secret"` // JWT 密钥
|
|
Expire string `yaml:"expire"` // JWT 过期时间（单位：小时）
|
|
} `yaml:"jwt"`
|
|
|
|
// mmdb 配置
|
|
Mmdb struct {
|
|
Path string `yaml:"path"` // mmdb 文件路径
|
|
} `yaml:"mmdb"`
|
|
|
|
// 日志配置
|
|
Log struct {
|
|
Path string `yaml:"path"` // 日志目录路径
|
|
} `yaml:"log"`
|
|
}
|
|
|
|
// LoadConfig 加载并返回应用程序配置
|
|
func LoadConfig() *Config {
|
|
// 首先尝试从配置文件加载
|
|
if cfg := loadConfigFromFile(); cfg != nil {
|
|
return cfg
|
|
}
|
|
|
|
// 如果没有配置文件，则使用环境变量或默认值
|
|
cfg := &Config{}
|
|
|
|
// 数据库配置
|
|
cfg.DB.Host = getEnv("DB_HOST", "localhost")
|
|
cfg.DB.Port = getEnv("DB_PORT", "3306")
|
|
cfg.DB.User = getEnv("DB_USER", "root")
|
|
cfg.DB.Password = getEnv("DB_PASSWORD", "123456")
|
|
cfg.DB.Name = getEnv("DB_NAME", "quincy")
|
|
|
|
// Redis 配置
|
|
cfg.Redis.Host = getEnv("REDIS_HOST", "localhost")
|
|
cfg.Redis.Port = getEnv("REDIS_PORT", "6379")
|
|
cfg.Redis.Password = getEnv("REDIS_PASSWORD", "")
|
|
cfg.Redis.DB = getEnv("REDIS_DB", "0")
|
|
|
|
// 服务器配置
|
|
cfg.Server.Port = getEnv("SERVER_PORT", "8080")
|
|
cfg.Server.Mode = getEnv("SERVER_MODE", "debug")
|
|
|
|
// Jwt 配置
|
|
cfg.Jwt.Secret = getEnv("JWT_SECRET", "your-secret-key")
|
|
cfg.Jwt.Expire = getEnv("JWT_EXPIRE", "24")
|
|
|
|
// mmdb 配置
|
|
cfg.Mmdb.Path = getEnv("MMDB_PATH", "./GeoLite2-City.mmdb")
|
|
|
|
cfg.Log.Path = getEnv("LOG_DIR", "./")
|
|
|
|
return cfg
|
|
}
|
|
|
|
// loadConfigFromFile 从配置文件加载配置
|
|
func loadConfigFromFile() *Config {
|
|
// 尝试加载 yaml 配置文件
|
|
if data, err := os.ReadFile("/home/app/quin_admin/default.yaml"); err == nil {
|
|
cfg := &Config{}
|
|
if err := yaml.Unmarshal(data, cfg); err == nil {
|
|
return cfg
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// getEnv 获取环境变量，如果不存在则返回默认值
|
|
func getEnv(key, defaultValue string) string {
|
|
value := os.Getenv(key)
|
|
if value == "" {
|
|
return defaultValue
|
|
}
|
|
return value
|
|
}
|