Add an option to inject additional measurements to load config

kubernetes · Jan 8, 2024 · 5a36f2c · 5a36f2c
1 parent 4e2bd7e
commit 5a36f2c
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/clusterloader2/testing/load/config.yaml b/clusterloader2/testing/load/config.yaml
@@ -77,6 +77,8 @@
 {{$latencyPodImage := DefaultParam .CL2_LATENCY_POD_IMAGE "registry.k8s.io/pause:3.9"}}
 {{$defaultQps := DefaultParam .CL2_DEFAULT_QPS (IfThenElse (le .Nodes 500) 10 100)}}
 
+{{$ADDITIONAL_MEASUREMENT_MODULES := DefaultParam .CL2_ADDITIONAL_MEASURMENT_MODULES nil}}
+
 name: load
 namespace:
   number: {{AddInt $namespaces $schedulerThroughputNamespaces}}
@@ -120,6 +122,15 @@ steps:
     params:
       action: start
 
+{{if $ADDITIONAL_MEASUREMENT_MODULES}}
+{{range $ADDITIONAL_MEASUREMENT_MODULES}}
+- module:
+    path: {{.}}
+    params:
+      action: start
+{{end}}
+{{end}}
+
 {{if $ENABLE_NETWORK_POLICY_ENFORCEMENT_LATENCY_TEST}}
 - module:
     path: modules/network-policy/net-policy-enforcement-latency.yaml
@@ -394,6 +405,15 @@ steps:
     params:
       action: gather
 
+{{if $ADDITIONAL_MEASUREMENT_MODULES}}
+{{range $ADDITIONAL_MEASUREMENT_MODULES}}
+- module:
+    path: {{.}}
+    params:
+      action: gather
+{{end}}
+{{end}}
+
 {{if $ENABLE_NETWORK_POLICY_ENFORCEMENT_LATENCY_TEST}}
 - module:
     path: modules/network-policy/net-policy-enforcement-latency.yaml