Add a nodeSelector for vegeta pods

kedark3 · Sep 2, 2021 · befbd51 · befbd51
1 parent abc48ea
commit befbd51
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/config/samples/vegeta/cr.yaml b/config/samples/vegeta/cr.yaml
@@ -12,6 +12,7 @@ spec:
   workload:
     name: vegeta
     args:
+      # node_selector: "vegeta=true"
       hostnetwork: false
       clients: 2
       targets:

diff --git a/docs/vegeta.md b/docs/vegeta.md
@@ -11,6 +11,9 @@ The option **runtime_class** can be set to specify an optional
 runtime_class to the podSpec runtimeClassName.  This is primarily
 intended for Kata containers.
 
+The **node_selector** option can be used to limit the nodes where
+the vegeta pods are deployed.
+
 ```yaml
 apiVersion: ripsaw.cloudbulldozer.io/v1alpha1
 kind: Benchmark
@@ -24,6 +27,7 @@ spec:
   workload:
     name: vegeta
     args:
+      # node_selector: "vegeta=true"
       clients: 2
       image: quay.io/cloud-bulldozer/vegeta:latest
       hostnetwork: false

diff --git a/roles/vegeta/templates/vegeta.yml.j2 b/roles/vegeta/templates/vegeta.yml.j2
@@ -13,6 +13,10 @@ spec:
         app: vegeta-benchmark-{{ trunc_uuid }}
         benchmark-uuid: {{ uuid }}
     spec:
+{% if workload_args.node_selector is defined %}
+      nodeSelector:
+        '{{ workload_args.node_selector.split("=")[0] }}': '{{ workload_args.node_selector.split("=")[1] }}'
+{% endif %}
 {% if workload_args.runtime_class is defined %}
       runtimeClassName: "{{ workload_args.runtime_class }}"
 {% endif %}