From b44e3316596994de74d957e9eecabb4ee0c7bcbd Mon Sep 17 00:00:00 2001
From: naiba <hi@nai.ba>
Date: Sat, 26 Dec 2020 17:58:10 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20chore:=20=E5=A2=9E=E5=BC=BA?=
 =?UTF-8?q?=E8=84=9A=E6=9C=AC=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 script/install.sh | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/script/install.sh b/script/install.sh
index edb1202..2009dd3 100755
--- a/script/install.sh
+++ b/script/install.sh
@@ -380,6 +380,16 @@ uninstall_agent() {
     fi
 }
 
+restart_agent() {
+    echo -e "> 重启Agent"
+
+    systemctl restart nezha-agent.service
+
+    if [[ $# == 0 ]]; then
+        before_show_menu
+    fi
+}
+
 clean_all() {
     if [ -z "$(ls -A ${NZ_BASE_PATH})" ]; then
         rm -rf ${NZ_BASE_PATH}
@@ -401,6 +411,7 @@ show_usage() {
     echo "./nbdomain.sh install_agent              - 安装监控Agent"
     echo "./nbdomain.sh modify_agent_config        - 修改Agent配置"
     echo "./nbdomain.sh uninstall_agent            - 卸载Agen"
+    echo "./nbdomain.sh restart_agent              - 重启Agen"
     echo "--------------------------------------------------------"
 }
 
@@ -421,8 +432,9 @@ show_menu() {
     ${green}8.${plain}  安装监控Agent
     ${green}9.${plain}  修改Agent配置
     ${green}10.${plain} 卸载Agent
+    ${green}11.${plain} 重启Agent
     "
-    echo && read -p "请输入选择 [0-8]: " num
+    echo && read -p "请输入选择 [0-11]: " num
 
     case "${num}" in
     0)
@@ -458,8 +470,11 @@ show_menu() {
     10)
         uninstall_agent
         ;;
+    11)
+        restart_agent
+        ;;
     *)
-        echo -e "${red}请输入正确的数字 [0-7]${plain}"
+        echo -e "${red}请输入正确的数字 [0-11]${plain}"
         ;;
     esac
 }
@@ -498,6 +513,9 @@ if [[ $# > 0 ]]; then
     "uninstall_agent")
         uninstall_agent 0
         ;;
+    "restart_agent")
+        restart_agent 0
+        ;;
     *) show_usage ;;
     esac
 else