d4c2b723a5
- 新增 Group 图片价格配置（image_price_1k/2k/4k） - BillingService 新增 CalculateImageCost 方法 - AntigravityGatewayService 支持识别图片生成模型并按次计费 - UsageLog 新增 image_count 和 image_size 字段 - 前端分组管理支持配置图片价格（antigravity 和 gemini 平台） - 图片计费复用通用计费能力（余额检查、扣费、倍率、订阅限额）
71 lines
1.5 KiB
Go
71 lines
1.5 KiB
Go
package service
|
||
|
||
import "time"
|
||
|
||
type Group struct {
|
||
ID int64
|
||
Name string
|
||
Description string
|
||
Platform string
|
||
RateMultiplier float64
|
||
IsExclusive bool
|
||
Status string
|
||
|
||
SubscriptionType string
|
||
DailyLimitUSD *float64
|
||
WeeklyLimitUSD *float64
|
||
MonthlyLimitUSD *float64
|
||
DefaultValidityDays int
|
||
|
||
// 图片生成计费配置（antigravity 和 gemini 平台使用）
|
||
ImagePrice1K *float64
|
||
ImagePrice2K *float64
|
||
ImagePrice4K *float64
|
||
|
||
CreatedAt time.Time
|
||
UpdatedAt time.Time
|
||
|
||
AccountGroups []AccountGroup
|
||
AccountCount int64
|
||
}
|
||
|
||
func (g *Group) IsActive() bool {
|
||
return g.Status == StatusActive
|
||
}
|
||
|
||
func (g *Group) IsSubscriptionType() bool {
|
||
return g.SubscriptionType == SubscriptionTypeSubscription
|
||
}
|
||
|
||
func (g *Group) IsFreeSubscription() bool {
|
||
return g.IsSubscriptionType() && g.RateMultiplier == 0
|
||
}
|
||
|
||
func (g *Group) HasDailyLimit() bool {
|
||
return g.DailyLimitUSD != nil && *g.DailyLimitUSD > 0
|
||
}
|
||
|
||
func (g *Group) HasWeeklyLimit() bool {
|
||
return g.WeeklyLimitUSD != nil && *g.WeeklyLimitUSD > 0
|
||
}
|
||
|
||
func (g *Group) HasMonthlyLimit() bool {
|
||
return g.MonthlyLimitUSD != nil && *g.MonthlyLimitUSD > 0
|
||
}
|
||
|
||
// GetImagePrice 根据 image_size 返回对应的图片生成价格
|
||
// 如果分组未配置价格，返回 nil（调用方应使用默认值）
|
||
func (g *Group) GetImagePrice(imageSize string) *float64 {
|
||
switch imageSize {
|
||
case "1K":
|
||
return g.ImagePrice1K
|
||
case "2K":
|
||
return g.ImagePrice2K
|
||
case "4K":
|
||
return g.ImagePrice4K
|
||
default:
|
||
// 未知尺寸默认按 2K 计费
|
||
return g.ImagePrice2K
|
||
}
|
||
}
|