Add a DocCount() method to the index client

This commit is contained in:
André Duffeck
2022-04-27 09:48:26 +02:00
parent d5084eea84
commit 827885abad
3 changed files with 48 additions and 4 deletions
@@ -66,6 +66,11 @@ func New(bleveIndex bleve.Index) (*Index, error) {
}, nil
}
// DocCount returns the number of elemenst in the index
func (i *Index) DocCount() (uint64, error) {
return i.bleveIndex.DocCount()
}
// Add adds a new entity to the Index
func (i *Index) Add(ref *sprovider.Reference, ri *sprovider.ResourceInfo) error {
entity := toEntity(ref, ri)
@@ -30,6 +30,41 @@ func (_m *IndexClient) Add(ref *providerv1beta1.Reference, ri *providerv1beta1.R
return r0
}
// DocCount provides a mock function with given fields:
func (_m *IndexClient) DocCount() (uint64, error) {
ret := _m.Called()
var r0 uint64
if rf, ok := ret.Get(0).(func() uint64); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(uint64)
}
var r1 error
if rf, ok := ret.Get(1).(func() error); ok {
r1 = rf()
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// Remove provides a mock function with given fields: ri
func (_m *IndexClient) Remove(ri *providerv1beta1.ResourceInfo) error {
ret := _m.Called(ri)
var r0 error
if rf, ok := ret.Get(0).(func(*providerv1beta1.ResourceInfo) error); ok {
r0 = rf(ri)
} else {
r0 = ret.Error(0)
}
return r0
}
// Search provides a mock function with given fields: ctx, req
func (_m *IndexClient) Search(ctx context.Context, req *v0.SearchIndexRequest) (*v0.SearchIndexResponse, error) {
ret := _m.Called(ctx, req)
@@ -66,11 +66,15 @@ func New(gwClient gateway.GatewayAPIClient, indexClient search.IndexClient, mach
}
}()
return &Provider{
gwClient: gwClient,
indexClient: indexClient,
machineAuthAPIKey: machineAuthAPIKey,
return p
}
func (p *Provider) logDocCount() {
c, err := p.indexClient.DocCount()
if err != nil {
p.logger.Error().Err(err).Msg("error getting document count from the index")
}
p.logger.Debug().Interface("count", c).Msg("new document count")
}
func (p *Provider) Search(ctx context.Context, req *searchsvc.SearchRequest) (*searchsvc.SearchResponse, error) {