elasticstream/search.go

package elasticstream

import (
	"context"
	"encoding/json"
	"fmt"
	"strings"

	"github.com/elastic/go-elasticsearch/esapi"
	"github.com/elastic/go-elasticsearch/v8"
)

// search is calling Elastic Search search API
func search(client *elasticsearch.Client, index string, offset, size *int) ([]Data, error) {
	query := fmt.Sprintf(`{
		"query": {
			"match_all": {}
		}
	}`)

	// Create the search request
	req := esapi.SearchRequest{
		Index: []string{index},
		Body:  strings.NewReader(query),
		From:  offset,
		Size:  size,
	}

	// Perform the request
	res, err := req.Do(context.Background(), client)
	if err != nil {
		return nil, fmt.Errorf("error getting response: %s", err)
	}
	defer res.Body.Close()

	if res.IsError() {
		return nil, fmt.Errorf("res.IsError() error: %s", res.String())
	}

	// Parse the response
	var result struct {
		Hits struct {
			Hits []struct {
				Source map[string]interface{} `json:"_source"`
			} `json:"hits"`
		} `json:"hits"`
	}

	if err := json.NewDecoder(res.Body).Decode(&result); err != nil {
		return nil, fmt.Errorf("error parsing the response body: %s", err)
	}

	// Collect the records
	newRecords := make([]map[string]interface{}, len(result.Hits.Hits))
	for i, hit := range result.Hits.Hits {
		newRecords[i] = hit.Source
	}

	header := map[string]string{"index": index}

	var records []Data
	for _, v := range newRecords {
		data := Data{
			Header:  header,
			Payload: v,
		}
		records = append(records, data)
	}

	// log.Println("records:", records)

	return records, nil
}