From 35c2c2eef8cadbd631ff9283b3d242c011ee40b8 Mon Sep 17 00:00:00 2001
From: Marwan Salem <marwan.saad@incorta.com>
Date: Mon, 13 Nov 2023 17:38:35 +0200
Subject: [PATCH] set kubernetes resources field for master and worker init
 containers

---
 charts/celeborn/templates/master-statefulset.yaml | 2 ++
 charts/celeborn/templates/worker-statefulset.yaml | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/charts/celeborn/templates/master-statefulset.yaml b/charts/celeborn/templates/master-statefulset.yaml
index fbd10192988..e988783f57b 100644
--- a/charts/celeborn/templates/master-statefulset.yaml
+++ b/charts/celeborn/templates/master-statefulset.yaml
@@ -88,6 +88,8 @@ spec:
         - chown
         - {{ .Values.securityContext.runAsUser | default 10006 }}:{{ .Values.securityContext.runAsGroup | default 10006 }}
         - {{ (index $dirs 0).mountPath }}
+        resources:
+          {{- toYaml .Values.resources.master | nindent 12 }}
         volumeMounts:
           - name: {{ $.Release.Name }}-master-vol-0
             mountPath: {{ (index $dirs 0).mountPath }}
diff --git a/charts/celeborn/templates/worker-statefulset.yaml b/charts/celeborn/templates/worker-statefulset.yaml
index 11efd50de14..a9a003adc89 100644
--- a/charts/celeborn/templates/worker-statefulset.yaml
+++ b/charts/celeborn/templates/worker-statefulset.yaml
@@ -90,6 +90,8 @@ spec:
         {{- range $dir := $dirs }}
         - {{ $dir.mountPath }}
         {{- end}}
+        resources:
+          {{- toYaml .Values.resources.worker | nindent 12 }}
         volumeMounts:
         {{- range $index, $dir := $dirs }}
         - name: {{ $.Release.Name }}-worker-vol-{{ $index }}