From 795c7c23594735c7d7bdfe5013d994fcf54e211c Mon Sep 17 00:00:00 2001
From: prafull01 <prafull.ladha@gmail.com>
Date: Wed, 26 Mar 2025 13:34:02 +0530
Subject: [PATCH] Add QoS for the init containers

---
 pkg/resource/statefulset.go                            | 10 ++++++++++
 .../TestStatefulSetBuilder/automount_sa.golden         |  8 +++++++-
 .../TestStatefulSetBuilder/default_secure.golden       |  8 +++++++-
 3 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/pkg/resource/statefulset.go b/pkg/resource/statefulset.go
index 838ef48f5..da78b73ea 100644
--- a/pkg/resource/statefulset.go
+++ b/pkg/resource/statefulset.go
@@ -263,6 +263,16 @@ func (b StatefulSetBuilder) MakeInitContainers() []corev1.Container {
 				RunAsUser:                ptr.Int64(0),
 				AllowPrivilegeEscalation: ptr.Bool(false),
 			},
+			Resources: corev1.ResourceRequirements{
+				Limits: corev1.ResourceList{
+					corev1.ResourceCPU:    resource.MustParse("100m"),
+					corev1.ResourceMemory: resource.MustParse("200Mi"),
+				},
+				Requests: corev1.ResourceList{
+					corev1.ResourceCPU:    resource.MustParse("50m"),
+					corev1.ResourceMemory: resource.MustParse("100Mi"),
+				},
+			},
 		},
 	}
 }
diff --git a/pkg/resource/testdata/TestStatefulSetBuilder/automount_sa.golden b/pkg/resource/testdata/TestStatefulSetBuilder/automount_sa.golden
index 406a9dad0..2a4295f7b 100644
--- a/pkg/resource/testdata/TestStatefulSetBuilder/automount_sa.golden
+++ b/pkg/resource/testdata/TestStatefulSetBuilder/automount_sa.golden
@@ -98,7 +98,13 @@ spec:
         image: cockroachdb/cockroach:v21.1.0
         imagePullPolicy: IfNotPresent
         name: db-init
-        resources: {}
+        resources:
+          limits:
+            cpu: 100m
+            memory: 200Mi
+          requests:
+            cpu: 50m
+            memory: 100Mi
         securityContext:
           allowPrivilegeEscalation: false
           runAsUser: 0
diff --git a/pkg/resource/testdata/TestStatefulSetBuilder/default_secure.golden b/pkg/resource/testdata/TestStatefulSetBuilder/default_secure.golden
index 08eef6f85..425470c11 100644
--- a/pkg/resource/testdata/TestStatefulSetBuilder/default_secure.golden
+++ b/pkg/resource/testdata/TestStatefulSetBuilder/default_secure.golden
@@ -98,7 +98,13 @@ spec:
         image: cockroachdb/cockroach:v21.1.0
         imagePullPolicy: IfNotPresent
         name: db-init
-        resources: {}
+        resources:
+          limits:
+            cpu: 100m
+            memory: 200Mi
+          requests:
+            cpu: 50m
+            memory: 100Mi
         securityContext:
           allowPrivilegeEscalation: false
           runAsUser: 0