lanyuanxiaoyao
d92db73937
## 高优先级修复 - stats_service_impl: 使用 strings.SplitN 替代错误的索引分割 - provider_handler: 使用 errors.Is(err, gorm.ErrDuplicatedKey) 替代字符串匹配 - client: 重写 isNetworkError 使用 errors.As/Is 类型安全判断 - proxy_handler: 使用 encoding/json 标准库解析 JSON（extractModelName、isStreamRequest） ## 中优先级修复 - stats_handler: 添加 parseDateParam 辅助函数消除重复日期解析 - pkg/errors: 新增 ErrRequestCreate/Send/ResponseRead 错误类型和 WithCause 方法 - client: 使用结构化错误替代 fmt.Errorf - ConversionEngine: logger 依赖注入，替换所有 zap.L() 调用 ## 低优先级修复 - encoder: 删除 joinStrings，使用 strings.Join - adapter: 删除 modelInfoRegex 正则，使用 isModelInfoPath 字符串函数 ## 文档更新 - README.md: 添加公共库使用指南和编码规范章节 - specs: 同步 delta specs 到 main specs（error-handling、structured-logging、request-validation） ## 归档 - openspec/changes/archive/2026-04-20-refactor-backend-code-quality/
88 lines
2.4 KiB
Go
88 lines
2.4 KiB
Go
package service
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"nex/backend/internal/domain"
|
|
"nex/backend/internal/repository"
|
|
)
|
|
|
|
type statsService struct {
|
|
statsRepo repository.StatsRepository
|
|
}
|
|
|
|
func NewStatsService(statsRepo repository.StatsRepository) StatsService {
|
|
return &statsService{statsRepo: statsRepo}
|
|
}
|
|
|
|
func (s *statsService) Record(providerID, modelName string) error {
|
|
return s.statsRepo.Record(providerID, modelName)
|
|
}
|
|
|
|
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
|
|
}
|