dual check v0.1

server/configs/config.go

@@ -0,0 +1,106 @@
+package configs
+
+import (
+	"fmt"
+	"os"
+	"strings"
+	"time"
+
+	"github.com/spf13/viper"
+)
+
+type Config struct {
+	Server    ServerConfig    `mapstructure:"server"`
+	Redis     RedisConfig     `mapstructure:"redis"`
+	RateLimit RateLimitConfig `mapstructure:"rate_limit"`
+	Session   SessionConfig   `mapstructure:"session"`
+	Log       LogConfig       `mapstructure:"log"`
+	CORS      CORSConfig      `mapstructure:"cors"`
+}
+
+type ServerConfig struct {
+	Host         string        `mapstructure:"host"`
+	Port         int           `mapstructure:"port"`
+	ReadTimeout  time.Duration `mapstructure:"read_timeout"`
+	WriteTimeout time.Duration `mapstructure:"write_timeout"`
+	IdleTimeout  time.Duration `mapstructure:"idle_timeout"`
+}
+
+type RedisConfig struct {
+	Addr     string `mapstructure:"addr"`
+	Password string `mapstructure:"password"`
+	DB       int    `mapstructure:"db"`
+	PoolSize int    `mapstructure:"pool_size"`
+}
+
+type RateLimitConfig struct {
+	GlobalRPS  float64 `mapstructure:"global_rps"`
+	PerIPRPS   float64 `mapstructure:"per_ip_rps"`
+	PerIPBurst int     `mapstructure:"burst"`
+}
+
+type SessionConfig struct {
+	TTL           int `mapstructure:"ttl"`
+	MaxTextLength int `mapstructure:"max_text_length"`
+}
+
+type LogConfig struct {
+	Level  string `mapstructure:"level"`
+	Format string `mapstructure:"format"`
+}
+
+type CORSConfig struct {
+	AllowedOrigins []string `mapstructure:"allowed_origins"`
+}
+
+func Load(configPath ...string) (*Config, error) {
+	v := viper.New()
+
+	if len(configPath) > 0 && configPath[0] != "" {
+		v.SetConfigFile(configPath[0])
+	} else {
+		v.SetConfigName("config")
+		v.SetConfigType("yaml")
+		v.AddConfigPath("configs/")
+		v.AddConfigPath("./configs/")
+		v.AddConfigPath(".")
+	}
+
+	v.SetDefault("server.host", "0.0.0.0")
+	v.SetDefault("server.port", 8080)
+	v.SetDefault("server.read_timeout", "15s")
+	v.SetDefault("server.write_timeout", "15s")
+	v.SetDefault("server.idle_timeout", "60s")
+	v.SetDefault("redis.addr", "localhost:6379")
+	v.SetDefault("redis.db", 0)
+	v.SetDefault("redis.pool_size", 100)
+	v.SetDefault("rate_limit.global_rps", 1000)
+	v.SetDefault("rate_limit.per_ip_rps", 50)
+	v.SetDefault("rate_limit.burst", 100)
+	v.SetDefault("session.ttl", 900)
+	v.SetDefault("session.max_text_length", 10000)
+	v.SetDefault("log.level", "info")
+	v.SetDefault("log.format", "json")
+
+	// 环境变量覆盖（前缀 DUALCHECK_）
+	// mapstructure 的 "." 和 "_" 都需要替换成 "_"，并转大写
+	v.SetEnvPrefix("DUALCHECK")
+	v.SetEnvKeyReplacer(strings.NewReplacer(".", "_", "-", "_"))
+	v.AutomaticEnv()
+
+	// 尝试读取配置文件（文件不存在不报错，用默认值+环境变量）
+	if err := v.ReadInConfig(); err != nil {
+		if _, ok := err.(viper.ConfigFileNotFoundError); !ok {
+			fmt.Fprintf(os.Stderr, "[config] warning: config file error: %v\n", err)
+		} else {
+			fmt.Fprintf(os.Stderr, "[config] config file not found, using defaults + env vars\n")
+		}
+	}
+
+	var cfg Config
+	if err := v.Unmarshal(&cfg); err != nil {
+		return nil, fmt.Errorf("unmarshal config: %w", err)
+	}
+
+	return &cfg, nil
+}

server/configs/config.yaml

@@ -0,0 +1,35 @@
+# === DualCheck Server Config ===
+# 生产环境敏感配置请用环境变量覆盖，不要把真实值提交到 git
+# 环境变量前缀：DUALCHECK_（如 DUALCHECK_REDIS_ADDR）
+
+server:
+  host: "0.0.0.0"
+  port: 8080
+  read_timeout: 15s
+  write_timeout: 15s
+  idle_timeout: 60s
+
+redis:
+  addr: "192.168.64.7:6379"
+  password: "123456"
+  db: 0
+  pool_size: 100
+
+rate_limit:
+  global_rps: 1000
+  per_ip_rps: 50
+  burst: 100
+
+session:
+  ttl: 900                  # 会话过期时间（秒），默认 15 分钟
+  max_text_length: 10000    # 单条文本最大长度
+
+log:
+  level: "info"             # debug / info / warn / error
+  format: "json"            # json / console
+
+cors:
+  allowed_origins:
+    - "http://localhost:3000"
+    - "http://localhost:8080"
+    - "http://localhost:5173"