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 - 新增单元测试覆盖缓存命中/失效/并发场景
86 lines
2.9 KiB
Go
86 lines
2.9 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: stats_repo.go
|
|
//
|
|
// Generated by this command:
|
|
//
|
|
// mockgen -source=stats_repo.go -destination=../../tests/mocks/mock_stats_repository.go -package=mocks
|
|
//
|
|
|
|
// Package mocks is a generated GoMock package.
|
|
package mocks
|
|
|
|
import (
|
|
domain "nex/backend/internal/domain"
|
|
reflect "reflect"
|
|
time "time"
|
|
|
|
gomock "go.uber.org/mock/gomock"
|
|
)
|
|
|
|
// MockStatsRepository is a mock of StatsRepository interface.
|
|
type MockStatsRepository struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockStatsRepositoryMockRecorder
|
|
isgomock struct{}
|
|
}
|
|
|
|
// MockStatsRepositoryMockRecorder is the mock recorder for MockStatsRepository.
|
|
type MockStatsRepositoryMockRecorder struct {
|
|
mock *MockStatsRepository
|
|
}
|
|
|
|
// NewMockStatsRepository creates a new mock instance.
|
|
func NewMockStatsRepository(ctrl *gomock.Controller) *MockStatsRepository {
|
|
mock := &MockStatsRepository{ctrl: ctrl}
|
|
mock.recorder = &MockStatsRepositoryMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockStatsRepository) EXPECT() *MockStatsRepositoryMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// BatchUpdate mocks base method.
|
|
func (m *MockStatsRepository) BatchUpdate(providerID, modelName string, date time.Time, delta int) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "BatchUpdate", providerID, modelName, date, delta)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// BatchUpdate indicates an expected call of BatchUpdate.
|
|
func (mr *MockStatsRepositoryMockRecorder) BatchUpdate(providerID, modelName, date, delta any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchUpdate", reflect.TypeOf((*MockStatsRepository)(nil).BatchUpdate), providerID, modelName, date, delta)
|
|
}
|
|
|
|
// Query mocks base method.
|
|
func (m *MockStatsRepository) Query(providerID, modelName string, startDate, endDate *time.Time) ([]domain.UsageStats, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Query", providerID, modelName, startDate, endDate)
|
|
ret0, _ := ret[0].([]domain.UsageStats)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Query indicates an expected call of Query.
|
|
func (mr *MockStatsRepositoryMockRecorder) Query(providerID, modelName, startDate, endDate any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Query", reflect.TypeOf((*MockStatsRepository)(nil).Query), providerID, modelName, startDate, endDate)
|
|
}
|
|
|
|
// Record mocks base method.
|
|
func (m *MockStatsRepository) Record(providerID, modelName string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Record", providerID, modelName)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Record indicates an expected call of Record.
|
|
func (mr *MockStatsRepositoryMockRecorder) Record(providerID, modelName any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Record", reflect.TypeOf((*MockStatsRepository)(nil).Record), providerID, modelName)
|
|
}
|