Skip to content

Commit

Permalink
Improve cloud_metadata
Browse files Browse the repository at this point in the history
Signed-off-by: chrismark <chrismarkou92@gmail.com>
  • Loading branch information
ChrsMark committed Jun 9, 2021
1 parent 34747af commit 8a8c6d1
Show file tree
Hide file tree
Showing 3 changed files with 359 additions and 9 deletions.
23 changes: 18 additions & 5 deletions libbeat/processors/add_cloud_metadata/add_cloud_metadata.go
Original file line number Diff line number Diff line change
Expand Up @@ -124,20 +124,35 @@ func (p *addCloudMetadata) Run(event *beat.Event) (*beat.Event, error) {
if !p.initData.overwrite {
cloudValue, _ := event.GetValue("cloud")
if cloudValue != nil {
err := p.extractECSMeta(event, meta)
if err != nil {
return nil, err
}
return event, nil
}
}

err := addMeta(event, meta)
err := p.extractECSMeta(event, meta)
if err != nil {
return nil, err
}
_, err = event.PutValue("cloud", meta)
return event, err
}

func (p *addCloudMetadata) String() string {
return "add_cloud_metadata=" + p.getMeta().String()
}

func addMeta(event *beat.Event, meta common.MapStr) error {
func (p *addCloudMetadata) extractECSMeta(event *beat.Event, meta common.MapStr) error {
// handle ECS fields first
if !p.initData.overwrite {
orchestratorValue, _ := event.GetValue("orchestrator")
if orchestratorValue != nil {
meta.Delete("orchestrator")
return nil
}
}
orchestratorFields, err := meta.GetValue("orchestrator")
if err == nil {
_, err = event.PutValue("orchestrator", orchestratorFields)
Expand All @@ -147,7 +162,5 @@ func addMeta(event *beat.Event, meta common.MapStr) error {
}
meta.Delete("orchestrator")

// put cloud.* metadata
_, err = event.PutValue("cloud", meta)
return err
return nil
}
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ var gceMetadataFetcher = provider{
} else {
out.Delete("orchestrator")
}

if project, ok := m["project"].(map[string]interface{}); ok {
s.Schema{
"project": s.Object{
Expand Down
Loading

0 comments on commit 8a8c6d1

Please sign in to comment.