From bc87f3152cab6863e800329ae80572ee34eed81e Mon Sep 17 00:00:00 2001
From: Andrew Hill <ahill@purestorage.com>
Date: Wed, 10 Jan 2024 15:30:09 +0000
Subject: [PATCH] add nodeSelector

---
 px-bench-env.yml  | 5 ++---
 px-bench-main.yml | 4 +++-
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/px-bench-env.yml b/px-bench-env.yml
index 1576fc9..14db60a 100644
--- a/px-bench-env.yml
+++ b/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
diff --git a/px-bench-main.yml b/px-bench-main.yml
index b8b6842..49c8103 100644
--- a/px-bench-main.yml
+++ b/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