From ea694684a0193740dc8215ee1b4db98f9cd3801d Mon Sep 17 00:00:00 2001 From: Elon Cheng Date: Sun, 6 Nov 2022 07:03:15 -0600 Subject: [PATCH] Calculating the Number of Logs in opensearch (#5352) Calculating the Number of Logs Signed-off-by: chengdehao Signed-off-by: chengdehao Co-authored-by: chengdehao --- pkg/simple/client/es/versions/opensearchv1/opensearchv1.go | 1 + pkg/simple/client/es/versions/opensearchv2/opensearchv2.go | 1 + 2 files changed, 2 insertions(+) diff --git a/pkg/simple/client/es/versions/opensearchv1/opensearchv1.go b/pkg/simple/client/es/versions/opensearchv1/opensearchv1.go index f2d438855..3537284ff 100644 --- a/pkg/simple/client/es/versions/opensearchv1/opensearchv1.go +++ b/pkg/simple/client/es/versions/opensearchv1/opensearchv1.go @@ -64,6 +64,7 @@ func (o *OpenSearch) Search(indices string, body []byte, scroll bool) ([]byte, e opts := []func(*opensearchapi.SearchRequest){ o.client.Search.WithContext(context.Background()), o.client.Search.WithIndex(indices), + o.client.Search.WithRestTotalHitsAsInt(true), o.client.Search.WithIgnoreUnavailable(true), o.client.Search.WithBody(bytes.NewBuffer(body)), } diff --git a/pkg/simple/client/es/versions/opensearchv2/opensearchv2.go b/pkg/simple/client/es/versions/opensearchv2/opensearchv2.go index 5f4d15900..eceb34118 100644 --- a/pkg/simple/client/es/versions/opensearchv2/opensearchv2.go +++ b/pkg/simple/client/es/versions/opensearchv2/opensearchv2.go @@ -63,6 +63,7 @@ func New(address string, basicAuth bool, username, password, index string) (*Ope func (o *OpenSearch) Search(indices string, body []byte, scroll bool) ([]byte, error) { opts := []func(*opensearchapi.SearchRequest){ o.client.Search.WithContext(context.Background()), + o.client.Search.WithRestTotalHitsAsInt(true), o.client.Search.WithIndex(indices), o.client.Search.WithIgnoreUnavailable(true), o.client.Search.WithBody(bytes.NewBuffer(body)),