From 59f60b33f3549d7ebf7204ef3da1f4341aa936ee Mon Sep 17 00:00:00 2001
From: Faiz <65137316+mdfaizsiddiqui@users.noreply.github.com>
Date: Thu, 23 Mar 2023 11:23:14 +0000
Subject: [PATCH] Support for affinity rules in helm chart

---
 helm/m3db-operator/templates/stateful_set.yaml | 4 ++++
 helm/m3db-operator/values.yaml                 | 1 +
 2 files changed, 5 insertions(+)

diff --git a/helm/m3db-operator/templates/stateful_set.yaml b/helm/m3db-operator/templates/stateful_set.yaml
index e4ed3366..b57105d9 100644
--- a/helm/m3db-operator/templates/stateful_set.yaml
+++ b/helm/m3db-operator/templates/stateful_set.yaml
@@ -28,3 +28,7 @@ spec:
             - name: ENVIRONMENT
               value: {{ .Values.environment }}
       serviceAccount: {{ .Values.operator.name }}
+      {{- with .Values.affinity }}
+      affinity:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
diff --git a/helm/m3db-operator/values.yaml b/helm/m3db-operator/values.yaml
index b0b6448f..7d1b7e91 100644
--- a/helm/m3db-operator/values.yaml
+++ b/helm/m3db-operator/values.yaml
@@ -4,3 +4,4 @@ image:
   repository: quay.io/m3db/m3db-operator
   tag: v0.14.0
 environment: production
+affinity: {}
\ No newline at end of file