c8ffbedf75
- config: Viper YAML 配置 + 环境变量覆盖 - logs: zap 结构化日志 + trace_id 链路追踪 + 敏感信息脱敏 - db: PostgreSQL (GORM) + Redis (go-redis) 连接管理 - middleware: Trace/Logger/CORS/Recovery 四层中间件 - provider: Wire 编译时依赖注入 - main.go: 优雅启动/关闭 HTTP 服务 Made-with: Cursor
99 lines
3.1 KiB
Go
99 lines
3.1 KiB
Go
// Package config 提供应用配置管理功能
|
||
// 使用 Viper 读取 YAML 配置文件，支持环境变量覆盖
|
||
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
// Config 应用全局配置结构体
|
||
type Config struct {
|
||
Server ServerConfig `mapstructure:"server"`
|
||
Database DatabaseConfig `mapstructure:"database"`
|
||
Redis RedisConfig `mapstructure:"redis"`
|
||
JWT JWTConfig `mapstructure:"jwt"`
|
||
Log LogConfig `mapstructure:"log"`
|
||
}
|
||
|
||
// ServerConfig HTTP 服务配置
|
||
type ServerConfig struct {
|
||
Port int `mapstructure:"port"` // 监听端口
|
||
Mode string `mapstructure:"mode"` // 运行模式: debug/release
|
||
}
|
||
|
||
// DatabaseConfig PostgreSQL 数据库配置
|
||
type DatabaseConfig struct {
|
||
Host string `mapstructure:"host"`
|
||
Port int `mapstructure:"port"`
|
||
User string `mapstructure:"user"`
|
||
Password string `mapstructure:"password"`
|
||
DBName string `mapstructure:"dbname"`
|
||
SSLMode string `mapstructure:"sslmode"`
|
||
MaxIdleConns int `mapstructure:"max_idle_conns"` // 最大空闲连接数
|
||
MaxOpenConns int `mapstructure:"max_open_conns"` // 最大打开连接数
|
||
}
|
||
|
||
// DSN 生成 PostgreSQL 连接字符串
|
||
func (d *DatabaseConfig) DSN() string {
|
||
return fmt.Sprintf(
|
||
"host=%s port=%d user=%s password=%s dbname=%s sslmode=%s",
|
||
d.Host, d.Port, d.User, d.Password, d.DBName, d.SSLMode,
|
||
)
|
||
}
|
||
|
||
// RedisConfig Redis 配置
|
||
type RedisConfig struct {
|
||
Host string `mapstructure:"host"`
|
||
Port int `mapstructure:"port"`
|
||
Password string `mapstructure:"password"`
|
||
DB int `mapstructure:"db"` // 数据库编号
|
||
}
|
||
|
||
// Addr 生成 Redis 连接地址
|
||
func (r *RedisConfig) Addr() string {
|
||
return fmt.Sprintf("%s:%d", r.Host, r.Port)
|
||
}
|
||
|
||
// JWTConfig JWT 认证配置
|
||
type JWTConfig struct {
|
||
Secret string `mapstructure:"secret"` // 签名密钥
|
||
AccessExpireMin int `mapstructure:"access_expire_min"` // Access Token 有效期（分钟）
|
||
RefreshExpireDay int `mapstructure:"refresh_expire_day"` // Refresh Token 有效期（天）
|
||
Issuer string `mapstructure:"issuer"` // 签发者
|
||
}
|
||
|
||
// LogConfig 日志配置
|
||
type LogConfig struct {
|
||
Level string `mapstructure:"level"` // debug/info/warn/error
|
||
Format string `mapstructure:"format"` // text(开发)/json(生产)
|
||
OutputPath string `mapstructure:"output_path"` // stdout 或文件路径
|
||
}
|
||
|
||
// Load 加载配置文件并返回 Config 实例
|
||
// configPath 为配置文件所在目录，configName 为文件名（不含扩展名）
|
||
func Load(configPath, configName string) (*Config, error) {
|
||
v := viper.New()
|
||
v.SetConfigName(configName)
|
||
v.SetConfigType("yaml")
|
||
v.AddConfigPath(configPath)
|
||
|
||
// 环境变量覆盖：ECHOCHAT_SERVER_PORT → server.port
|
||
v.SetEnvPrefix("ECHOCHAT")
|
||
v.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
|
||
v.AutomaticEnv()
|
||
|
||
if err := v.ReadInConfig(); err != nil {
|
||
return nil, fmt.Errorf("读取配置文件失败: %w", err)
|
||
}
|
||
|
||
var cfg Config
|
||
if err := v.Unmarshal(&cfg); err != nil {
|
||
return nil, fmt.Errorf("解析配置文件失败: %w", err)
|
||
}
|
||
|
||
return &cfg, nil
|
||
}
|