Enable countErrorMargin parameter for waitForRunningPodsMeasurement. (#34)

Author: sumanthreddy29

clusterloader2/Dockerfile

@@ -1,4 +1,4 @@
-ARG GOLANG_VERSION=1.22
+ARG GOLANG_VERSION=1.24
 FROM golang:${GOLANG_VERSION} AS builder
 
 WORKDIR /root/perf-tests/clusterloader2

clusterloader2/pkg/measurement/common/wait_for_pods.go

@@ -30,6 +30,7 @@ import (
 const (
 	defaultWaitForPodsTimeout         = 60 * time.Second
 	defaultWaitForPodsInterval        = 5 * time.Second
+	defaultCountErrorMargin           = 0
 	defaultIsFatal                    = false
 	waitForRunningPodsMeasurementName = "WaitForRunningPods"
 )
@@ -54,6 +55,10 @@ func (w *waitForRunningPodsMeasurement) Execute(config *measurement.Config) ([]m
 	if err != nil {
 		return nil, err
 	}
+	countErrorMargin, err := util.GetIntOrDefault(config.Params, "countErrorMargin", defaultCountErrorMargin)
+	if err != nil {
+		return nil, err
+	}
 	selector := util.NewObjectSelector()
 	if err := selector.Parse(config.Params); err != nil {
 		return nil, err
@@ -76,6 +81,7 @@ func (w *waitForRunningPodsMeasurement) Execute(config *measurement.Config) ([]m
 	defer cancel()
 	options := &measurementutil.WaitForPodOptions{
 		DesiredPodCount:     func() int { return desiredPodCount },
+		CountErrorMargin:    countErrorMargin,
 		CallerName:          w.String(),
 		WaitForPodsInterval: refreshInterval,
 	}