feat: add restartPolicy prop to BackgroundWorker

techfromsage · Oct 17, 2022 · 5165c0a · 5165c0a
1 parent 0470c26
commit 5165c0a
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 0 deletions.
diff --git a/lib/background-worker/background-worker-props.ts b/lib/background-worker/background-worker-props.ts
@@ -15,4 +15,10 @@ export interface BackgroundWorkerProps
    * @default { app: "<app label from chart>", role: "worker", instance: "<construct id>" }
    */
   readonly selectorLabels?: { [key: string]: string };
+
+  /**
+   * Restart policy for all containers within the pod.
+   * @default "Always"
+   */
+  readonly restartPolicy?: "Always" | "OnFailure" | "Never";
 }
diff --git a/lib/background-worker/background-worker.ts b/lib/background-worker/background-worker.ts
@@ -51,6 +51,7 @@ export class BackgroundWorker extends Construct {
               props.automountServiceAccountToken ?? false,
             imagePullSecrets: props.imagePullSecrets,
             priorityClassName: props.priorityClassName,
+            restartPolicy: props.restartPolicy,
             terminationGracePeriodSeconds: props.terminationGracePeriodSeconds,
             volumes: props.volumes,
             hostAliases: props.hostAliases,

diff --git a/test/background-worker/__snapshots__/background-worker.test.ts.snap b/test/background-worker/__snapshots__/background-worker.test.ts.snap
@@ -222,6 +222,7 @@ exports[`BackgroundWorker > Props > All the props 1`] = `
             },
           ],
           "priorityClassName": "high-priority",
+          "restartPolicy": "OnFailure",
           "terminationGracePeriodSeconds": 300,
           "volumes": [
             {

diff --git a/test/background-worker/background-worker.test.ts b/test/background-worker/background-worker.test.ts
@@ -59,6 +59,7 @@ describe("BackgroundWorker", () => {
         imagePullSecrets: [{ name: "foo-secret" }],
         priorityClassName: "high-priority",
         revisionHistoryLimit: 5,
+        restartPolicy: "OnFailure",
         affinity: {
           podAntiAffinity: {
             preferredDuringSchedulingIgnoredDuringExecution: [