From bc208b2f890b73f5471e3a9f3373d060ff638375 Mon Sep 17 00:00:00 2001
From: "felix.fengmin" <felix.fengmin@bytedance.com>
Date: Wed, 27 Mar 2024 17:19:31 +0800
Subject: [PATCH] feat: exit benchmark on unsupported-env

---
 scripts/benchmark_compare.sh | 2 ++
 scripts/env.sh               | 2 ++
 scripts/util.sh              | 7 +++++++
 3 files changed, 11 insertions(+)
 create mode 100644 scripts/util.sh

diff --git a/scripts/benchmark_compare.sh b/scripts/benchmark_compare.sh
index 329bf7e..9cdada8 100755
--- a/scripts/benchmark_compare.sh
+++ b/scripts/benchmark_compare.sh
@@ -15,6 +15,8 @@
 cd `dirname $0`/..
 PROJECT_ROOT=`pwd`
 
+source $PROJECT_ROOT/scripts/util.sh && check_supported_env
+
 old=$1
 new=$2
 
diff --git a/scripts/env.sh b/scripts/env.sh
index 27a2e0f..f3253ea 100755
--- a/scripts/env.sh
+++ b/scripts/env.sh
@@ -9,6 +9,8 @@ sleep=0
 
 CURDIR=$(cd $(dirname $0); pwd)
 
+source $CURDIR/util.sh && check_supported_env
+
 if ! [ -x "$(command -v taskset)" ]; then
   echo "Error: taskset is not installed." >&2
   exit 1
diff --git a/scripts/util.sh b/scripts/util.sh
new file mode 100644
index 0000000..8cb402a
--- /dev/null
+++ b/scripts/util.sh
@@ -0,0 +1,7 @@
+function check_supported_env() {
+  case "$OSTYPE" in
+      linux*)   ;;
+      darwin*)  ;;
+      *)        echo "[ERROR] kitex benchmark is not supported on $OSTYPE"; exit 1;;
+  esac
+}
\ No newline at end of file