Adding configurable cpu requests/limit for the debug daemonset

redhat-best-practices-for-k8s · Jan 17, 2024 · 5bde1f4 · 5bde1f4
1 parent 5439207
commit 5bde1f4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/pkg/configuration/configuration.go b/pkg/configuration/configuration.go
@@ -105,4 +105,8 @@ type TestParameters struct {
 	OmitArtifactsZipFile          bool   `split_words:"true" default:"false"`
 	EnableDataCollection          bool   `split_words:"true" envconfig:"ENABLE_DATA_COLLECTION" default:"false"`
 	EnableXMLCreation             bool   `split_words:"true" envconfig:"TNF_ENABLE_XML_CREATION" default:"false"`
+	DaemonsetCPUReq               string `default:"100m" split_words:"true"`
+	DaemonsetCPULim               string `default:"100m" split_words:"true"`
+	DaemonsetMemReq               string `default:"100M" split_words:"true"`
+	DaemonsetMemLim               string `default:"100M" split_words:"true"`
 }
diff --git a/pkg/provider/provider.go b/pkg/provider/provider.go
@@ -189,7 +189,12 @@ func deployDaemonSet(namespace string) error {
 	matchLabels := make(map[string]string)
 	matchLabels["name"] = DaemonSetName
 	matchLabels["test-network-function.com/app"] = DaemonSetName
-	_, err := k8sPrivilegedDs.CreateDaemonSet(DaemonSetName, namespace, containerName, dsImage, matchLabels, debugPodsTimeout)
+	_, err := k8sPrivilegedDs.CreateDaemonSet(DaemonSetName, namespace, containerName, dsImage, matchLabels, debugPodsTimeout,
+		configuration.GetTestParameters().DaemonsetCPUReq,
+		configuration.GetTestParameters().DaemonsetCPULim,
+		configuration.GetTestParameters().DaemonsetMemReq,
+		configuration.GetTestParameters().DaemonsetMemLim,
+	)
 	if err != nil {
 		return fmt.Errorf("could not deploy tnf daemonset, err=%v", err)
 	}