New

internal/config/config.go

@@ -0,0 +1,81 @@
+package config
+
+import (
+	"fmt"
+	"strings"
+	"time"
+
+	"github.com/spf13/viper"
+)
+
+// Config 应用所有配置的集合
+type Config struct {
+	Database      DatabaseConfig
+	Server        ServerConfig
+	Log           LogConfig
+	Redis         RedisConfig `mapstructure:"redis"`
+	SessionSecret string      `mapstructure:"session_secret"`
+	EncryptionKey string      `mapstructure:"encryption_key"`
+}
+
+// DatabaseConfig 存储数据库连接信息
+type DatabaseConfig struct {
+	DSN             string        `mapstructure:"dsn"`
+	MaxIdleConns    int           `mapstructure:"max_idle_conns"`
+	MaxOpenConns    int           `mapstructure:"max_open_conns"`
+	ConnMaxLifetime time.Duration `mapstructure:"conn_max_lifetime"`
+}
+
+// ServerConfig 存储HTTP服务器配置
+type ServerConfig struct {
+	Port string `mapstructure:"port"`
+}
+
+// LogConfig 存储日志配置
+type LogConfig struct {
+	Level      string `mapstructure:"level"       json:"level"`
+	Format     string `mapstructure:"format"      json:"format"`
+	EnableFile bool   `mapstructure:"enable_file" json:"enable_file"`
+	FilePath   string `mapstructure:"file_path"   json:"file_path"`
+}
+
+type RedisConfig struct {
+	DSN string `mapstructure:"dsn"`
+}
+
+// LoadConfig 从文件和环境变量加载配置
+func LoadConfig() (*Config, error) {
+	// 设置配置文件名和路径
+	viper.SetConfigName("config")
+	viper.SetConfigType("yaml")
+	viper.AddConfigPath(".")
+
+	// 允许从环境变量读取
+	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
+	viper.AutomaticEnv()
+
+	// 设置默认值
+	viper.SetDefault("server.port", "8080")
+	viper.SetDefault("log.level", "info")
+	viper.SetDefault("log.format", "text")
+	viper.SetDefault("log.enable_file", false)
+	viper.SetDefault("log.file_path", "logs/gemini-balancer.log")
+	viper.SetDefault("database.type", "sqlite")
+	viper.SetDefault("database.dsn", "gemini-balancer.db")
+	viper.SetDefault("database.max_idle_conns", 10)
+	viper.SetDefault("database.max_open_conns", 100)
+	viper.SetDefault("database.conn_max_lifetime", "1h")
+	viper.SetDefault("encryption_key", "")
+
+	// 读取配置文件
+	if err := viper.ReadInConfig(); err != nil {
+		if _, ok := err.(viper.ConfigFileNotFoundError); !ok {
+			return nil, fmt.Errorf("error reading config file: %w", err)
+		}
+	}
+	var cfg Config
+	if err := viper.Unmarshal(&cfg); err != nil {
+		return nil, fmt.Errorf("unable to decode config into struct: %w", err)
+	}
+	return &cfg, nil
+}