gRPC bugfix for Dirigent

Signed-off-by: Lazar Cvetković <[email protected]>

pkg/common/trace_types.go

@@ -79,6 +79,8 @@ type DirigentMetadata struct {
 	ScalingLowerBound   int    `csv:"ScalingLowerBound"`
 	IterationMultiplier int    `csv:"IterationMultiplier"`
 	IOPercentage        int    `csv:"IOPercentage"`
+	EnvVars             string `csv:"EnvVars"`
+	ProgramArgs         string `csv:"ProgramArgs"`
 }
 
 type Function struct {

pkg/driver/clients/grpc_client.go

@@ -68,6 +68,9 @@ func (i *grpcInvoker) Invoke(function *common.Function, runtimeSpec *common.Runt
 
 	var dialOptions []grpc.DialOption
 	dialOptions = append(dialOptions, grpc.WithTransportCredentials(insecure.NewCredentials()))
+	if strings.Contains(strings.ToLower(i.cfg.Platform), "dirigent") {
+		dialOptions = append(dialOptions, grpc.WithAuthority(function.Name)) // Dirigent specific
+	}
 	if i.cfg.EnableZipkinTracing {
 		dialOptions = append(dialOptions, grpc.WithStatsHandler(otelgrpc.NewClientHandler()))
 	}

pkg/driver/deployment/dirigent.go

@@ -53,6 +53,18 @@ var registrationClient = &http.Client{
 	},
 }
 
+const CSVColumnDelimiter = " "
+
+func prepareEnvVars(data string) []string {
+	// FORMAT: arg1=value1 arg2=value2
+	return strings.Split(data, CSVColumnDelimiter)
+}
+
+func prepareProgramArgs(data string) []string {
+	// FORMAT: arg1 val1 arg2 val2
+	return strings.Split(data, CSVColumnDelimiter)
+}
+
 func deployDirigent(function *common.Function, controlPlaneAddress string, busyLoopOnColdStart bool) {
 	metadata := function.DirigentMetadata
 
@@ -68,6 +80,8 @@ func deployDirigent(function *common.Function, controlPlaneAddress string, busyL
 		"scaling_lower_bound": {strconv.Itoa(metadata.ScalingLowerBound)},
 		"requested_cpu":       {strconv.Itoa(function.CPURequestsMilli)},
 		"requested_memory":    {strconv.Itoa(function.MemoryRequestsMiB)},
+		"env_vars":            prepareEnvVars(metadata.EnvVars),
+		"program_args":        prepareProgramArgs(metadata.ProgramArgs),
 	}
 
 	if busyLoopOnColdStart {