From 9425a23bac52ae403ae52f60028a4c5222191e89 Mon Sep 17 00:00:00 2001
From: sachinsaju <33017477+sachinsaju@users.noreply.github.com>
Date: Fri, 21 Jul 2023 23:05:10 +0530
Subject: [PATCH] feat(cron): support nodeselector, affinity and toleration
 capabilities  (#342)

* feat : add tolerations parameter in datahub-ingestion-cron chart

---------

Co-authored-by: david-leifker <114954101+david-leifker@users.noreply.github.com>
---
 charts/datahub/Chart.yaml                          |  4 ++--
 .../subcharts/datahub-ingestion-cron/Chart.yaml    |  2 +-
 .../datahub-ingestion-cron/templates/cron.yaml     | 12 ++++++++++++
 .../subcharts/datahub-ingestion-cron/values.yaml   | 14 ++++++++++++++
 4 files changed, 29 insertions(+), 3 deletions(-)

diff --git a/charts/datahub/Chart.yaml b/charts/datahub/Chart.yaml
index 78afa2c2b..251e85465 100644
--- a/charts/datahub/Chart.yaml
+++ b/charts/datahub/Chart.yaml
@@ -4,7 +4,7 @@ description: A Helm chart for LinkedIn DataHub
 type: application
 # This is the chart version. This version number should be incremented each time you make changes
 # to the chart and its templates, including the app version.
-version: 0.2.179
+version: 0.2.180
 # This is the version number of the application being deployed. This version number should be
 # incremented each time you make changes to the application.
 appVersion: 0.10.4
@@ -26,7 +26,7 @@ dependencies:
     repository: file://./subcharts/datahub-mce-consumer
     condition: global.datahub_standalone_consumers_enabled
   - name: datahub-ingestion-cron
-    version: 0.2.133
+    version: 0.2.134
     repository: file://./subcharts/datahub-ingestion-cron
     condition: datahub-ingestion-cron.enabled
   - name: acryl-datahub-actions
diff --git a/charts/datahub/subcharts/datahub-ingestion-cron/Chart.yaml b/charts/datahub/subcharts/datahub-ingestion-cron/Chart.yaml
index f24e507fe..95eca2002 100644
--- a/charts/datahub/subcharts/datahub-ingestion-cron/Chart.yaml
+++ b/charts/datahub/subcharts/datahub-ingestion-cron/Chart.yaml
@@ -12,7 +12,7 @@ description: A Helm chart for Kubernetes
 type: application
 # This is the chart version. This version number should be incremented each time you make changes
 # to the chart and its templates, including the app version.
-version: 0.2.133
+version: 0.2.134
 # This is the version number of the application being deployed. This version number should be
 # incremented each time you make changes to the application.
 appVersion: v0.10.0
diff --git a/charts/datahub/subcharts/datahub-ingestion-cron/templates/cron.yaml b/charts/datahub/subcharts/datahub-ingestion-cron/templates/cron.yaml
index ad8fdb425..5b2c417db 100644
--- a/charts/datahub/subcharts/datahub-ingestion-cron/templates/cron.yaml
+++ b/charts/datahub/subcharts/datahub-ingestion-cron/templates/cron.yaml
@@ -65,6 +65,18 @@ spec:
               {{- end }}
               {{- end }}
           restartPolicy: {{ default "OnFailure" .restartPolicy }}
+          {{- if .nodeSelector }}
+          nodeSelector:
+            {{- toYaml .nodeSelector | nindent 12 }}
+          {{- end }}
+          {{- if .affinity }}
+          affinity:
+            {{- toYaml .affinity | nindent 12 }}
+          {{- end }}
+          {{- if .tolerations }}
+          tolerations:
+            {{- toYaml .tolerations | nindent 12 }}
+          {{- end }}
           volumes:
             - name: recipe
               configMap:
diff --git a/charts/datahub/subcharts/datahub-ingestion-cron/values.yaml b/charts/datahub/subcharts/datahub-ingestion-cron/values.yaml
index 3e5db76e3..f4ad84c1f 100644
--- a/charts/datahub/subcharts/datahub-ingestion-cron/values.yaml
+++ b/charts/datahub/subcharts/datahub-ingestion-cron/values.yaml
@@ -86,6 +86,20 @@ crons: {}
     ##
     #backoffLimit:
 
+    ## Node Selector parameters for job pod
+    ##
+    #nodeSelector: {}
+    #  region: us-west-1
+    #  disk: ssd
+
+    ## Affinity settings for job pod
+    ##
+    #affinity : {}
+
+    ## Toleration settings  for job pod
+    ##
+    #tolerations: []
+
 # Add extra sidecar containers to deployment pod
 extraSidecars: []
   # - name: my-image-name