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 - 新增单元测试覆盖缓存命中/失效/并发场景
17 lines
508 B
Go
17 lines
508 B
Go
package repository
|
|
|
|
import (
|
|
"time"
|
|
|
|
"nex/backend/internal/domain"
|
|
)
|
|
|
|
//go:generate go run go.uber.org/mock/mockgen -source=stats_repo.go -destination=../../tests/mocks/mock_stats_repository.go -package=mocks
|
|
|
|
// StatsRepository 统计数据仓库接口
|
|
type StatsRepository interface {
|
|
Record(providerID, modelName string) error
|
|
BatchUpdate(providerID, modelName string, date time.Time, delta int) error
|
|
Query(providerID, modelName string, startDate, endDate *time.Time) ([]domain.UsageStats, error)
|
|
}
|