Skip to content

Commit

Permalink
Add RemoveEventReceived processor
Browse files Browse the repository at this point in the history
  • Loading branch information
carsonip committed Sep 7, 2023
1 parent 3d5c2a7 commit 09b80b1
Show file tree
Hide file tree
Showing 2 changed files with 100 additions and 0 deletions.
42 changes: 42 additions & 0 deletions model/modelprocessor/eventreceived.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.

package modelprocessor

import (
"context"

"github.com/elastic/apm-data/model/modelpb"
)

// RemoveEventReceived is a processor that removes event.received
// field as it should not be indexed.
type RemoveEventReceived struct{}

// ProcessBatch removes event.received field for events.
func (RemoveEventReceived) ProcessBatch(ctx context.Context, b *modelpb.Batch) error {
for i := range *b {
removeEventReceived((*b)[i])
}
return nil
}

func removeEventReceived(event *modelpb.APMEvent) {
if event.Event != nil {
event.Event.Received = 0
}
}
58 changes: 58 additions & 0 deletions model/modelprocessor/eventreceived_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,58 @@
// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.

package modelprocessor_test

import (
"testing"

"github.com/elastic/apm-data/model/modelpb"
"github.com/elastic/apm-data/model/modelprocessor"
)

func TestRemoveEventReceived(t *testing.T) {
processor := modelprocessor.RemoveEventReceived{}

for _, tc := range []struct {
name string
input *modelpb.APMEvent
expected *modelpb.APMEvent
}{
{
name: "with event",
input: &modelpb.APMEvent{
Event: &modelpb.Event{
Received: 1000,
},
},
expected: &modelpb.APMEvent{
Event: &modelpb.Event{
Received: 0,
},
},
},
{
name: "without event",
input: &modelpb.APMEvent{},
expected: &modelpb.APMEvent{},
},
} {
t.Run(tc.name, func(t *testing.T) {
testProcessBatch(t, processor, tc.input, tc.expected)
})
}
}

0 comments on commit 09b80b1

Please sign in to comment.