This repository has been archived by the owner on Aug 31, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 353
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Willem Borgesius
authored and
Willem Borgesius
committed
Dec 16, 2021
1 parent
43dde85
commit 99860f1
Showing
7 changed files
with
195 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,145 @@ | ||
| package sinks | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
| opensearch "github.com/opensearch-project/opensearch-go" | ||
| "github.com/opsgenie/kubernetes-event-exporter/pkg/kube" | ||
| "github.com/rs/zerolog/log" | ||
| "io/ioutil" | ||
| "net/http" | ||
| "regexp" | ||
| "strings" | ||
| "time" | ||
| ) | ||
|
|
||
| type OpenSearchConfig struct { | ||
| // Connection specific | ||
| Hosts []string `yaml:"hosts"` | ||
| Username string `yaml:"username"` | ||
| Password string `yaml:"password"` | ||
| // Indexing preferences | ||
| UseEventID bool `yaml:"useEventID"` | ||
| // DeDot all labels and annotations in the event. For both the event and the involvedObject | ||
| DeDot bool `yaml:"deDot"` | ||
| Index string `yaml:"index"` | ||
| IndexFormat string `yaml:"indexFormat"` | ||
| Type string `yaml:"type"` | ||
| TLS TLS `yaml:"tls"` | ||
| Layout map[string]interface{} `yaml:"layout"` | ||
| } | ||
|
|
||
| func NewOpenSearch(cfg *OpenSearchConfig) (*OpenSearch, error) { | ||
|
|
||
| tlsClientConfig, err := setupTLS(&cfg.TLS) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to setup TLS: %w", err) | ||
| } | ||
|
|
||
| client, err := opensearch.NewClient(opensearch.Config{ | ||
| Addresses: cfg.Hosts, | ||
| Username: cfg.Username, | ||
| Password: cfg.Password, | ||
| Transport: &http.Transport{ | ||
| TLSClientConfig: tlsClientConfig, | ||
| }, | ||
| }) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return &OpenSearch{ | ||
| client: client, | ||
| cfg: cfg, | ||
| }, nil | ||
| } | ||
|
|
||
| type OpenSearch struct { | ||
| client *opensearch.Client | ||
| cfg *OpenSearchConfig | ||
| } | ||
|
|
||
| var osRegex = regexp.MustCompile(`(?s){(.*)}`) | ||
|
|
||
| func osFormatIndexName(pattern string, when time.Time) string { | ||
| m := osRegex.FindAllStringSubmatchIndex(pattern, -1) | ||
| current := 0 | ||
| var builder strings.Builder | ||
|
|
||
| for i := 0; i < len(m); i++ { | ||
| pair := m[i] | ||
|
|
||
| builder.WriteString(pattern[current:pair[0]]) | ||
| builder.WriteString(when.Format(pattern[pair[0]+1 : pair[1]-1])) | ||
| current = pair[1] | ||
| } | ||
|
|
||
| builder.WriteString(pattern[current:]) | ||
|
|
||
| return builder.String() | ||
| } | ||
|
|
||
| func (e *OpenSearch) Send(ctx context.Context, ev *kube.EnhancedEvent) error { | ||
| var toSend []byte | ||
|
|
||
| if e.cfg.DeDot { | ||
| de := ev.DeDot() | ||
| ev = &de | ||
| } | ||
| if e.cfg.Layout != nil { | ||
| res, err := convertLayoutTemplate(e.cfg.Layout, ev) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| toSend, err = json.Marshal(res) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| } else { | ||
| toSend = ev.ToJSON() | ||
| } | ||
|
|
||
| var index string | ||
| if len(e.cfg.IndexFormat) > 0 { | ||
| now := time.Now() | ||
| index = osFormatIndexName(e.cfg.IndexFormat, now) | ||
| } else { | ||
| index = e.cfg.Index | ||
| } | ||
|
|
||
| req := esapi.IndexRequest{ | ||
| Body: bytes.NewBuffer(toSend), | ||
| Index: index, | ||
| } | ||
|
|
||
| // This should not be used for clusters with ES8.0+. | ||
| if len(e.cfg.Type) > 0 { | ||
| req.DocumentType = e.cfg.Type | ||
| } | ||
|
|
||
| if e.cfg.UseEventID { | ||
| req.DocumentID = string(ev.UID) | ||
| } | ||
|
|
||
| resp, err := req.Do(ctx, e.client) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| defer resp.Body.Close() | ||
| if resp.StatusCode > 399 { | ||
| rb, err := ioutil.ReadAll(resp.Body) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| log.Error().Msgf("Indexing failed: %s", string(rb)) | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func (e *OpenSearch) Close() { | ||
| // No-op | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters