Fix UDP_STREAM

Signed-off-by: Joe Talerico aka rook <[email protected]>
cloud-bulldozer · Sep 5, 2024 · 08c9654 · 08c9654
1 parent 133e2b9
commit 08c9654
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/pkg/drivers/netperf.go b/pkg/drivers/netperf.go
@@ -49,13 +49,19 @@ func (n *netperf) Run(c *kubernetes.Clientset, rc rest.Config, nc config.Config,
 		"-k", fmt.Sprint(omniOptions)}
 	var additionalOptions []string
 	if strings.Contains(nc.Profile, "STREAM") {
-		additionalOptions = []string {
-			"-m", fmt.Sprint(nc.MessageSize)}
+		if strings.Contains(nc.Profile, "UDP") {
+			additionalOptions = []string{
+				"-m", fmt.Sprint(nc.MessageSize),
+				"-R", "1"}
+		} else {
+			additionalOptions = []string{
+				"-m", fmt.Sprint(nc.MessageSize)}
+		}
 	} else {
-		additionalOptions = []string {
+		additionalOptions = []string{
 			"-r", fmt.Sprint(nc.MessageSize, ",", nc.MessageSize)}
 		if strings.Contains(nc.Profile, "TCP_RR") && (nc.Burst > 0) {
-			burst := []string {"-b", fmt.Sprint(nc.Burst)}
+			burst := []string{"-b", fmt.Sprint(nc.Burst)}
 			additionalOptions = append(additionalOptions, burst...)
 		}
 	}