| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152 |
- package service
- import "time"
- // APIKeyAuthSnapshot API Key 认证缓存快照（仅包含认证所需字段）
- type APIKeyAuthSnapshot struct {
- Version int `json:"version"`
- APIKeyID int64 `json:"api_key_id"`
- UserID int64 `json:"user_id"`
- GroupID *int64 `json:"group_id,omitempty"`
- Name string `json:"name"`
- Status string `json:"status"`
- IPWhitelist []string `json:"ip_whitelist,omitempty"`
- IPBlacklist []string `json:"ip_blacklist,omitempty"`
- User APIKeyAuthUserSnapshot `json:"user"`
- Group *APIKeyAuthGroupSnapshot `json:"group,omitempty"`
- // Quota fields for API Key independent quota feature
- Quota float64 `json:"quota"` // Quota limit in USD (0 = unlimited)
- QuotaUsed float64 `json:"quota_used"` // Used quota amount
- // Expiration field for API Key expiration feature
- ExpiresAt *time.Time `json:"expires_at,omitempty"` // Expiration time (nil = never expires)
- // Rate limit configuration (only limits, not usage - usage read from Redis at check time)
- RateLimit5h float64 `json:"rate_limit_5h"`
- RateLimit1d float64 `json:"rate_limit_1d"`
- RateLimit7d float64 `json:"rate_limit_7d"`
- }
- // APIKeyAuthUserSnapshot 用户快照
- type APIKeyAuthUserSnapshot struct {
- ID int64 `json:"id"`
- Status string `json:"status"`
- Role string `json:"role"`
- Balance float64 `json:"balance"`
- Concurrency int `json:"concurrency"`
- AllowedGroups []int64 `json:"allowed_groups,omitempty"`
- // Balance notification fields (required for CheckBalanceAfterDeduction)
- Email string `json:"email"`
- Username string `json:"username"`
- BalanceNotifyEnabled bool `json:"balance_notify_enabled"`
- RestrictPublicGroups bool `json:"restrict_public_groups"`
- BalanceNotifyThresholdType string `json:"balance_notify_threshold_type"`
- BalanceNotifyThreshold *float64 `json:"balance_notify_threshold,omitempty"`
- BalanceNotifyExtraEmails []NotifyEmailEntry `json:"balance_notify_extra_emails,omitempty"`
- TotalRecharged float64 `json:"total_recharged"`
- // RPMLimit 用户级每分钟请求数上限（0 = 不限制）；用于 billing_cache_service.checkRPM 兜底判断。
- RPMLimit int `json:"rpm_limit"`
- // UserGroupRPMOverride 该 API Key 对应的 (user, group) 专属 RPM 覆盖值。
- // nil = 无 override（回退到 group/user 级）；0 = 不限流；>0 = 专属上限。
- UserGroupRPMOverride *int `json:"user_group_rpm_override,omitempty"`
- }
- // APIKeyAuthGroupSnapshot 分组快照
- type APIKeyAuthGroupSnapshot struct {
- SecurityPolicyEnabled bool `json:"security_policy_enabled"`
- SecurityPolicyMode string `json:"security_policy_mode"`
- SecurityPolicyEmailEnabled bool `json:"security_policy_email_enabled"`
- ID int64 `json:"id"`
- Name string `json:"name"`
- Platform string `json:"platform"`
- IsExclusive bool `json:"is_exclusive"`
- Status string `json:"status"`
- SubscriptionType string `json:"subscription_type"`
- RateMultiplier float64 `json:"rate_multiplier"`
- DailyLimitUSD *float64 `json:"daily_limit_usd,omitempty"`
- WeeklyLimitUSD *float64 `json:"weekly_limit_usd,omitempty"`
- MonthlyLimitUSD *float64 `json:"monthly_limit_usd,omitempty"`
- AllowImageGeneration bool `json:"allow_image_generation"`
- AllowBatchImageGeneration bool `json:"allow_batch_image_generation"`
- ImageRateIndependent bool `json:"image_rate_independent"`
- ImageRateMultiplier float64 `json:"image_rate_multiplier"`
- ImagePrice1K *float64 `json:"image_price_1k,omitempty"`
- ImagePrice2K *float64 `json:"image_price_2k,omitempty"`
- ImagePrice4K *float64 `json:"image_price_4k,omitempty"`
- VideoRateIndependent bool `json:"video_rate_independent"`
- VideoRateMultiplier float64 `json:"video_rate_multiplier"`
- VideoPrice480P *float64 `json:"video_price_480p,omitempty"`
- VideoPrice720P *float64 `json:"video_price_720p,omitempty"`
- VideoPrice1080P *float64 `json:"video_price_1080p,omitempty"`
- VideoModelPrices map[string]map[string]float64 `json:"video_model_prices,omitempty"`
- WebSearchPricePerCall *float64 `json:"web_search_price_per_call,omitempty"`
- SearchPricePer1k *float64 `json:"search_price_per_1k,omitempty"`
- AudioRealtimePricePerMin *float64 `json:"audio_realtime_price_per_min,omitempty"`
- AudioTTSPricePerMillionChars *float64 `json:"audio_tts_price_per_million_chars,omitempty"`
- AudioSTTPricePerHour *float64 `json:"audio_stt_price_per_hour,omitempty"`
- LongContextPricingEnabled bool `json:"long_context_pricing_enabled"`
- ModelPricing []ChannelModelPricing `json:"model_pricing,omitempty"`
- ClaudeCodeOnly bool `json:"claude_code_only"`
- FallbackGroupID *int64 `json:"fallback_group_id,omitempty"`
- FallbackGroupIDOnInvalidRequest *int64 `json:"fallback_group_id_on_invalid_request,omitempty"`
- // Model routing is used by gateway account selection, so it must be part of auth cache snapshot.
- // Only anthropic groups use these fields; others may leave them empty.
- ModelRouting map[string][]int64 `json:"model_routing,omitempty"`
- ModelRoutingEnabled bool `json:"model_routing_enabled"`
- MCPXMLInject bool `json:"mcp_xml_inject"`
- // 支持的模型系列（仅 antigravity 平台使用）
- SupportedModelScopes []string `json:"supported_model_scopes,omitempty"`
- // OpenAI Messages 调度配置（仅 openai 平台使用）
- AllowMessagesDispatch bool `json:"allow_messages_dispatch"`
- AllowLive bool `json:"allow_live"`
- ForceOpenAIFast bool `json:"force_openai_fast"`
- FreeOpenAIFast bool `json:"free_openai_fast"`
- DefaultMappedModel string `json:"default_mapped_model,omitempty"`
- MessagesDispatchModelConfig OpenAIMessagesDispatchModelConfig `json:"messages_dispatch_model_config,omitempty"`
- ModelAllowlist GroupModelAllowlist `json:"model_allowlist,omitempty"`
- // CodexModelsManifestConfig 与 ModelAllowlist 一样在认证快照分组里透传，
- // Codex /models handler 直接读认证分组对象。
- CodexModelsManifestConfig GroupCodexModelsManifestConfig `json:"codex_models_manifest_config,omitempty"`
- // RPMLimit 分组级每分钟请求数上限（0 = 不限制）；用于 billing_cache_service.checkRPM 级联判断。
- RPMLimit int `json:"rpm_limit"`
- // MaxReasoningEffort Anthropic/OpenAI 请求的推理强度上限，空字符串表示不限制。
- MaxReasoningEffort string `json:"max_reasoning_effort,omitempty"`
- // MaxReasoningEffortOverLimit 超过上限时的访问控制：downgrade（默认）或 deny。
- MaxReasoningEffortOverLimit string `json:"max_reasoning_effort_over_limit,omitempty"`
- // ReasoningEffortMappings rewrites explicit effort values before the ceiling.
- ReasoningEffortMappings []ReasoningEffortMapping `json:"reasoning_effort_mappings"`
- // 高峰时段倍率：PeakRateEnabled 为 true 且请求时刻处于 [PeakStart, PeakEnd) 时，
- // token 计费倍率额外乘以 PeakRateMultiplier（详见 Group.PeakMultiplierAt）。
- // 必须随快照缓存，否则扣费路径拿到的 apiKey.Group 缺字段、高峰倍率失效。
- PeakRateEnabled bool `json:"peak_rate_enabled"`
- PeakStart string `json:"peak_start"`
- PeakEnd string `json:"peak_end"`
- PeakRateMultiplier float64 `json:"peak_rate_multiplier"`
- // 分组利润控制：调度准入门在直连热路径上读的就是这份快照——门解析
- // （resolveOpenAIProfitControlGate / resolveProfitControlGroup）优先取
- // 认证中间件放入 ctx 的 Group，而它正是本快照物化出来的对象，生产绝大
- // 多数流量走的都是这条路；只有 composite/模型路由等被调度分组与认证分组
- // 不一致时才回源 schedulerSnapshot。
- // 因此这三个字段与 GetByKeyForAuth 的投影都不得删减：漏掉任何一个，
- // 门会拿到零值 ProfitControlEnabled=false 而静默失效（有集成测试兜底）。
- ProfitControlEnabled bool `json:"profit_control_enabled"`
- ProfitMinMargin float64 `json:"profit_min_margin"`
- ProfitSafetyBuffer float64 `json:"profit_safety_buffer"`
- }
- // APIKeyAuthCacheEntry 缓存条目，支持负缓存
- type APIKeyAuthCacheEntry struct {
- NotFound bool `json:"not_found"`
- Snapshot *APIKeyAuthSnapshot `json:"snapshot,omitempty"`
- }
|
