From 8337f8e77fa9cf22dcd105c0382d20211e8bec67 Mon Sep 17 00:00:00 2001
From: wusamzong <t3742238@gmail.com>
Date: Fri, 10 Nov 2023 14:34:46 +0800
Subject: [PATCH 1/3] [YUNIKORN-2037] Testing the performance of Throughput

---
 deployments/kwok-perf-test/kwok-setup.sh      |  76 +++++++++++
 .../kwok-perf-test/throughput/deploy-tool.sh  | 121 ++++++++++++++++++
 2 files changed, 197 insertions(+)
 create mode 100755 deployments/kwok-perf-test/kwok-setup.sh
 create mode 100755 deployments/kwok-perf-test/throughput/deploy-tool.sh

diff --git a/deployments/kwok-perf-test/kwok-setup.sh b/deployments/kwok-perf-test/kwok-setup.sh
new file mode 100755
index 000000000..558524b8e
--- /dev/null
+++ b/deployments/kwok-perf-test/kwok-setup.sh
@@ -0,0 +1,76 @@
+#!/bin/bash
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+if [ $# -eq 0 ]; then
+  echo "Error: Please provide the number of nodes to create."
+  echo "Usage: $0 <number_of_nodes>"
+  exit 1
+fi
+
+KWOK_REPO=kubernetes-sigs/kwok
+KWOK_LATEST_RELEASE=$(curl "https://api.github.com/repos/${KWOK_REPO}/releases/latest" | jq -r '.tag_name')
+kubectl apply -f "https://github.com/${KWOK_REPO}/releases/download/${KWOK_LATEST_RELEASE}/kwok.yaml"
+kubectl apply -f "https://github.com/${KWOK_REPO}/releases/download/${KWOK_LATEST_RELEASE}/stage-fast.yaml"
+
+for (( i=0;i<$1; i++))
+do
+  kubectl apply -f - <<EOF
+  apiVersion: v1
+  kind: Node
+  metadata:
+    annotations:
+      node.alpha.kubernetes.io/ttl: "0"
+      kwok.x-k8s.io/node: fake
+    labels:
+      beta.kubernetes.io/arch: amd64
+      beta.kubernetes.io/os: linux
+      kubernetes.io/arch: amd64
+      kubernetes.io/hostname: kwok-node-$i
+      kubernetes.io/os: linux
+      kubernetes.io/role: agent
+      node-role.kubernetes.io/agent: ""
+      type: kwok
+    name: kwok-node-$i
+  spec:
+    taints: # Avoid scheduling actual running pods to fake Node
+    - effect: NoSchedule
+      key: kwok.x-k8s.io/node
+      value: fake
+  status:
+    allocatable:
+      cpu: 32
+      memory: 256Gi
+      pods: 110
+    capacity:
+      cpu: 32
+      memory: 256Gi
+      pods: 110
+    nodeInfo:
+      architecture: amd64
+      bootID: ""
+      containerRuntimeVersion: ""
+      kernelVersion: ""
+      kubeProxyVersion: fake
+      kubeletVersion: fake
+      machineID: ""
+      operatingSystem: linux
+      osImage: ""
+      systemUUID: ""
+    phase: Running
+EOF
+done
\ No newline at end of file
diff --git a/deployments/kwok-perf-test/throughput/deploy-tool.sh b/deployments/kwok-perf-test/throughput/deploy-tool.sh
new file mode 100755
index 000000000..d2c5a1c75
--- /dev/null
+++ b/deployments/kwok-perf-test/throughput/deploy-tool.sh
@@ -0,0 +1,121 @@
+#!/bin/bash
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+show_help() {
+  cat << EOF
+Invalid option: -$OPTARG
+Usage: $0 [-d] [-i <interval>] <deployment_count> <replicas_count>
+
+Options:
+  -d, --delete               Delete the specified number of deployments.
+  -i, --interval <interval>  Set the interval between deployments in seconds.
+
+Arguments:
+  <deployment_count>         Number of deployments to create or delete (required).
+  <replicas_count>           Number of replicas for each deployment (required).
+EOF
+}
+
+deploy_deployments() {
+  for (( i=0; i<$deployment_count; i++ )); do
+    kubectl apply -f - <<EOF
+apiVersion: apps/v1
+kind: Deployment
+metadata:
+  name: sleep-deployment-$i
+  labels:
+    app: sleep
+    applicationId: "sleep-deployment-$i"
+    queue: root.default
+spec:
+  replicas: $replicas_count
+  selector:
+    matchLabels:
+      app: nginx
+  template:
+    metadata:
+      labels:
+        app: nginx
+        applicationId: "sleep-deployment-$i"
+        queue: root.default
+    spec:
+      containers:
+      - name: sleep300
+        image: "alpine:latest"
+      tolerations:
+      - key: "kwok.x-k8s.io/node"
+        operator: "Exists"
+        effect: "NoSchedule"
+EOF
+    sleep $interval
+  done
+}
+
+delete_deployments(){
+    for (( i=0; i<$deployment_count; i++ )); do
+        kubectl delete deploy/sleep-deployment-$i
+    done
+}
+
+# Default values
+delete=false
+interval=0
+
+# Process command-line options
+while getopts ":di:" opt; do
+  case $opt in
+    d)
+      delete=true
+      ;;
+    i)
+      interval=$OPTARG
+      ;;
+    \?)
+      show_help
+      exit 1
+      ;;
+    :)
+      show_help
+      exit 1
+      ;;
+  esac
+done
+
+# Shift the processed options out of the command-line arguments
+shift $((OPTIND-1))
+
+# Check if deployment count and replicas count are provided
+if [ $# -ne 2 ]; then
+  show_help
+  exit 1
+fi
+
+# Assign provided values to variables
+deployment_count=$1
+replicas_count=$2
+
+# Check if delete flag is set
+if [ "$delete" = true ]; then
+  echo "Deleting $deployment_count deployments with $replicas_count replicas each."
+  delete_deployments
+else
+  echo "Deploying $deployment_count deployments with $replicas_count replicas each."
+  deploy_deployments 
+fi
+
+exit 0

From a6bd8d4e601209d6dbc4f17d2686125b093f6af6 Mon Sep 17 00:00:00 2001
From: wusamzong <t3742238@gmail.com>
Date: Fri, 10 Nov 2023 14:46:09 +0800
Subject: [PATCH 2/3] fix shell check

---
 deployments/kwok-perf-test/throughput/deploy-tool.sh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/deployments/kwok-perf-test/throughput/deploy-tool.sh b/deployments/kwok-perf-test/throughput/deploy-tool.sh
index d2c5a1c75..64bb64fde 100755
--- a/deployments/kwok-perf-test/throughput/deploy-tool.sh
+++ b/deployments/kwok-perf-test/throughput/deploy-tool.sh
@@ -32,7 +32,7 @@ EOF
 }
 
 deploy_deployments() {
-  for (( i=0; i<$deployment_count; i++ )); do
+  for (( i=0; i<deployment_count; i++ )); do
     kubectl apply -f - <<EOF
 apiVersion: apps/v1
 kind: Deployment
@@ -62,12 +62,12 @@ spec:
         operator: "Exists"
         effect: "NoSchedule"
 EOF
-    sleep $interval
+    sleep "$interval"
   done
 }
 
 delete_deployments(){
-    for (( i=0; i<$deployment_count; i++ )); do
+    for (( i=0; i<deployment_count; i++ )); do
         kubectl delete deploy/sleep-deployment-$i
     done
 }

