add limit

internal/config/config.go

@@ -17,7 +17,8 @@ type Config struct {
 	Email       EmailConfig       `mapstructure:"email"`
 	Storage     StorageConfig     `mapstructure:"storage"`
 	OCR         OCRConfig         `mapstructure:"ocr"`
-	RateLimit   RateLimitConfig   `mapstructure:"ratelimit"`
+	RateLimit       RateLimitConfig       `mapstructure:"ratelimit"`
+	DailyRateLimit  DailyRateLimitConfig  `mapstructure:"daily_ratelimit"`
 	Monitoring  MonitoringConfig  `mapstructure:"monitoring"`
 	Health      HealthConfig      `mapstructure:"health"`
 	Resilience  ResilienceConfig  `mapstructure:"resilience"`
@@ -118,6 +119,27 @@ type RateLimitConfig struct {
 	Burst    int           `mapstructure:"burst"`
 }
 
+// DailyRateLimitConfig 每日限流配置
+type DailyRateLimitConfig struct {
+	MaxRequestsPerDay int           `mapstructure:"max_requests_per_day"` // 每日最大请求次数
+	MaxRequestsPerIP  int           `mapstructure:"max_requests_per_ip"`  // 每个IP每日最大请求次数
+	KeyPrefix         string        `mapstructure:"key_prefix"`           // Redis键前缀
+	TTL              time.Duration `mapstructure:"ttl"`                  // 键过期时间
+	// 新增安全配置
+	EnableIPWhitelist bool          `mapstructure:"enable_ip_whitelist"` // 是否启用IP白名单
+	IPWhitelist       []string      `mapstructure:"ip_whitelist"`        // IP白名单
+	EnableIPBlacklist bool          `mapstructure:"enable_ip_blacklist"` // 是否启用IP黑名单
+	IPBlacklist       []string      `mapstructure:"ip_blacklist"`        // IP黑名单
+	EnableUserAgent   bool          `mapstructure:"enable_user_agent"`   // 是否检查User-Agent
+	BlockedUserAgents []string      `mapstructure:"blocked_user_agents"` // 被阻止的User-Agent
+	EnableReferer     bool          `mapstructure:"enable_referer"`      // 是否检查Referer
+	AllowedReferers   []string      `mapstructure:"allowed_referers"`    // 允许的Referer
+	EnableGeoBlock    bool          `mapstructure:"enable_geo_block"`    // 是否启用地理位置阻止
+	BlockedCountries  []string      `mapstructure:"blocked_countries"`   // 被阻止的国家/地区
+	EnableProxyCheck  bool          `mapstructure:"enable_proxy_check"`  // 是否检查代理
+	MaxConcurrent     int           `mapstructure:"max_concurrent"`      // 最大并发请求数
+}
+
 // MonitoringConfig 监控配置
 type MonitoringConfig struct {
 	MetricsEnabled  bool    `mapstructure:"metrics_enabled"`