lanyuanxiaoyao
915b004924
实现支持 OpenAI 和 Anthropic 双协议的统一大模型 API 网关 MVP 版本，包含： - OpenAI 和 Anthropic 协议代理 - 供应商和模型管理 - 用量统计 - 前端配置界面
80 lines
1.6 KiB
Go
80 lines
1.6 KiB
Go
package config
|
|
|
|
import (
|
|
"errors"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// RecordRequest 记录请求统计
|
|
func RecordRequest(providerID, modelName string) error {
|
|
db := GetDB()
|
|
if db == nil {
|
|
return errors.New("数据库未初始化")
|
|
}
|
|
|
|
today := time.Now().Format("2006-01-02")
|
|
todayTime, _ := time.Parse("2006-01-02", today)
|
|
|
|
// 使用事务确保并发安全
|
|
return db.Transaction(func(tx *gorm.DB) error {
|
|
var stats UsageStats
|
|
|
|
// 查找或创建统计记录
|
|
err := tx.Where("provider_id = ? AND model_name = ? AND date = ?",
|
|
providerID, modelName, todayTime).
|
|
First(&stats).Error
|
|
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
// 创建新记录
|
|
stats = UsageStats{
|
|
ProviderID: providerID,
|
|
ModelName: modelName,
|
|
RequestCount: 1,
|
|
Date: todayTime,
|
|
}
|
|
return tx.Create(&stats).Error
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 更新计数
|
|
return tx.Model(&stats).Update("request_count", gorm.Expr("request_count + 1")).Error
|
|
})
|
|
}
|
|
|
|
// GetStats 查询统计
|
|
func GetStats(providerID, modelName string, startDate, endDate *time.Time) ([]UsageStats, error) {
|
|
db := GetDB()
|
|
if db == nil {
|
|
return nil, errors.New("数据库未初始化")
|
|
}
|
|
|
|
var stats []UsageStats
|
|
query := db.Model(&UsageStats{})
|
|
|
|
if providerID != "" {
|
|
query = query.Where("provider_id = ?", providerID)
|
|
}
|
|
|
|
if modelName != "" {
|
|
query = query.Where("model_name = ?", modelName)
|
|
}
|
|
|
|
if startDate != nil {
|
|
query = query.Where("date >= ?", startDate)
|
|
}
|
|
|
|
if endDate != nil {
|
|
query = query.Where("date <= ?", endDate)
|
|
}
|
|
|
|
err := query.Order("date DESC").Find(&stats).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return stats, nil
|
|
}
|