Add a DocCount() method to the index client
This commit is contained in:
@@ -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) {
|
||||
|
||||
Reference in New Issue
Block a user