From ce7e8d5eedf16bbf4c7034322f409d266e9a3993 Mon Sep 17 00:00:00 2001 From: Mathieu Benoit Date: Fri, 6 Sep 2024 13:23:51 +0000 Subject: [PATCH] Create a new WorkloadServiceNameAnnotation to allow the override of the Service name --- internal/annotations.go | 5 +++-- internal/convert/workloads.go | 5 ++++- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/internal/annotations.go b/internal/annotations.go index d843246..05a334b 100644 --- a/internal/annotations.go +++ b/internal/annotations.go @@ -19,8 +19,9 @@ import ( ) const ( - AnnotationPrefix = "k8s.score.dev/" - WorkloadKindAnnotation = AnnotationPrefix + "kind" + AnnotationPrefix = "k8s.score.dev/" + WorkloadKindAnnotation = AnnotationPrefix + "kind" + WorkloadServiceNameAnnotation = AnnotationPrefix + "service-name" ) func ListAnnotations(metadata map[string]interface{}) []string { diff --git a/internal/convert/workloads.go b/internal/convert/workloads.go index dd27921..59abe22 100644 --- a/internal/convert/workloads.go +++ b/internal/convert/workloads.go @@ -280,7 +280,10 @@ func ConvertWorkload(state *project.State, workloadName string) ([]machineryMeta } func WorkloadServiceName(workloadName string) string { - return fmt.Sprintf("%s", workloadName) + if d, ok := internal.FindAnnotation(spec.Metadata, internal.WorkloadServiceNameAnnotation); ok { + return d + } + return workloadName } func buildProbe(input scoretypes.HttpProbe) coreV1.ProbeHandler {