Support LB spec & coordinator-only annotations

Author: jirislav

charts/trino/templates/service-coordinator.yaml

@@ -1,4 +1,6 @@
 {{- $coordinatorJmx := merge .Values.jmx.coordinator (omit .Values.jmx "coordinator" "worker") -}}
+{{- $coordinatorSvcSpecOverride := index .Values.coordinator.service "spec" | default dict -}}
+{{- $coordinatorSvcAnnotations := deepCopy .Values.service.annotations | merge (index .Values.coordinator.service "annotations" | default dict) -}}
 apiVersion: v1
 kind: Service
 metadata:
@@ -8,9 +10,17 @@ metadata:
     {{- include "trino.labels" . | nindent 4 }}
     app.kubernetes.io/component: coordinator
   annotations:
-    {{- toYaml .Values.service.annotations | nindent 4 }}
+    {{- toYaml $coordinatorSvcAnnotations | nindent 4 }}
 spec:
-  type: {{ .Values.service.type }}
+  {{- if gt (len $coordinatorSvcSpecOverride) 0 }}
+  {{- toYaml $coordinatorSvcSpecOverride | nindent 2 }}
+  {{- end }}
+
+  {{- if not (hasKey $coordinatorSvcSpecOverride "type") }}
+  type: {{ default .Values.service.type }}
+  {{- end }}
+
+  {{- if not (hasKey $coordinatorSvcSpecOverride "ports") }}
   ports:
     - port: {{ .Values.service.port }}
       targetPort: http
@@ -43,6 +53,10 @@ spec:
       nodePort: {{ $value.nodePort }}
       {{- end }}
     {{- end }}
+  {{- end }}
+
+  {{- if not (hasKey $coordinatorSvcSpecOverride "selector") }}
   selector:
     {{- include "trino.selectorLabels" . | nindent 4 }}
     app.kubernetes.io/component: coordinator
+  {{- end }}

charts/trino/templates/service-worker.yaml

@@ -1,4 +1,7 @@
 {{- $workerJmx := merge .Values.jmx.worker (omit .Values.jmx "coordinator" "worker") -}}
+{{- $workerSvcSpecOverride := index .Values.worker.service "spec" | default dict -}}
+{{- $workerSvcAnnotations := deepCopy .Values.service.annotations | merge (index .Values.worker.service "annotations" | default dict) -}}
+{{- $workerSvcIsLoadBalancer := eq (index $workerSvcSpecOverride "type" | default "") "LoadBalancer" -}}
 apiVersion: v1
 kind: Service
 metadata:
@@ -8,9 +11,17 @@ metadata:
     {{- include "trino.labels" . | nindent 4 }}
     app.kubernetes.io/component: worker
   annotations:
-    {{- toYaml .Values.service.annotations | nindent 4 }}
+    {{- toYaml $workerSvcAnnotations | nindent 4 }}
 spec:
+  {{- if gt (len $workerSvcSpecOverride) 0 }}
+  {{- toYaml $workerSvcSpecOverride | nindent 2 }}
+  {{- end }}
+
+  {{- if not $workerSvcIsLoadBalancer }}
   clusterIP: None
+  {{- end }}
+
+  {{- if not (hasKey $workerSvcSpecOverride "ports") }}
   ports:
     - port: {{ .Values.service.port }}
       targetPort: http
@@ -31,6 +42,10 @@ spec:
       nodePort: {{ $value.nodePort }}
       {{- end }}
     {{- end }}
+  {{- end }}
+
+  {{- if not (hasKey $workerSvcSpecOverride "selector") }}
   selector:
     {{- include "trino.selectorLabels" . | nindent 4 }}
     app.kubernetes.io/component: worker
+  {{- end }}

charts/trino/values.yaml

@@ -550,6 +550,21 @@ secretMounts: []
 # ```
 
 coordinator:
+  service: {}
+  # coordinator.service -- Service overrides just for the coordinator service. It is merged with the root-level service specification. Beware that specifying spec.ports will override the dynamically generated ports for JMX exporter as well as the additionalExposedPorts.
+  # @raw
+  # Example:
+  # ```yaml
+  # annotations:
+  #   external-dns.alpha.kubernetes.io/hostname: trino.example.com
+  # spec:
+  #   type: LoadBalancer
+  #   externalTrafficPolicy: Local
+  #   loadBalancerSourceRanges:
+  #     - 1.2.3.4/32
+  #     - 10.0.0.0/8
+  # ```
+
   deployment:
     annotations: {}
     progressDeadlineSeconds: 600
@@ -735,6 +750,20 @@ coordinator:
   # ```
 
 worker:
+  service: {}
+  # worker.service -- Service overrides just for the worker service. It is merged with the root-level service specification. Beware that specifying spec.ports will override the dynamically generated ports for JMX exporter as well as the additionalExposedPorts.
+  # @raw
+  # Example:
+  # ```yaml
+  # annotations:
+  #   external-dns.alpha.kubernetes.io/hostname: trino-workers.example.com
+  # spec:
+  #   type: LoadBalancer
+  #   externalTrafficPolicy: Local
+  #   loadBalancerSourceRanges:
+  #     - 1.2.3.4/32
+  #     - 10.0.0.0/8
+  # ```
   deployment:
     annotations: {}
     progressDeadlineSeconds: 600