lanyuanxiaoyao
df253559a5
- 新增 RoutingCache 组件，使用 sync.Map 缓存 Provider 和 Model - 新增 StatsBuffer 组件，使用 sync.Map + atomic.Int64 缓冲统计数据 - 扩展 StatsRepository.BatchUpdate 支持批量增量更新 - 改造 RoutingService/StatsService/ProviderService/ModelService 集成缓存 - 更新 usage-statistics spec，新增 routing-cache 和 stats-buffer spec - 新增单元测试覆盖缓存命中/失效/并发场景
90 lines
2.4 KiB
Go
90 lines
2.4 KiB
Go
package service
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"nex/backend/internal/domain"
|
|
"nex/backend/internal/repository"
|
|
)
|
|
|
|
type statsService struct {
|
|
statsRepo repository.StatsRepository
|
|
buffer *StatsBuffer
|
|
}
|
|
|
|
func NewStatsService(statsRepo repository.StatsRepository, buffer *StatsBuffer) StatsService {
|
|
return &statsService{statsRepo: statsRepo, buffer: buffer}
|
|
}
|
|
|
|
func (s *statsService) Record(providerID, modelName string) error {
|
|
s.buffer.Increment(providerID, modelName)
|
|
return nil
|
|
}
|
|
|
|
func (s *statsService) Get(providerID, modelName string, startDate, endDate *time.Time) ([]domain.UsageStats, error) {
|
|
return s.statsRepo.Query(providerID, modelName, startDate, endDate)
|
|
}
|
|
|
|
func (s *statsService) Aggregate(stats []domain.UsageStats, groupBy string) []map[string]interface{} {
|
|
switch groupBy {
|
|
case "provider":
|
|
return s.aggregateByProvider(stats)
|
|
case "model":
|
|
return s.aggregateByModel(stats)
|
|
case "date":
|
|
return s.aggregateByDate(stats)
|
|
default:
|
|
return s.aggregateByProvider(stats)
|
|
}
|
|
}
|
|
|
|
func (s *statsService) aggregateByProvider(stats []domain.UsageStats) []map[string]interface{} {
|
|
aggregated := make(map[string]int)
|
|
for _, stat := range stats {
|
|
aggregated[stat.ProviderID] += stat.RequestCount
|
|
}
|
|
result := make([]map[string]interface{}, 0, len(aggregated))
|
|
for providerID, count := range aggregated {
|
|
result = append(result, map[string]interface{}{
|
|
"provider_id": providerID,
|
|
"request_count": count,
|
|
})
|
|
}
|
|
return result
|
|
}
|
|
|
|
func (s *statsService) aggregateByModel(stats []domain.UsageStats) []map[string]interface{} {
|
|
aggregated := make(map[string]int)
|
|
for _, stat := range stats {
|
|
key := stat.ProviderID + "/" + stat.ModelName
|
|
aggregated[key] += stat.RequestCount
|
|
}
|
|
result := make([]map[string]interface{}, 0, len(aggregated))
|
|
for key, count := range aggregated {
|
|
parts := strings.SplitN(key, "/", 2)
|
|
result = append(result, map[string]interface{}{
|
|
"provider_id": parts[0],
|
|
"model_name": parts[1],
|
|
"request_count": count,
|
|
})
|
|
}
|
|
return result
|
|
}
|
|
|
|
func (s *statsService) aggregateByDate(stats []domain.UsageStats) []map[string]interface{} {
|
|
aggregated := make(map[string]int)
|
|
for _, stat := range stats {
|
|
key := stat.Date.Format("2006-01-02")
|
|
aggregated[key] += stat.RequestCount
|
|
}
|
|
result := make([]map[string]interface{}, 0, len(aggregated))
|
|
for date, count := range aggregated {
|
|
result = append(result, map[string]interface{}{
|
|
"date": date,
|
|
"request_count": count,
|
|
})
|
|
}
|
|
return result
|
|
}
|