writer.go

package search

import (
	"context"
	"crypto/tls"
	"encoding/json"
	"net/http"
	"strings"

	opensearch "github.com/opensearch-project/opensearch-go"
	opensearchapi "github.com/opensearch-project/opensearch-go/opensearchapi"
	"gitlab.com/uafrica/go-utils/errors"
)

type Writer interface {
	TimeSeries(name string, tmpl interface{}) (TimeSeries, error) //tmpl must embed TimeSeriesHeader as first unanymous field
	DelOldTimeSeries(name string, olderThanDays int) ([]string, error)
}

func New(config Config) (Writer, error) {
	if err := config.Validate(); err != nil {
		return nil, errors.Wrapf(err, "invalid config")
	}
	w := &writer{
		config:           config,
		timeSeriesByName: map[string]TimeSeries{},
	}

	// Initialize the client with SSL/TLS enabled.
	var err error
	w.client, err = opensearch.NewClient(opensearch.Config{
		Transport: &http.Transport{
			TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
		},
		Addresses: config.Addresses,
		Username:  config.Username,
		Password:  config.Password,
	})
	if err != nil {
		return nil, errors.Wrapf(err, "cannot initialize opensearch connection")
	}
	// Print OpenSearch version information on console.
	//fmt.Println(client.Info())

	w.api = opensearchapi.New(w.client)
	return w, nil
}

//implements audit.Auditor
type writer struct {
	config           Config
	client           *opensearch.Client
	api              *opensearchapi.API
	timeSeriesByName map[string]TimeSeries
}

func (writer writer) Write(indexName string, doc interface{}) error {
	if writer.client == nil {
		return errors.Errorf("writer closed")
	}
	jsonDoc, err := json.Marshal(doc)
	if err != nil {
		return errors.Wrapf(err, "failed to JSON encode document")
	}
	indexResponse, err := writer.api.Index(
		indexName,
		strings.NewReader(string(jsonDoc)),
	)
	if err != nil {
		return errors.Wrapf(err, "failed to index document")