EchoChat/backend/go-service/config/config.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"`
	Minio    MinioConfig    `mapstructure:"minio"`
}

// MinioConfig MinIO 对象存储配置
type MinioConfig struct {
	Endpoint  string `mapstructure:"endpoint"`   // MinIO 服务地址（host:port）
	AccessKey string `mapstructure:"access_key"` // 访问密钥
	SecretKey string `mapstructure:"secret_key"` // 密钥
	Bucket    string `mapstructure:"bucket"`     // 存储桶名称
	UseSSL    bool   `mapstructure:"use_ssl"`    // 是否使用 HTTPS
}

// 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 或文件路径
	File       LogFileConfig `mapstructure:"file"`        // 日志文件轮转配置
}

// LogFileConfig 日志文件轮转配置（基于 lumberjack）
type LogFileConfig struct {
	Enable     bool   `mapstructure:"enable"`      // 是否启用文件日志
	Dir        string `mapstructure:"dir"`          // 日志文件目录
	MaxSize    int    `mapstructure:"max_size"`     // 单个文件最大大小（MB），超过后自动切割
	MaxBackups int    `mapstructure:"max_backups"`  // 保留的旧日志文件最大数量
	MaxAge     int    `mapstructure:"max_age"`      // 旧日志文件保留天数
	Compress   bool   `mapstructure:"compress"`     // 是否压缩归档的旧日志文件
}

// 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
}