Add extraEnvVarsCMs for multiple ConfigMaps

Adds support for definining multiple ConfigMaps in the Worker Deployment. Related to #421
PrefectHQ · Jan 9, 2025 · 921fc95 · 921fc95
1 parent 04b1fbc
commit 921fc95
Show file tree

Hide file tree

Showing 5 changed files with 36 additions and 3 deletions.
diff --git a/charts/prefect-worker/README.md b/charts/prefect-worker/README.md
@@ -308,7 +308,8 @@ worker:
 | worker.extraArgs | list | `[]` | array with extra Arguments for the worker container to start with |
 | worker.extraContainers | list | `[]` | additional sidecar containers |
 | worker.extraEnvVars | list | `[]` | array with extra environment variables to add to worker nodes |
-| worker.extraEnvVarsCM | string | `""` | name of existing ConfigMap containing extra env vars to add to worker nodes |
+| worker.extraEnvVarsCM | string | `""` | name of existing ConfigMap containing extra env vars to add to worker nodes (deprecated, use extraEnvVarsCMs) |
+| worker.extraEnvVarsCMs | list | `[]` | names of existing ConfigMaps containing extra env vars to add to worker nodes |
 | worker.extraEnvVarsSecret | string | `""` | name of existing Secret containing extra env vars to add to worker nodes |
 | worker.extraVolumeMounts | list | `[]` | array with extra volumeMounts for the worker pod |
 | worker.extraVolumes | list | `[]` | array with extra volumes for the worker pod |

diff --git a/charts/prefect-worker/templates/deployment.yaml b/charts/prefect-worker/templates/deployment.yaml
@@ -209,6 +209,10 @@ spec:
             - configMapRef:
                 name: {{ include "common.tplvalues.render" (dict "value" .Values.worker.extraEnvVarsCM "context" $) }}
             {{- end }}
+            {{- range .Values.worker.extraEnvVarsCMs }}
+            - configMapRef:
+                name: {{ include "common.tplvalues.render" (dict "value" . "context" $) }}
+            {{- end }}
             {{- if .Values.worker.extraEnvVarsSecret }}
             - secretRef:
                 name: {{ include "common.tplvalues.render" (dict "value" .Values.worker.extraEnvVarsSecret "context" $) }}

diff --git a/charts/prefect-worker/tests/worker_test.yaml b/charts/prefect-worker/tests/worker_test.yaml
@@ -319,6 +319,26 @@ tests:
           path: .spec.template.spec.containers[0].envFrom[0].configMapRef.name
           value: my-config-map
 
+  - it: Should set extra environment variables from ConfigMaps
+    set:
+      worker:
+        extraEnvVarsCMs:
+          - my-config-map
+          - my-other-config-map
+    asserts:
+      - template: deployment.yaml
+        contains:
+          path: .spec.template.spec.containers[0].envFrom
+          content:
+            configMapRef:
+              name: my-config-map
+      - template: deployment.yaml
+        contains:
+          path: .spec.template.spec.containers[0].envFrom
+          content:
+            configMapRef:
+              name: my-other-config-map
+
   - it: Should set extra volumes
     set:
       worker:

diff --git a/charts/prefect-worker/values.schema.json b/charts/prefect-worker/values.schema.json
@@ -621,7 +621,12 @@
         "extraEnvVarsCM": {
           "type": "string",
           "title": "Extra Env Vars ConfigMap",
-          "description": "name of existing ConfigMap containing extra env vars to add to worker nodes"
+          "description": "name of existing ConfigMap containing extra env vars to add to worker nodes (deprecated, use extraEnvVarsCMs)"
+        },
+        "extraEnvVarsCMs": {
+          "type": "array",
+          "title": "Extra Env Vars ConfigMaps",
+          "description": "names of existing ConfigMaps containing extra env vars to add to worker nodes"
         },
         "extraEnvVarsSecret": {
           "type": "string",

diff --git a/charts/prefect-worker/values.yaml b/charts/prefect-worker/values.yaml
@@ -252,9 +252,12 @@ worker:
   # -- array with extra environment variables to add to worker nodes
   extraEnvVars: []
 
-  # -- name of existing ConfigMap containing extra env vars to add to worker nodes
+  # -- name of existing ConfigMap containing extra env vars to add to worker nodes (deprecated, use extraEnvVarsCMs)
   extraEnvVarsCM: ""
 
+  # -- names of existing ConfigMaps containing extra env vars to add to worker nodes
+  extraEnvVarsCMs: []
+
   # -- name of existing Secret containing extra env vars to add to worker nodes
   extraEnvVarsSecret: ""