Skip to content
Snippets Groups Projects
Select Git revision
  • d25f8523fde5e74b888a932d1185c254cf7c527f
  • main default protected
  • trading_hours
  • refactor_trading_hours
  • audit_cleaning_cater_for_non_struct_fields
  • remove-info-logs
  • sl-refactor
  • 18-use-scan-for-param-values
  • 17-order-search-results
  • 4-simplify-framework-2
  • 1-http-error
  • v1.297.0
  • v1.296.0
  • v1.295.0
  • v1.294.0
  • v1.293.0
  • v1.292.0
  • v1.291.0
  • v1.290.0
  • v1.289.0
  • v1.288.0
  • v1.287.0
  • v1.286.0
  • v1.285.0
  • v1.284.0
  • v1.283.0
  • v1.282.0
  • v1.281.0
  • v1.280.0
  • v1.279.0
  • v1.278.0
31 results

writer.go

Blame
  • writer.go 4.54 KiB
    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"
    	"gitlab.com/uafrica/go-utils/logger"
    )
    
    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{},
    	}
    
    	searchConfig := opensearch.Config{
    		Transport: &http.Transport{
    			TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
    		},
    		Addresses: config.Addresses,
    		Username:  config.Username,
    		Password:  config.Password,
    	}
    	// Initialize the client with SSL/TLS enabled.
    	var err error
    	w.client, err = opensearch.NewClient(searchConfig)
    	if err != nil {
    		return nil, errors.Wrapf(err, "cannot initialize opensearch connection")
    	}
    	// Print OpenSearch version information on console.
    	logger.Debugf("Search client created with config: %+v", searchConfig)
    
    	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")
    	}
    	jsonDocStr, ok := doc.(string)
    	if !ok {
    		jsonDoc, err := json.Marshal(doc)
    		if err != nil {
    			return errors.Wrapf(err, "failed to JSON encode document")
    		}
    		jsonDocStr = string(jsonDoc)
    	}