Adding option for defining extra Init Containers

Signed-off-by: Nikhil Jiju <[email protected]>
PrefectHQ · Oct 23, 2024 · dae31eb · dae31eb
1 parent 0393728
commit dae31eb
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/charts/prefect-worker/templates/deployment.yaml b/charts/prefect-worker/templates/deployment.yaml
@@ -54,8 +54,9 @@ spec:
       {{- if .Values.worker.priorityClassName }}
       priorityClassName: {{ .Values.worker.priorityClassName }}
       {{- end }}
-      {{- if include "worker.baseJobTemplateName" . }}
+      {{- if or (include "worker.baseJobTemplateName" .) (.Values.worker.initContainer.extraContainers) }}
       initContainers:
+        {{- if include "worker.baseJobTemplateName" . }}
         - name: sync-base-job-template
           image: "{{ .Values.worker.image.repository }}:{{ .Values.worker.image.prefectTag }}"
           imagePullPolicy: {{ .Values.worker.image.pullPolicy }}
@@ -125,6 +126,10 @@ spec:
           {{- with .Values.worker.initContainer.containerSecurityContext }}
           securityContext: {{- toYaml . | nindent 12 }}
           {{- end }}
+        {{- end }}
+        {{- if .Values.worker.initContainer.extraContainers }}
+        {{- include "common.tplvalues.render" (dict "value" .Values.worker.initContainer.extraContainers "context" $) | nindent 8 }}
+        {{- end }}
       {{- end }}
       containers:
         - name: prefect-worker

diff --git a/charts/prefect-worker/values.schema.json b/charts/prefect-worker/values.schema.json
@@ -172,6 +172,14 @@
                   }
                 }
               }
+            },
+            "extraContainers": {
+              "type": "array",
+              "title": "Extra Containers",
+              "description": "additional init containers",
+              "items": {
+                "type": "object"
+              }
             }
           }
         },

diff --git a/charts/prefect-worker/values.yaml b/charts/prefect-worker/values.yaml
@@ -52,6 +52,9 @@ worker:
       allowPrivilegeEscalation: false
       # -- set init container's security context capabilities
       capabilities: {}
+
+    # -- additional sidecar containers
+    extraContainers: []
 
   image:
     # -- worker image repository