feat(chart): add scheduling controls to lago-data-rev-rec (#232)

Author: d-tw

charts/lago-data-rev-rec/templates/deployment.yaml

@@ -33,12 +33,40 @@ spec:
     replicas: {{ .Values.jobManager.replicas }}
     resource:
       {{- toYaml .Values.jobManager.resource | nindent 6 }}
+    {{- if or .Values.nodeSelector .Values.tolerations .Values.affinity }}
+    podTemplate:
+      spec:
+        {{- with .Values.nodeSelector }}
+        nodeSelector:
+          {{- toYaml . | nindent 10 }}
+        {{- end }}
+        {{- with .Values.tolerations }}
+        tolerations:
+          {{- toYaml . | nindent 10 }}
+        {{- end }}
+        {{- with .Values.affinity }}
+        affinity:
+          {{- toYaml . | nindent 10 }}
+        {{- end }}
+    {{- end }}
   taskManager:
     replicas: {{ .Values.taskManager.replicas }}
     resource:
       {{- toYaml .Values.taskManager.resource | nindent 6 }}
     podTemplate:
       spec:
+        {{- with .Values.nodeSelector }}
+        nodeSelector:
+          {{- toYaml . | nindent 10 }}
+        {{- end }}
+        {{- with .Values.tolerations }}
+        tolerations:
+          {{- toYaml . | nindent 10 }}
+        {{- end }}
+        {{- with .Values.affinity }}
+        affinity:
+          {{- toYaml . | nindent 10 }}
+        {{- end }}
         containers:
           - name: flink-main-container
             # operator injects flink config; nothing else to set here

charts/lago-data-rev-rec/templates/submitter-job.yaml

@@ -18,6 +18,18 @@ spec:
         {{- include "lago-data-rev-rec.labels" . | nindent 8 }}
         app.kubernetes.io/component: submitter
     spec:
+      {{- with .Values.nodeSelector }}
+      nodeSelector:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
+      {{- with .Values.tolerations }}
+      tolerations:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
+      {{- with .Values.affinity }}
+      affinity:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
       restartPolicy: Never
       serviceAccountName: {{ include "lago-data-rev-rec.serviceAccountName" . }}
       containers:

charts/lago-data-rev-rec/values.yaml

@@ -15,6 +15,12 @@ image:
   tag: ""
   pullPolicy: IfNotPresent
 
+# Applied to every pod this chart creates: the FlinkDeployment's JobManager
+# and TaskManagers, and the submitter Job. Empty by default.
+nodeSelector: {}
+tolerations: []
+affinity: {}
+
 # Annotations applied to the FlinkDeployment resource.
 annotations: {}