opensearch

type AggregationBucket struct {
	Key      string `json:"key"`
	DocCount uint64 `json:"doc_count"`
}

type AggregationResponse struct {
	GroupBy TermsAggregation `json:"group_by"`
}

type Config struct {
	URL   string `json:"url"`
	Index string `json:"index"`
}

type CountResponse struct {
	Count int `json:"count"`
}

type Hit struct {
	ID     string          `json:"_id"`
	Score  float64         `json:"_score"`
	Source json.RawMessage `json:"_source"`
}

type Hits struct {
	Total `json:"total"`
	Hits  []Hit `json:"hits"`
}

type OpenSearchClientRetriever interface {
	Search(ctx context.Context, index string, query []byte, pageSize int) (*SearchResponse, error)
	Count(ctx context.Context, index string, query []byte) (*CountResponse, error)
	AggregationSearch(ctx context.Context, index string, query []byte) (*AggregationResponse, error)
	IsReady(ctx context.Context) error
}

type OpenSearchSearcher struct {
	// contains filtered or unexported fields
}

type SearchResponse struct {
	Hits      `json:"hits"`
	PageToken *string `json:"last_item_id,omitempty"`
}

type TermsAggregation struct {
	DocCountErrorUpperBound uint64              `json:"doc_count_error_upper_bound"`
	SumOtherDocCount        uint64              `json:"sum_other_doc_count"`
	Buckets                 []AggregationBucket `json:"buckets"`
}

type Total struct {
	Value int `json:"value"`
}