From e148499a5f064890bd5ba39d82916940c10c280b Mon Sep 17 00:00:00 2001
From: Smit Thakkar <smitthakkar96@gmail.com>
Date: Thu, 25 Jun 2020 12:18:50 +0400
Subject: [PATCH] added tolerations, nodeSelectors and affinity

---
 charts/argo-rollouts/README.md                     |  4 ++++
 .../templates/argo-rollouts-deployment.yaml        | 14 ++++++++++++++
 charts/argo-rollouts/values.yaml                   |  7 +++++++
 3 files changed, 25 insertions(+)

diff --git a/charts/argo-rollouts/README.md b/charts/argo-rollouts/README.md
index 86292c74..6a775bba 100644
--- a/charts/argo-rollouts/README.md
+++ b/charts/argo-rollouts/README.md
@@ -35,6 +35,10 @@ $ helm install --name my-release argo/argo-rollouts
 | controller.image.repository | string | `"argoproj/argo-rollouts"` |  |
 | controller.image.tag | string | `"v0.8.0"` |  |
 | controller.name | string | `"argo-rollouts"` |  |
+| controller.resources | Resource limits and requests for the controller pods. | `{}` |
+| controller.tolerations | [Tolerations for use with node taints](https://kubernetes.io/docs/concepts/configuration/taint-and-toleration/) | `[]` |
+| controller.affinity | [Assign custom affinity rules to the deployment](https://kubernetes.io/docs/concepts/configuration/assign-pod-node/) | `{}` |
+| controller.nodeSelector | [Node selector](https://kubernetes.io/docs/user-guide/node-selection/) | `{}` |
 | imagePullSecrets | list | `[]` |  |
 | installCRDs | bool | `true` |  |
 | podAnnotations | object | `{}` |  |
diff --git a/charts/argo-rollouts/templates/argo-rollouts-deployment.yaml b/charts/argo-rollouts/templates/argo-rollouts-deployment.yaml
index 1fb1a2d2..595fe120 100644
--- a/charts/argo-rollouts/templates/argo-rollouts-deployment.yaml
+++ b/charts/argo-rollouts/templates/argo-rollouts-deployment.yaml
@@ -38,6 +38,20 @@ spec:
         volumeMounts:
         - name: tmp
           mountPath: /tmp
+        resources:
+          {{- toYaml .Values.controller.resources | nindent 10 }}
+      {{- if .Values.controller.nodeSelector }}
+      nodeSelector:
+        {{- toYaml .Values.controller.nodeSelector | nindent 8 }}
+      {{- end }}
+      {{- if .Values.controller.tolerations }}
+      tolerations:
+        {{- toYaml .Values.controller.tolerations | nindent 8 }}
+      {{- end }}
+      {{- if .Values.controller.affinity }}
+      affinity:
+        {{- toYaml .Values.controller.affinity | nindent 8 }}
+      {{- end }}
       volumes:
       - name: tmp
         emptyDir: {}
diff --git a/charts/argo-rollouts/values.yaml b/charts/argo-rollouts/values.yaml
index 18b7dff4..9f86dc19 100644
--- a/charts/argo-rollouts/values.yaml
+++ b/charts/argo-rollouts/values.yaml
@@ -5,6 +5,13 @@ clusterInstall: true
 controller:
   name: argo-rollouts
   component: rollouts-controller
+  ## Node selectors and tolerations for server scheduling to nodes with taints
+  ## Ref: https://kubernetes.io/docs/concepts/configuration/assign-pod-node/
+  ##
+  nodeSelector:
+    dubizzle.com/group: ops
+  tolerations: []
+  affinity: {}
   image:
     repository: argoproj/argo-rollouts
     tag: v0.8.0