improve handling of same port translation

workflow_builder/argo_services.go

@@ -7,6 +7,7 @@ import (
 
 	"cloud.o-forge.io/core/oc-lib/models/resource_model"
 	"cloud.o-forge.io/core/oc-lib/models/resources/workflow/graph"
+	"github.com/nwtgck/go-fakelish"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/bson/primitive"
 	"gopkg.in/yaml.v3"
@@ -60,10 +61,10 @@ func (b *ArgoBuilder) CreateService(processing graph.GraphItem) models.Service{
 	new_service := models.Service{APIVersion: "v1", 
 							Kind: "Service", 
 							Metadata: models.Metadata{
-								Name: "workflow-service" ,
+								Name: "workflow-service-" ,
 							}, 
 							Spec: models.ServiceSpec{
-								Selector: map[string]string{"app": "oc-service"},
+								Selector: map[string]string{"app": "service-" + fakelish.GenerateFakeWord(5, 8)},
 								Ports: []models.ServicePort{
 								},
 								Type: "NodePort",
@@ -115,25 +116,28 @@ func completeServicePorts(service *models.Service, processing graph.GraphItem) {
 	return
 }
 
-func (b *ArgoBuilder) createService(service models.Service, processing_name string, processing_id string) {
-	if b.Services != nil{
-		b.Services.Spec.Ports = append(b.Services.Spec.Ports, service.Spec.Ports...)
-		}else {
-			b.Services = &service
-		}
-		
-	b.addLabel(processing_name,processing_id)
+// The k8s service passed as the parameter only expose one port because it is the result of CreateService()
+// we check if this port is already exposed by a service in the workflow and we proceed to the creation of a new service OR 
+// add the port to the list of port exposed by an existing if portAlreadyExposed is false 
+func (b *ArgoBuilder) addServiceToWorkflow(service models.Service, processing_name string, processing_id string) (label string) {
+	if exposed, service_available_port := b.portAlreadyExposed(service.Spec.Ports[0].TargetPort); exposed && service_available_port != nil{
+		// The port you want to expose is already exposed by all the existing services
+		service_available_port.Spec.Ports = append(service_available_port.Spec.Ports, service.Spec.Ports...)
+		return service_available_port.Spec.Selector["app"]
+	} 
 
+	b.Services = append(b.Services, service)
+	return service.Spec.Selector["app"]	
 }
 
-func (b *ArgoBuilder) addServiceToArgo() error {
-	service_manifest, err := yaml.Marshal(b.Services)
+func (b *ArgoBuilder) addServiceToArgo(service models.Service) error {
+	service_manifest, err := yaml.Marshal(service)
 	if err != nil {
 		logger.Error().Msg("Could not marshal service manifest")
 		return err
 	}
 	
-	service_template := models.Template{Name: "workflow-service-pod", 
+	service_template := models.Template{Name: "workflow-service-pod-" + strconv.Itoa(len(b.Services)), 
 								Resource: models.ServiceResource{
 									Action: "create", 
 									SuccessCondition: "status.succeeded > 0",
@@ -145,4 +149,33 @@ func (b *ArgoBuilder) addServiceToArgo() error {
 	b.Workflow.Spec.Templates = append(b.Workflow.Spec.Templates, service_template)
 
 	return nil
+}
+
+
+func (b *ArgoBuilder) addLabel(name string, id string) {
+	argo_name := getArgoName(name,id)
+	for _, template := range b.Workflow.Spec.Templates{
+		if template.Name == argo_name{
+			template.Metadata.Labels["app"] = "service-workflow"			
+			return
+		}
+	}
+}
+
+func (b *ArgoBuilder) portAlreadyExposed(port int64) (exposed bool, service *models.Service ){
+	// For all already existing k8s services, test if the port in parameter is already exposed and returns the first service that doesn't yet expose this port 
+	for _, s := range b.Services {
+		i := 0
+		port_exposed := false
+		for !port_exposed {
+			if s.Spec.Ports[i].TargetPort == port {
+				port_exposed = true
+			}
+		}
+		if !port_exposed {
+			return false, &s
+		}
+	}
+
+	return true, nil
 }