Add contains function

.chloggen/lkwronski.issue-30420-contains.yaml

@@ -0,0 +1,27 @@
+# Use this changelog template to create an entry for release notes.
+
+# One of 'breaking', 'deprecation', 'new_component', 'enhancement', 'bug_fix'
+change_type: enhancement
+
+# The name of the component, or a single word describing the area of concern, (e.g. filelogreceiver)
+component: pkg/ottl
+
+# A brief description of the change.  Surround your text with quotes ("") if it needs to start with a backtick (`).
+note: Add new Contains function to check item in a slice.
+
+# Mandatory: One or more tracking issues related to the change. You can use the PR number here if no issue exists.
+issues: [30420]
+
+# (Optional) One or more lines of additional information to render under the primary note.
+# These lines will be padded with 2 spaces and then inserted directly into the document.
+# Use pipe (|) for multiline entries.
+subtext:
+
+# If your change doesn't affect end users or the exported elements of any package,
+# you should instead start your pull request title with [chore] or use the "Skip Changelog" label.
+# Optional: The change log or logs in which this entry should be included.
+# e.g. '[user]' or '[user, api]'
+# Include 'user' if the change is relevant to end users.
+# Include 'api' if there is a change to a library API.
+# Default: '[user]'
+change_logs: [user]

pkg/ottl/e2e/e2e_test.go

