-
Notifications
You must be signed in to change notification settings - Fork 2.1k
/
func_convert_case.go
71 lines (54 loc) · 1.87 KB
/
func_convert_case.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
// 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"
"strings"
"github.com/iancoleman/strcase"
"github.com/open-telemetry/opentelemetry-collector-contrib/pkg/ottl"
)
type ConvertCaseArguments[K any] struct {
Target ottl.StringGetter[K]
ToCase string
}
func NewConvertCaseFactory[K any]() ottl.Factory[K] {
return ottl.NewFactory("ConvertCase", &ConvertCaseArguments[K]{}, createConvertCaseFunction[K])
}
func createConvertCaseFunction[K any](_ ottl.FunctionContext, oArgs ottl.Arguments) (ottl.ExprFunc[K], error) {
args, ok := oArgs.(*ConvertCaseArguments[K])
if !ok {
return nil, fmt.Errorf("ConvertCaseFactory args must be of type *ConvertCaseArguments[K]")
}
return convertCase(args.Target, args.ToCase)
}
func convertCase[K any](target ottl.StringGetter[K], toCase string) (ottl.ExprFunc[K], error) {
if toCase != "lower" && toCase != "upper" && toCase != "snake" && toCase != "camel" {
return nil, fmt.Errorf("invalid case: %s, allowed cases are: lower, upper, snake, camel", toCase)
}
return func(ctx context.Context, tCtx K) (any, error) {
val, err := target.Get(ctx, tCtx)
if err != nil {
return nil, err
}
if val == "" {
return val, nil
}
switch toCase {
// Convert string to lowercase (SOME_NAME -> some_name)
case "lower":
return strings.ToLower(val), nil
// Convert string to uppercase (some_name -> SOME_NAME)
case "upper":
return strings.ToUpper(val), nil
// Convert string to snake case (someName -> some_name)
case "snake":
return strcase.ToSnake(val), nil
// Convert string to camel case (some_name -> SomeName)
case "camel":
return strcase.ToCamel(val), nil
default:
return nil, fmt.Errorf("error handling unexpected case: %s", toCase)
}
}, nil
}