68 lines
2.9 KiB
Go
68 lines
2.9 KiB
Go
package model
|
||
|
||
import (
|
||
"time"
|
||
)
|
||
|
||
// 消费类型常量
|
||
const (
|
||
BillingTypeBalance int8 = 0 // 钱包余额
|
||
BillingTypeSubscription int8 = 1 // 订阅套餐
|
||
)
|
||
|
||
type UsageLog struct {
|
||
ID int64 `gorm:"primaryKey" json:"id"`
|
||
UserID int64 `gorm:"index;not null" json:"user_id"`
|
||
ApiKeyID int64 `gorm:"index;not null" json:"api_key_id"`
|
||
AccountID int64 `gorm:"index;not null" json:"account_id"`
|
||
RequestID string `gorm:"size:64" json:"request_id"`
|
||
Model string `gorm:"size:100;index;not null" json:"model"`
|
||
|
||
// 订阅关联（可选）
|
||
GroupID *int64 `gorm:"index" json:"group_id"`
|
||
SubscriptionID *int64 `gorm:"index" json:"subscription_id"`
|
||
|
||
// Token使用量（4类）
|
||
InputTokens int `gorm:"default:0;not null" json:"input_tokens"`
|
||
OutputTokens int `gorm:"default:0;not null" json:"output_tokens"`
|
||
CacheCreationTokens int `gorm:"default:0;not null" json:"cache_creation_tokens"`
|
||
CacheReadTokens int `gorm:"default:0;not null" json:"cache_read_tokens"`
|
||
|
||
// 详细的缓存创建分类
|
||
CacheCreation5mTokens int `gorm:"default:0;not null" json:"cache_creation_5m_tokens"`
|
||
CacheCreation1hTokens int `gorm:"default:0;not null" json:"cache_creation_1h_tokens"`
|
||
|
||
// 费用（USD）
|
||
InputCost float64 `gorm:"type:decimal(20,10);default:0;not null" json:"input_cost"`
|
||
OutputCost float64 `gorm:"type:decimal(20,10);default:0;not null" json:"output_cost"`
|
||
CacheCreationCost float64 `gorm:"type:decimal(20,10);default:0;not null" json:"cache_creation_cost"`
|
||
CacheReadCost float64 `gorm:"type:decimal(20,10);default:0;not null" json:"cache_read_cost"`
|
||
TotalCost float64 `gorm:"type:decimal(20,10);default:0;not null" json:"total_cost"` // 原始总费用
|
||
ActualCost float64 `gorm:"type:decimal(20,10);default:0;not null" json:"actual_cost"` // 实际扣除费用
|
||
RateMultiplier float64 `gorm:"type:decimal(10,4);default:1;not null" json:"rate_multiplier"` // 计费倍率
|
||
|
||
// 元数据
|
||
BillingType int8 `gorm:"type:smallint;default:0;not null" json:"billing_type"` // 0=余额 1=订阅
|
||
Stream bool `gorm:"default:false;not null" json:"stream"`
|
||
DurationMs *int `json:"duration_ms"`
|
||
FirstTokenMs *int `json:"first_token_ms"` // 首字时间（流式请求）
|
||
|
||
CreatedAt time.Time `gorm:"index;not null" json:"created_at"`
|
||
|
||
// 关联
|
||
User *User `gorm:"foreignKey:UserID" json:"user,omitempty"`
|
||
ApiKey *ApiKey `gorm:"foreignKey:ApiKeyID" json:"api_key,omitempty"`
|
||
Account *Account `gorm:"foreignKey:AccountID" json:"account,omitempty"`
|
||
Group *Group `gorm:"foreignKey:GroupID" json:"group,omitempty"`
|
||
Subscription *UserSubscription `gorm:"foreignKey:SubscriptionID" json:"subscription,omitempty"`
|
||
}
|
||
|
||
func (UsageLog) TableName() string {
|
||
return "usage_logs"
|
||
}
|
||
|
||
// TotalTokens 总token数
|
||
func (u *UsageLog) TotalTokens() int {
|
||
return u.InputTokens + u.OutputTokens + u.CacheCreationTokens + u.CacheReadTokens
|
||
}
|