[pkg/ottl] Add function ProfileID()

Author: rockdaboot

pkg/ottl/ottlfuncs/func_profile_id.go

@@ -0,0 +1,43 @@
+// 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"
+	"errors"
+	"fmt"
+
+	"go.opentelemetry.io/collector/pdata/pprofile"
+
+	"github.com/open-telemetry/opentelemetry-collector-contrib/pkg/ottl"
+)
+
+type ProfileIDArguments[K any] struct {
+	Bytes []byte
+}
+
+func NewProfileIDFactory[K any]() ottl.Factory[K] {
+	return ottl.NewFactory("ProfileID", &ProfileIDArguments[K]{}, createProfileIDFunction[K])
+}
+
+func createProfileIDFunction[K any](_ ottl.FunctionContext, oArgs ottl.Arguments) (ottl.ExprFunc[K], error) {
+	args, ok := oArgs.(*ProfileIDArguments[K])
+
+	if !ok {
+		return nil, errors.New("ProfileIDFactory args must be of type *ProfileIDArguments[K]")
+	}
+
+	return profileID[K](args.Bytes)
+}
+
+func profileID[K any](bytes []byte) (ottl.ExprFunc[K], error) {
+	id := pprofile.ProfileID{}
+	if len(bytes) != len(id) {
+		return nil, fmt.Errorf("profile ids must be %d bytes", len(id))
+	}
+	copy(id[:], bytes)
+	return func(context.Context, K) (any, error) {
+		return id, nil
+	}, nil
+}

pkg/ottl/ottlfuncs/func_profile_id_test.go

@@ -0,0 +1,58 @@
+// Copyright The OpenTelemetry Authors
+// SPDX-License-Identifier: Apache-2.0
+
+package ottlfuncs
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"go.opentelemetry.io/collector/pdata/pprofile"
+)
+
+func Test_profileID(t *testing.T) {
+	tests := []struct {
+		name  string
+		bytes []byte
+		want  pprofile.ProfileID
+	}{
+		{
+			name:  "create profile id",
+			bytes: []byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16},
+			want:  pprofile.ProfileID([16]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16}),
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			exprFunc, err := profileID[any](tt.bytes)
+			assert.NoError(t, err)
+			result, err := exprFunc(nil, nil)
+			assert.NoError(t, err)
+			assert.Equal(t, tt.want, result)
+		})
+	}
+}
+
+func Test_profileID_validation(t *testing.T) {
+	tests := []struct {
+		name  string
+		bytes []byte
+	}{
+		{
+			name:  "byte slice less than 16",
+			bytes: []byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15},
+		},
+		{
+			name:  "byte slice longer than 16",
+			bytes: []byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			_, err := profileID[any](tt.bytes)
+			require.Error(t, err)
+			assert.ErrorContains(t, err, "profile ids must be 16 bytes")
+		})
+	}
+}

pkg/ottl/ottlfuncs/functions.go

@@ -114,5 +114,6 @@ func converters[K any]() []ottl.Factory[K] {
 		NewYearFactory[K](),
 		NewHexFactory[K](),
 		NewSliceToMapFactory[K](),
+		NewProfileIDFactory[K](),
 	}
 }