@@ -5,6 +5,7 @@ package e2e
 
 import (
 	"context"
+	"strings"
 	"testing"
 	"time"
 
@@ -674,80 +675,106 @@ func Test_e2e_converters(t *testing.T) {
 func Test_e2e_ottl_features(t *testing.T) {
 	tests := []struct {
 		name      string
-		statement string
+		statement []string
 		want      func(tCtx ottllog.TransformContext)
 	}{
 		{
 			name:      "where clause",
-			statement: `set(attributes["test"], "pass") where body == "operationB"`,
+			statement: []string{`set(attributes["test"], "pass") where body == "operationB"`},
 			want:      func(_ ottllog.TransformContext) {},
 		},
 		{
 			name:      "reach upwards",
-			statement: `set(attributes["test"], "pass") where resource.attributes["host.name"] == "localhost"`,
+			statement: []string{`set(attributes["test"], "pass") where resource.attributes["host.name"] == "localhost"`},
 			want: func(tCtx ottllog.TransformContext) {
 				tCtx.GetLogRecord().Attributes().PutStr("test", "pass")
 			},
 		},
 		{
 			name:      "Using enums",
-			statement: `set(severity_number, SEVERITY_NUMBER_TRACE2) where severity_number == SEVERITY_NUMBER_TRACE`,
+			statement: []string{`set(severity_number, SEVERITY_NUMBER_TRACE2) where severity_number == SEVERITY_NUMBER_TRACE`},
 			want: func(tCtx ottllog.TransformContext) {
 				tCtx.GetLogRecord().SetSeverityNumber(2)
 			},
 		},
 		{
 			name:      "Using hex",
-			statement: `set(attributes["test"], "pass") where trace_id == TraceID(0x0102030405060708090a0b0c0d0e0f10)`,
+			statement: []string{`set(attributes["test"], "pass") where trace_id == TraceID(0x0102030405060708090a0b0c0d0e0f10)`},
 			want: func(tCtx ottllog.TransformContext) {
 				tCtx.GetLogRecord().Attributes().PutStr("test", "pass")
 			},
 		},
 		{
 			name:      "where clause without comparator",
-			statement: `set(attributes["test"], "pass") where IsMatch(body, "operation[AC]")`,
+			statement: []string{`set(attributes["test"], "pass") where IsMatch(body, "operation[AC]")`},
 			want: func(tCtx ottllog.TransformContext) {
 				tCtx.GetLogRecord().Attributes().PutStr("test", "pass")
 			},
 		},
 		{
 			name:      "where clause with Converter return value",
-			statement: `set(attributes["test"], "pass") where body == Concat(["operation", "A"], "")`,
+			statement: []string{`set(attributes["test"], "pass") where body == Concat(["operation", "A"], "")`},
 			want: func(tCtx ottllog.TransformContext) {
 				tCtx.GetLogRecord().Attributes().PutStr("test", "pass")
 			},
 		},
 		{
 			name:      "composing functions",
-			statement: `merge_maps(attributes, ParseJSON("{\"json_test\":\"pass\"}"), "insert") where body == "operationA"`,
+			statement: []string{`merge_maps(attributes, ParseJSON("{\"json_test\":\"pass\"}"), "insert") where body == "operationA"`},
 			want: func(tCtx ottllog.TransformContext) {
 				tCtx.GetLogRecord().Attributes().PutStr("json_test", "pass")
 			},
 		},
+		{
+			name:      "where clause with Contains return value",
+			statement: []string{`set(attributes["test"], "pass") where Contains(["hello", "world"], "hello")`},
+			want: func(tCtx ottllog.TransformContext) {
+				tCtx.GetLogRecord().Attributes().PutStr("test", "pass")
+			},
+		},
+		{
+			name: `set attribute when tag "staging" is in tags attributes slice using Contains`,
+			statement: []string{
+				`set(attributes["tags"], ["staging", "hello", "world", "work"])`,
+				`set(attributes["staging"], "true") where Contains(attributes["tags"], "staging")`,
+			},
+			want: func(tCtx ottllog.TransformContext) {
+				var tags = tCtx.GetLogRecord().Attributes().PutEmptySlice("tags")
+				tags.AppendEmpty().SetStr("staging")
+				tags.AppendEmpty().SetStr("hello")
+				tags.AppendEmpty().SetStr("world")
+				tags.AppendEmpty().SetStr("work")
+
+				tCtx.GetLogRecord().Attributes().PutStr("staging", "true")
+			},
+		},
 		{
 			name:      "complex indexing found",
-			statement: `set(attributes["test"], attributes["foo"]["bar"])`,
+			statement: []string{`set(attributes["test"], attributes["foo"]["bar"])`},
 			want: func(tCtx ottllog.TransformContext) {
 				tCtx.GetLogRecord().Attributes().PutStr("test", "pass")
 			},
 		},
 		{
 			name:      "complex indexing not found",
-			statement: `set(attributes["test"], attributes["metadata"]["uid"])`,
+			statement: []string{`set(attributes["test"], attributes["metadata"]["uid"])`},
 			want:      func(_ ottllog.TransformContext) {},
 		},
 	}
 
 	for _, tt := range tests {
-		t.Run(tt.statement, func(t *testing.T) {
+		t.Run(strings.Join(tt.statement, "\n"), func(t *testing.T) {
 			settings := componenttest.NewNopTelemetrySettings()
 			logParser, err := ottllog.NewParser(ottlfuncs.StandardFuncs[ottllog.TransformContext](), settings)
 			assert.NoError(t, err)
-			logStatements, err := logParser.ParseStatement(tt.statement)
+			logStatements, err := logParser.ParseStatements(tt.statement)
 			assert.NoError(t, err)
 
 			tCtx := constructLogTransformContext()
-			_, _, _ = logStatements.Execute(context.Background(), tCtx)
+
+			for i := range logStatements {
+				_, _, _ = logStatements[i].Execute(context.Background(), tCtx)
+			}
 
 			exTCtx := constructLogTransformContext()
 			tt.want(exTCtx)

pkg/ottl/expression.go

@@ -141,6 +141,58 @@ func (l *listGetter[K]) Get(ctx context.Context, tCtx K) (any, error) {
 	return evaluated, nil
 }
 
+// PSliceGetter is a Getter that must return a pcommon.Slice.
+type PSliceGetter[K any] interface {
+	Get(ctx context.Context, tCtx K) (pcommon.Slice, error)
+}
+
+// PStandardSliceGetter is a basic implementation of PSliceGetter
+type StandardPSliceGetter[K any] struct {
+	Getter func(ctx context.Context, tCtx K) (any, error)
+}
+
+// Get retrieves a pcommon.Slice value.
+// If the value is not a pcommon.Slice a new TypeError is returned.
+// If there is an error getting the value it will be returned.
+func (g StandardPSliceGetter[K]) Get(ctx context.Context, tCtx K) (pcommon.Slice, error) {
+	val, err := g.Getter(ctx, tCtx)
+	if err != nil {
+		return pcommon.Slice{}, fmt.Errorf("error getting value in %T: %w", g, err)
+	}
+	if val == nil {
+		return pcommon.Slice{}, TypeError("expected pcommon.Slice but got nil")
+	}
+	switch v := val.(type) {
+	case pcommon.Slice:
+		return v, nil
+	case pcommon.Value:
+		if v.Type() == pcommon.ValueTypeSlice {
+			return v.Slice(), nil
+		}
+		return pcommon.Slice{}, TypeError(fmt.Sprintf("expected pcommon.Slice but got %v", v.Type()))
+	default:
+		if reflect.ValueOf(val).Kind() == reflect.Slice {
+			return convertToSlice(val)
+		}
+		return pcommon.Slice{}, TypeError(fmt.Sprintf("expected pcommon.Slice but got %T", val))
+	}
+}
+
+// converts a generic slice to pcommon.Slice. It uses reflection to handle any slice type, creating a pcommon.Slice from it.
+func convertToSlice(val any) (pcommon.Slice, error) {
+	reflectV := reflect.ValueOf(val)
+	var output []any
+	for i := 0; i < reflectV.Len(); i++ {
+		output = append(output, reflectV.Index(i).Interface())
+	}
+	s := pcommon.NewSlice()
+	err := s.FromRaw(output)
+	if err != nil {
+		return pcommon.Slice{}, err
+	}
+	return s, nil
+}
+
 // TypeError represents that a value was not an expected type.
 type TypeError string
 

pkg/ottl/functions.go

@@ -469,6 +469,12 @@ func (p *Parser[K]) buildArg(argVal value, argType reflect.Type) (any, error) {
 			return nil, err
 		}
 		return StandardPMapGetter[K]{Getter: arg.Get}, nil
+	case strings.HasPrefix(name, "PSliceGetter"):
+		arg, err := p.newGetter(argVal)
+		if err != nil {
+			return nil, err
+		}
+		return StandardPSliceGetter[K]{Getter: arg.Get}, nil
 	case strings.HasPrefix(name, "DurationGetter"):
 		arg, err := p.newGetter(argVal)
 		if err != nil {

pkg/ottl/ottlfuncs/README.md

@@ -379,6 +379,7 @@ Available Converters:
 
 - [Base64Decode](#base64decode)
 - [Concat](#concat)
+- [Contains](#contains)
 - [ConvertCase](#convertcase)
 - [ExtractPatterns](#extractpatterns)
 - [FNV](#fnv)
@@ -456,6 +457,23 @@ Examples:
 
 - `Concat(["HTTP method is: ", attributes["http.method"]], "")`
 
+
+### Contains
+
+`Contains(target, item)`
+
+The Contains function checks if an item is present in a given slice `target`. It returns true if the `item` is found, and false otherwise.
+
+`target` is a slices of values passed as arguments. It supports primitive types.
+
+`item` is the string value to check for in the `target`.
+
+Examples:
+
+- `Contains(attributes["tags"], "staging")`
+- `Contains([1, 2, 3, 4, 5], "3")`
+- `Contains(["GET", "PUT", "POST"], "GET")`
+
 ### ConvertCase
 
 `ConvertCase(target, toCase)`

pkg/ottl/ottlfuncs/func_contains.go

@@ -0,0 +1,48 @@
+// Copyright The OpenTelemetry Authors
+// SPDX-License-Identifier: Apache-2.0
+
+package ottlfuncs // import "github.com/open-telemetry/opentelemetry-collector-contrib/pkg/ottl/ottlfuncs"
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/open-telemetry/opentelemetry-collector-contrib/pkg/ottl"
+)
+
+type ContainsArguments[K any] struct {
+	Target ottl.PSliceGetter[K]
+	Item   string
+}
+
+func NewContainsFactory[K any]() ottl.Factory[K] {
+	return ottl.NewFactory("Contains", &ContainsArguments[K]{}, createContainsFunction[K])
+}
+
+func createContainsFunction[K any](_ ottl.FunctionContext, oArgs ottl.Arguments) (ottl.ExprFunc[K], error) {
+	args, ok := oArgs.(*ContainsArguments[K])
+
+	if !ok {
+		return nil, fmt.Errorf("ContainsFactory args must be of type *ContainsArguments[K]")
+	}
+
+	return contains(args.Target, args.Item), nil
+}
+
+// nolint:errorlint
+func contains[K any](target ottl.PSliceGetter[K], item string) ottl.ExprFunc[K] {
+	return func(ctx context.Context, tCtx K) (any, error) {
+		slice, err := target.Get(ctx, tCtx)
+		if err != nil {
+			return nil, err
+		}
+
+		for i := 0; i < slice.Len(); i++ {
+			val := slice.At(i).AsString()
+			if val == item {
+				return true, nil
+			}
+		}
+		return false, nil
+	}
+}