add nodeSelector

px-bench-env.yml

@@ -3,10 +3,9 @@ kind: ConfigMap
 metadata:
   name: env
   namespace: px-bench
-######
-# Options that control the underlying fio benchmark utility. These equate to the same-named option in fio
-#
 data:
+  #node_selector: "kubernetes.io/hostname=node-1-1" # uncomment to pin fio to node(s)
+  # Options that control the underlying fio benchmark utility. These equate to the same-named option in fio
   file_size: "10GiB"        # size of region of IO testing, will also be the size of the file
   pvc_size: "50Gi"          # If the storageclass is backed by a larger disk from infrastructure,
                             # such as a Portworx pool, then pvc_size should be the same size as

px-bench-main.yml

@@ -34,6 +34,7 @@ data:
         export time_based=
         envsubst < /px-bench/main.fio.tmpl > $filename
     done
+    nodeSelector=$(echo $node_selector | sed 's/=/: /')
     kubectl create configmap fio-config --from-file=/tmp/fio -n px-bench
     for sc in $storageclass_list; do
       volumes="$volumes\
@@ -60,7 +61,6 @@ data:
            storage: $pvc_size
     EOF
     done
-    #echo $pvcs >/pvc.yml
     kubectl apply -f - <<EOF
     apiVersion: batch/v1
     kind: Job
@@ -74,6 +74,8 @@ data:
           labels:
             px-bench: fio
         spec:
+          nodeSelector:
+            $nodeSelector
           securityContext:
             fsGroupChangePolicy: "OnRootMismatch"
           serviceAccountName: px-bench