Remote Labels Configuration

api/grpc/mpi/v1/helpers.go

@@ -6,6 +6,8 @@
 package v1
 
 import (
+	"log/slog"
+
 	"google.golang.org/protobuf/types/known/structpb"
 )
 
@@ -45,3 +47,28 @@ func ConvertToStructs(input map[string]any) ([]*structpb.Struct, error) {
 
 	return structs, nil
 }
+
+func ConvertToMap(input []*structpb.Struct) map[string]any {
+	convertedMap := make(map[string]any)
+	for _, value := range input {
+		for key, field := range value.GetFields() {
+			kind := field.GetKind()
+			switch kind.(type) {
+			case *structpb.Value_StringValue:
+				convertedMap[key] = field.GetStringValue()
+			case *structpb.Value_NumberValue:
+				convertedMap[key] = int(field.GetNumberValue())
+			case *structpb.Value_StructValue:
+				convertedMap[key] = field.GetStructValue()
+			case *structpb.Value_ListValue:
+				convertedMap[key] = field.GetListValue()
+			case *structpb.Value_BoolValue:
+				convertedMap[key] = field.GetBoolValue()
+			default:
+				slog.Warn("unknown type for map conversion", "value", kind)
+			}
+		}
+	}
+
+	return convertedMap
+}

api/grpc/mpi/v1/helpers_test.go

@@ -70,3 +70,85 @@ func TestConvertToStructs(t *testing.T) {
 		})
 	}
 }
+
+func TestConvertToMaps(t *testing.T) {
+	tests := []struct {
+		name     string
+		expected map[string]any
+		input    []*structpb.Struct
+	}{
+		{
+			name: "Test 1: Valid input with simple key-value pairs",
+			expected: map[string]any{
+				"key1": "value1",
+				"key2": 123,
+				"key3": true,
+			},
+			input: []*structpb.Struct{
+				{
+					Fields: map[string]*structpb.Value{
+						"key1": structpb.NewStringValue("value1"),
+					},
+				},
+				{
+					Fields: map[string]*structpb.Value{
+						"key2": structpb.NewNumberValue(123),
+					},
+				},
+				{
+					Fields: map[string]*structpb.Value{
+						"key3": structpb.NewBoolValue(true),
+					},
+				},
+			},
+		},
+		{
+			name: "Test 2: Valid input with simple key-value pairs, struct and list ",
+			expected: map[string]any{
+				"key1": "value1",
+				"key2": &structpb.ListValue{Values: []*structpb.Value{
+					structpb.NewStringValue("value1"),
+					structpb.NewStringValue("value2"),
+				}},
+				"key3": &structpb.Struct{
+					Fields: map[string]*structpb.Value{
+						"key1": structpb.NewStringValue("value1"),
+					},
+				},
+			},
+			input: []*structpb.Struct{
+				{
+					Fields: map[string]*structpb.Value{
+						"key1": structpb.NewStringValue("value1"),
+					},
+				},
+				{
+					Fields: map[string]*structpb.Value{
+						"key2": structpb.NewListValue(&structpb.ListValue{
+							Values: []*structpb.Value{
+								structpb.NewStringValue("value1"),
+								structpb.NewStringValue("value2"),
+							},
+						}),
+					},
+				},
+
+				{
+					Fields: map[string]*structpb.Value{
+						"key3": structpb.NewStructValue(&structpb.Struct{
+							Fields: map[string]*structpb.Value{"key1": structpb.NewStringValue("value1")},
+						}),
+					},
+				},
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := ConvertToMap(tt.input)
+
+			assert.Equal(t, tt.expected, got)
+		})
+	}
+}

internal/app.go

@@ -50,7 +50,7 @@ func (a *App) Run(ctx context.Context) error {
 			slog.String("commit", a.commit),
 		)
 
-		messagePipe := bus.NewMessagePipe(defaultMessagePipeChannelSize)
+		messagePipe := bus.NewMessagePipe(defaultMessagePipeChannelSize, agentConfig)
 		err = messagePipe.Register(defaultQueueSize, plugin.LoadPlugins(ctx, agentConfig))
 		if err != nil {
 			slog.ErrorContext(ctx, "Failed to register plugins", "error", err)