From 56ebbf8bc97fb6899caa91ec182fd83ee612bd73 Mon Sep 17 00:00:00 2001
From: ZhangYu0123 <67053339+ZhangYu0123@users.noreply.github.com>
Date: Wed, 22 Feb 2023 19:34:40 +0800
Subject: [PATCH] [chore](tools) fix load-clickbench-data script cannot be
 interrupted #17000

---
 tools/clickbench-tools/load-clickbench-data.sh | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/tools/clickbench-tools/load-clickbench-data.sh b/tools/clickbench-tools/load-clickbench-data.sh
index 53d6d61b4b..dad68720b3 100755
--- a/tools/clickbench-tools/load-clickbench-data.sh
+++ b/tools/clickbench-tools/load-clickbench-data.sh
@@ -86,6 +86,8 @@ check_prerequest "wget --version" "wget"
 
 source $CURDIR/conf/doris-cluster.conf
 
+wget_pids=()
+
 echo "FE_HOST: $FE_HOST"
 echo "FE_HTTP_PORT: $FE_HTTP_PORT"
 echo "USER: $USER"
@@ -113,6 +115,8 @@ function load() {
             echo "will download hits_split${i} to $DATA_DIR"
             wget --continue "https://doris-test-data.oss-cn-hongkong.aliyuncs.com/ClickBench/hits_split${i}" &
             # wget --continue "https://doris-test-data.oss-cn-hongkong-internal.aliyuncs.com/ClickBench/hits_split${i}" &
+            PID=$!
+            wget_pids[${#wget_pids[@]}]=$PID
         fi
     done
 
@@ -132,6 +136,15 @@ function load() {
     done
 }
 
+function signal_handler() {
+
+    for PID in ${wget_pids[@]}; do
+        kill -9 $PID
+    done
+}
+
+trap signal_handler 2 3 6 15
+
 echo "start..."
 start=$(date +%s)
 check_doirs_conf