Skip to content

Commit

Permalink
[chore][pkg/stanza] Cleanup remove operator files (#32067)
Browse files Browse the repository at this point in the history
Contributes to #32058
  • Loading branch information
djaglowski committed Apr 4, 2024
1 parent de3e1e7 commit 261b13a
Show file tree
Hide file tree
Showing 3 changed files with 42 additions and 31 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@
package remove // import "github.com/open-telemetry/opentelemetry-collector-contrib/pkg/stanza/operator/transformer/remove"

import (
"context"
"fmt"

"go.uber.org/zap"
Expand Down Expand Up @@ -55,33 +54,3 @@ func (c Config) Build(logger *zap.SugaredLogger) (operator.Operator, error) {
Field: c.Field,
}, nil
}

// Transformer is an operator that deletes a field
type Transformer struct {
helper.TransformerOperator
Field rootableField
}

// Process will process an entry with a remove transformation.
func (p *Transformer) Process(ctx context.Context, entry *entry.Entry) error {
return p.ProcessWith(ctx, entry, p.Transform)
}

// Transform will apply the remove operation to an entry
func (p *Transformer) Transform(entry *entry.Entry) error {
if p.Field.allAttributes {
entry.Attributes = nil
return nil
}

if p.Field.allResource {
entry.Resource = nil
return nil
}

_, exist := entry.Delete(p.Field.Field)
if !exist {
return fmt.Errorf("remove: field does not exist: %s", p.Field.Field.String())
}
return nil
}
42 changes: 42 additions & 0 deletions pkg/stanza/operator/transformer/remove/transformer.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0

package remove // import "github.com/open-telemetry/opentelemetry-collector-contrib/pkg/stanza/operator/transformer/remove"

import (
"context"
"fmt"

"github.com/open-telemetry/opentelemetry-collector-contrib/pkg/stanza/entry"
"github.com/open-telemetry/opentelemetry-collector-contrib/pkg/stanza/operator/helper"
)

// Transformer is an operator that deletes a field
type Transformer struct {
helper.TransformerOperator
Field rootableField
}

// Process will process an entry with a remove transformation.
func (t *Transformer) Process(ctx context.Context, entry *entry.Entry) error {
return t.ProcessWith(ctx, entry, t.Transform)
}

// Transform will apply the remove operation to an entry
func (t *Transformer) Transform(entry *entry.Entry) error {
if t.Field.allAttributes {
entry.Attributes = nil
return nil
}

if t.Field.allResource {
entry.Resource = nil
return nil
}

_, exist := entry.Delete(t.Field.Field)
if !exist {
return fmt.Errorf("remove: field does not exist: %s", t.Field.Field.String())
}
return nil
}

0 comments on commit 261b13a

Please sign in to comment.