From 0b8d4cb944e04e74fad6271f7f14bb684c1056d9 Mon Sep 17 00:00:00 2001
From: wusamzong <t3742238@gmail.com>
Date: Mon, 4 Dec 2023 12:19:35 +0800
Subject: [PATCH 3/3] add newline at the end of kwok-setup.sh & remove a
 separate directory

---
 deployments/kwok-perf-test/{throughput => }/deploy-tool.sh | 0
 deployments/kwok-perf-test/kwok-setup.sh                   | 2 +-
 2 files changed, 1 insertion(+), 1 deletion(-)
 rename deployments/kwok-perf-test/{throughput => }/deploy-tool.sh (100%)

diff --git a/deployments/kwok-perf-test/throughput/deploy-tool.sh b/deployments/kwok-perf-test/deploy-tool.sh
similarity index 100%
rename from deployments/kwok-perf-test/throughput/deploy-tool.sh
rename to deployments/kwok-perf-test/deploy-tool.sh
diff --git a/deployments/kwok-perf-test/kwok-setup.sh b/deployments/kwok-perf-test/kwok-setup.sh
index 558524b8e..6e5288838 100755
--- a/deployments/kwok-perf-test/kwok-setup.sh
+++ b/deployments/kwok-perf-test/kwok-setup.sh
@@ -73,4 +73,4 @@ do
       systemUUID: ""
     phase: Running
 EOF
-done
\ No newline at end of file
+done