6146be1474
This feature allows API Keys to have their own quota limits and expiration times, independent of the user's balance. Backend: - Add quota, quota_used, expires_at fields to api_key schema - Implement IsExpired() and IsQuotaExhausted() checks in middleware - Add ResetQuota and ClearExpiration API endpoints - Integrate quota billing in gateway handlers (OpenAI, Anthropic, Gemini) - Include quota/expiration fields in auth cache for performance - Expiration check returns 403, quota exhausted returns 429 Frontend: - Add quota and expiration inputs to key create/edit dialog - Add quick-select buttons for expiration (+7, +30, +90 days) - Add reset quota confirmation dialog - Add expires_at column to keys list - Add i18n translations for new features (en/zh) Migration: - Add 045_add_api_key_quota.sql for new columns
61 lines
2.5 KiB
Go
61 lines
2.5 KiB
Go
package service
|
|
|
|
import "time"
|
|
|
|
// APIKeyAuthSnapshot API Key 认证缓存快照（仅包含认证所需字段）
|
|
type APIKeyAuthSnapshot struct {
|
|
APIKeyID int64 `json:"api_key_id"`
|
|
UserID int64 `json:"user_id"`
|
|
GroupID *int64 `json:"group_id,omitempty"`
|
|
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)
|
|
}
|
|
|
|
// APIKeyAuthUserSnapshot 用户快照
|
|
type APIKeyAuthUserSnapshot struct {
|
|
ID int64 `json:"id"`
|
|
Status string `json:"status"`
|
|
Role string `json:"role"`
|
|
Balance float64 `json:"balance"`
|
|
Concurrency int `json:"concurrency"`
|
|
}
|
|
|
|
// APIKeyAuthGroupSnapshot 分组快照
|
|
type APIKeyAuthGroupSnapshot struct {
|
|
ID int64 `json:"id"`
|
|
Name string `json:"name"`
|
|
Platform string `json:"platform"`
|
|
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"`
|
|
ImagePrice1K *float64 `json:"image_price_1k,omitempty"`
|
|
ImagePrice2K *float64 `json:"image_price_2k,omitempty"`
|
|
ImagePrice4K *float64 `json:"image_price_4k,omitempty"`
|
|
ClaudeCodeOnly bool `json:"claude_code_only"`
|
|
FallbackGroupID *int64 `json:"fallback_group_id,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"`
|
|
}
|
|
|
|
// APIKeyAuthCacheEntry 缓存条目，支持负缓存
|
|
type APIKeyAuthCacheEntry struct {
|
|
NotFound bool `json:"not_found"`
|
|
Snapshot *APIKeyAuthSnapshot `json:"snapshot,omitempty"`
|
|
}
|