song
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
|
|
}
|
|
}
|