From 1f547f2e1d76dcd9d155a1beefcb9c59dbbec12b Mon Sep 17 00:00:00 2001
From: Hui Kang <hui.kang@salesforce.com>
Date: Mon, 16 Aug 2021 10:34:04 -0400
Subject: [PATCH] feat(argo-rollouts): add replicas variable for controller and
 dashboard  pods

Signed-off-by: Hui Kang <hui.kang@salesforce.com>
---
 charts/argo-rollouts/Chart.yaml                              | 3 ++-
 charts/argo-rollouts/README.md                               | 2 ++
 charts/argo-rollouts/templates/argo-rollouts-deployment.yaml | 1 +
 charts/argo-rollouts/templates/dashboard/deployment.yaml     | 1 +
 charts/argo-rollouts/values.yaml                             | 2 ++
 5 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/charts/argo-rollouts/Chart.yaml b/charts/argo-rollouts/Chart.yaml
index 7bdb6ea9..c0a78e3b 100644
--- a/charts/argo-rollouts/Chart.yaml
+++ b/charts/argo-rollouts/Chart.yaml
@@ -2,7 +2,7 @@ apiVersion: v2
 appVersion: "v1.0.2"
 description: A Helm chart for Argo Rollouts
 name: argo-rollouts
-version: 2.0.0
+version: 2.0.1
 icon: https://argoproj.github.io/argo-rollouts/assets/logo.png
 home: https://github.com/argoproj/argo-helm
 maintainers:
@@ -12,3 +12,4 @@ maintainers:
 annotations:
   artifacthub.io/changes: |
     - "[Added]: Dashboard manifests"
+    - "[Added]: replicas variables for controller and dashboard"
diff --git a/charts/argo-rollouts/README.md b/charts/argo-rollouts/README.md
index 9fcd862a..f8eb0c35 100644
--- a/charts/argo-rollouts/README.md
+++ b/charts/argo-rollouts/README.md
@@ -34,6 +34,7 @@ If dashboard is installed by `--set dashboard.enabled=true`, checkout the argo-r
 |-----|------|---------|-------------|
 | clusterInstall | bool | `true` | `false` runs controller in namespaced mode (does not require cluster RBAC) |
 | controller.component | string | `"rollouts-controller"` | Value of label `app.kubernetes.io/component` |
+| controller.replicas | int | `1` | The number of controller pods to run |
 | controller.image.pullPolicy | string | `"IfNotPresent"` | Image pull policy |
 | controller.image.registry | string | `quay.io` | Registry to use |
 | controller.image.repository | string | `"argoproj/argo-rollouts"` | Repository to use |
@@ -58,6 +59,7 @@ If dashboard is installed by `--set dashboard.enabled=true`, checkout the argo-r
 | containerSecurityContext | object | `{}` | Security Context to set on container level |
 | dashboard.enabled | bool | `false` | Deploy dashboard server |
 | dashboard.component | string | `"rollouts-dashboard"` | Value of label `app.kubernetes.io/component` |
+| dashboard.replicas | int | `1` | The number of dashboard pods to run |
 | dashboard.image.pullPolicy | string | `"IfNotPresent"` | Image pull policy |
 | dashboard.image.registry | string | `quay.io` | Registry to use |
 | dashboard.image.repository | string | `"argoproj/kubectl-argo-rollouts"` | Repository to use |
diff --git a/charts/argo-rollouts/templates/argo-rollouts-deployment.yaml b/charts/argo-rollouts/templates/argo-rollouts-deployment.yaml
index 71d73352..e7d29d98 100644
--- a/charts/argo-rollouts/templates/argo-rollouts-deployment.yaml
+++ b/charts/argo-rollouts/templates/argo-rollouts-deployment.yaml
@@ -12,6 +12,7 @@ spec:
       {{- include "argo-rollouts.selectorLabels" . | nindent 6 }}
   strategy:
     type: Recreate
+  replicas: {{ .Values.controller.replicas }}
   template:
     metadata:
       {{- with .Values.podAnnotations }}
diff --git a/charts/argo-rollouts/templates/dashboard/deployment.yaml b/charts/argo-rollouts/templates/dashboard/deployment.yaml
index 48364587..36964d09 100644
--- a/charts/argo-rollouts/templates/dashboard/deployment.yaml
+++ b/charts/argo-rollouts/templates/dashboard/deployment.yaml
@@ -13,6 +13,7 @@ spec:
       app.kubernetes.io/component: {{ .Values.dashboard.component }}
   strategy:
     type: Recreate
+  replicas: {{ .Values.dashboard.replicas }}
   template:
     metadata:
       {{- with .Values.podAnnotations }}
diff --git a/charts/argo-rollouts/values.yaml b/charts/argo-rollouts/values.yaml
index 4c288b98..9e0738d1 100644
--- a/charts/argo-rollouts/values.yaml
+++ b/charts/argo-rollouts/values.yaml
@@ -4,6 +4,7 @@ clusterInstall: true
 
 controller:
   component: rollouts-controller
+  replicas: 1
   ## Node selectors and tolerations for server scheduling to nodes with taints
   ## Ref: https://kubernetes.io/docs/concepts/configuration/assign-pod-node/
   ##
@@ -97,6 +98,7 @@ imagePullSecrets: []
 
 dashboard:
   enabled: false
+  replicas: 1
   component: rollouts-dashboard
   ## Node selectors and tolerations for server scheduling to nodes with taints
   ## Ref: https://kubernetes.io/docs/concepts/configuration/assign-pod-node/