233 lines
13 KiB
YAML
233 lines
13 KiB
YAML
{{- if or (eq .Values.executor "CeleryExecutor") (eq .Values.executor "CeleryKubernetesExecutor") }}
|
|
apiVersion: {{ include "common.capabilities.statefulset.apiVersion" . }}
|
|
kind: StatefulSet
|
|
metadata:
|
|
name: {{ printf "%s-worker" (include "common.names.fullname" .) }}
|
|
namespace: {{ .Release.Namespace | quote }}
|
|
labels: {{- include "common.labels.standard" . | nindent 4 }}
|
|
app.kubernetes.io/component: worker
|
|
{{- if .Values.commonLabels }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.commonLabels "context" $) | nindent 4 }}
|
|
{{- end }}
|
|
{{- if .Values.commonAnnotations }}
|
|
annotations: {{- include "common.tplvalues.render" (dict "value" .Values.commonAnnotations "context" $) | nindent 4 }}
|
|
{{- end }}
|
|
spec:
|
|
{{- if .Values.worker.podManagementPolicy }}
|
|
podManagementPolicy: {{ .Values.worker.podManagementPolicy | quote }}
|
|
{{- end }}
|
|
replicas: {{ .Values.worker.replicaCount }}
|
|
selector:
|
|
matchLabels: {{- include "common.labels.matchLabels" . | nindent 6 }}
|
|
app.kubernetes.io/component: worker
|
|
serviceName: {{ printf "%s-worker-hl" (include "common.names.fullname" .) | trunc 63 | trimSuffix "-" }}
|
|
{{- if .Values.worker.updateStrategy }}
|
|
updateStrategy: {{- toYaml .Values.worker.updateStrategy | nindent 4 }}
|
|
{{- end }}
|
|
template:
|
|
metadata:
|
|
annotations:
|
|
checksum/configmap: {{ include (print $.Template.BasePath "/config/configmap.yaml") . | sha256sum }}
|
|
{{- if .Values.worker.podAnnotations }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.worker.podAnnotations "context" $) | nindent 8 }}
|
|
{{- end }}
|
|
labels: {{- include "common.labels.standard" . | nindent 8 }}
|
|
app.kubernetes.io/component: worker
|
|
{{- if .Values.worker.podLabels }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.worker.podLabels "context" $) | nindent 8 }}
|
|
{{- end }}
|
|
spec:
|
|
{{- include "airflow.imagePullSecrets" . | nindent 6 }}
|
|
{{- if .Values.worker.hostAliases }}
|
|
hostAliases: {{- include "common.tplvalues.render" (dict "value" .Values.worker.hostAliases "context" $) | nindent 8 }}
|
|
{{- end }}
|
|
{{- if .Values.worker.affinity }}
|
|
affinity: {{- include "common.tplvalues.render" (dict "value" .Values.worker.affinity "context" $) | nindent 8 }}
|
|
{{- else }}
|
|
affinity:
|
|
podAffinity: {{- include "common.affinities.pods" (dict "type" .Values.worker.podAffinityPreset "component" "worker" "context" $) | nindent 10 }}
|
|
podAntiAffinity: {{- include "common.affinities.pods" (dict "type" .Values.worker.podAntiAffinityPreset "component" "worker" "context" $) | nindent 10 }}
|
|
nodeAffinity: {{- include "common.affinities.nodes" (dict "type" .Values.worker.nodeAffinityPreset.type "key" .Values.worker.nodeAffinityPreset.key "values" .Values.worker.nodeAffinityPreset.values) | nindent 10 }}
|
|
{{- end }}
|
|
{{- if .Values.worker.nodeSelector }}
|
|
nodeSelector: {{- include "common.tplvalues.render" (dict "value" .Values.worker.nodeSelector "context" $) | nindent 8 }}
|
|
{{- end }}
|
|
{{- if .Values.worker.terminationGracePeriodSeconds }}
|
|
terminationGracePeriodSeconds: {{ .Values.worker.terminationGracePeriodSeconds }}
|
|
{{- end }}
|
|
{{- if .Values.worker.tolerations }}
|
|
tolerations: {{- include "common.tplvalues.render" (dict "value" .Values.worker.tolerations "context" $) | nindent 8 }}
|
|
{{- end }}
|
|
{{- if .Values.worker.topologySpreadConstraints }}
|
|
topologySpreadConstraints: {{- include "common.tplvalues.render" (dict "value" .Values.worker.topologySpreadConstraints "context" .) | nindent 8 }}
|
|
{{- end }}
|
|
{{- if .Values.worker.priorityClassName }}
|
|
priorityClassName: {{ .Values.worker.priorityClassName | quote }}
|
|
{{- end }}
|
|
{{- if .Values.worker.schedulerName }}
|
|
schedulerName: {{ .Values.worker.schedulerName }}
|
|
{{- end }}
|
|
serviceAccountName: {{ include "airflow.serviceAccountName" . }}
|
|
{{- if .Values.worker.podSecurityContext.enabled }}
|
|
securityContext: {{- omit .Values.worker.podSecurityContext "enabled" | toYaml | nindent 8 }}
|
|
{{- end }}
|
|
initContainers:
|
|
{{- include "airflow.git.containers.clone" (dict "securityContext" .Values.worker.containerSecurityContext "context" $) | trim | nindent 8 }}
|
|
{{- if .Values.dags.existingConfigmap }}
|
|
{{- include "airflow.loadDAGsInitContainer" (dict "component" "worker" "context" . ) | trim | nindent 8 }}
|
|
{{- end }}
|
|
{{- if .Values.initContainers }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.initContainers "context" $) | trim | nindent 8 }}
|
|
{{- end }}
|
|
{{- if .Values.worker.initContainers }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.worker.initContainers "context" $) | trim | nindent 8 }}
|
|
{{- end }}
|
|
containers:
|
|
{{- include "airflow.git.containers.sync" (dict "securityContext" .Values.worker.containerSecurityContext "context" $) | trim | nindent 8 }}
|
|
- name: airflow-worker
|
|
image: {{ include "airflow.workerImage" . }}
|
|
imagePullPolicy: {{ .Values.worker.image.pullPolicy | quote }}
|
|
{{- if .Values.worker.containerSecurityContext.enabled }}
|
|
securityContext: {{- omit .Values.worker.containerSecurityContext "enabled" | toYaml | nindent 12 }}
|
|
{{- end }}
|
|
{{- if .Values.diagnosticMode.enabled }}
|
|
command: {{- include "common.tplvalues.render" (dict "value" .Values.diagnosticMode.command "context" $) | nindent 12 }}
|
|
{{- else if .Values.worker.command }}
|
|
command: {{- include "common.tplvalues.render" (dict "value" .Values.worker.command "context" $) | nindent 12 }}
|
|
{{- end }}
|
|
{{- if .Values.diagnosticMode.enabled }}
|
|
args: {{- include "common.tplvalues.render" (dict "value" .Values.diagnosticMode.args "context" $) | nindent 12 }}
|
|
{{- else if .Values.worker.args }}
|
|
args: {{- include "common.tplvalues.render" (dict "value" .Values.worker.args "context" $) | nindent 12 }}
|
|
{{- end }}
|
|
env:
|
|
{{- include "airflow.configure.airflow.common" . | nindent 12 }}
|
|
{{- include "airflow.configure.database" . | nindent 12 }}
|
|
{{- include "airflow.configure.redis" . | nindent 12 }}
|
|
- name: AIRFLOW_EXECUTOR
|
|
value: {{ include "airflow.worker.executor" . }}
|
|
- name: AIRFLOW_WEBSERVER_HOST
|
|
value: {{ include "common.names.fullname" . }}
|
|
- name: AIRFLOW_WEBSERVER_PORT_NUMBER
|
|
value: {{ .Values.service.ports.http | quote }}
|
|
{{- if .Values.extraEnvVars }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.extraEnvVars "context" $) | nindent 12 }}
|
|
{{- end }}
|
|
{{- if .Values.worker.extraEnvVars }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.worker.extraEnvVars "context" $) | nindent 12 }}
|
|
{{- end }}
|
|
{{- if or .Values.worker.extraEnvVarsCM .Values.worker.extraEnvVarsSecret .Values.worker.extraEnvVarsSecrets .Values.extraEnvVarsCM .Values.extraEnvVarsSecret .Values.extraEnvVarsSecrets }}
|
|
envFrom:
|
|
{{- if .Values.extraEnvVarsCM }}
|
|
- configMapRef:
|
|
name: {{ .Values.extraEnvVarsCM }}
|
|
{{- end }}
|
|
{{- if .Values.extraEnvVarsSecret }}
|
|
- secretRef:
|
|
name: {{ .Values.extraEnvVarsSecret }}
|
|
{{- end }}
|
|
{{- if .Values.worker.extraEnvVarsCM }}
|
|
- configMapRef:
|
|
name: {{ .Values.worker.extraEnvVarsCM }}
|
|
{{- end }}
|
|
{{- if .Values.worker.extraEnvVarsSecret }}
|
|
- secretRef:
|
|
name: {{ .Values.worker.extraEnvVarsSecret }}
|
|
{{- end }}
|
|
{{- if .Values.extraEnvVarsSecrets }}
|
|
{{- range .Values.extraEnvVarsSecrets }}
|
|
- secretRef:
|
|
name: {{ . }}
|
|
{{- end }}
|
|
{{- end }}
|
|
{{- if .Values.worker.extraEnvVarsSecrets }}
|
|
{{- range .Values.worker.extraEnvVarsSecrets }}
|
|
- secretRef:
|
|
name: {{ . }}
|
|
{{- end }}
|
|
{{- end }}
|
|
{{- end }}
|
|
ports:
|
|
- name: worker
|
|
containerPort: {{ .Values.worker.containerPorts.http }}
|
|
{{- if not .Values.diagnosticMode.enabled }}
|
|
{{- if .Values.worker.customStartupProbe }}
|
|
startupProbe: {{- include "common.tplvalues.render" (dict "value" .Values.worker.customStartupProbe "context" $) | nindent 12 }}
|
|
{{- else if .Values.worker.startupProbe.enabled }}
|
|
startupProbe: {{- include "common.tplvalues.render" (dict "value" (omit .Values.worker.startupProbe "enabled") "context" $) | nindent 12 }}
|
|
tcpSocket:
|
|
port: worker
|
|
{{- end }}
|
|
{{- if .Values.worker.customLivenessProbe }}
|
|
livenessProbe: {{- include "common.tplvalues.render" (dict "value" .Values.worker.customLivenessProbe "context" $) | nindent 12 }}
|
|
{{- else if .Values.worker.livenessProbe.enabled }}
|
|
livenessProbe: {{- include "common.tplvalues.render" (dict "value" (omit .Values.worker.livenessProbe "enabled") "context" $) | nindent 12 }}
|
|
tcpSocket:
|
|
port: worker
|
|
{{- end }}
|
|
{{- if .Values.worker.customReadinessProbe }}
|
|
readinessProbe: {{- include "common.tplvalues.render" (dict "value" .Values.worker.customReadinessProbe "context" $) | nindent 12 }}
|
|
{{- else if .Values.worker.readinessProbe.enabled }}
|
|
readinessProbe: {{- include "common.tplvalues.render" (dict "value" (omit .Values.worker.readinessProbe "enabled") "context" $) | nindent 12 }}
|
|
tcpSocket:
|
|
port: worker
|
|
{{- end }}
|
|
{{- end }}
|
|
{{- if .Values.worker.lifecycleHooks }}
|
|
lifecycle: {{- include "common.tplvalues.render" (dict "value" .Values.worker.lifecycleHooks "context" $) | nindent 12 }}
|
|
{{- end }}
|
|
{{- if .Values.worker.resources }}
|
|
resources: {{- toYaml .Values.worker.resources | nindent 12 }}
|
|
{{- end }}
|
|
volumeMounts:
|
|
{{- if .Files.Glob "files/dags/*.py" }}
|
|
- name: local-dag-files
|
|
mountPath: /opt/bitnami/airflow/dags/local
|
|
{{- end }}
|
|
{{- if .Values.dags.existingConfigmap }}
|
|
- name: external-dag-files
|
|
mountPath: /opt/bitnami/airflow/dags/external
|
|
{{- end }}
|
|
{{- if or .Values.configuration .Values.existingConfigmap }}
|
|
- name: custom-configuration-file
|
|
mountPath: /opt/bitnami/airflow/airflow.cfg
|
|
subPath: airflow.cfg
|
|
{{- end }}
|
|
{{- include "airflow.git.maincontainer.volumeMounts" . | trim | nindent 12 }}
|
|
{{- if .Values.extraVolumeMounts }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.extraVolumeMounts "context" $) | nindent 12 }}
|
|
{{- end }}
|
|
{{- if .Values.worker.extraVolumeMounts }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.worker.extraVolumeMounts "context" $) | nindent 12 }}
|
|
{{- end }}
|
|
{{- if .Values.sidecars }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.sidecars "context" $) | trim | nindent 8 }}
|
|
{{- end }}
|
|
{{- if .Values.worker.sidecars }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.worker.sidecars "context" $) | trim | nindent 8 }}
|
|
{{- end }}
|
|
volumes:
|
|
{{- if .Values.dags.existingConfigmap }}
|
|
- name: load-external-dag-files
|
|
configMap:
|
|
name: {{ tpl .Values.dags.existingConfigmap $ }}
|
|
- name: external-dag-files
|
|
emptyDir: {}
|
|
{{- end }}
|
|
{{- if or .Values.configuration .Values.existingConfigmap }}
|
|
- name: custom-configuration-file
|
|
configMap:
|
|
name: {{ include "airflow.configMapName" . }}
|
|
{{- end }}
|
|
{{- if .Values.extraVolumes }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.extraVolumes "context" $) | nindent 8 }}
|
|
{{- end }}
|
|
{{- if .Values.worker.extraVolumes }}
|
|
{{- include "common.tplvalues.render" (dict "value" .Values.worker.extraVolumes "context" $) | nindent 8 }}
|
|
{{- end }}
|
|
{{- include "airflow.git.volumes" . | trim | nindent 8 }}
|
|
{{- if .Values.worker.extraVolumeClaimTemplates }}
|
|
volumeClaimTemplates: {{- include "common.tplvalues.render" (dict "value" .Values.worker.extraVolumeClaimTemplates "context" $) | nindent 8 }}
|
|
{{- end }}
|
|
{{- end }}
|