Linux下Shell脚本实现统一管理服务启停重启

#!/bin/bash

 

###################################

#                                 #

#       OWNER: 一袭白衣一         #

#                                 #

#       TIME:2024/8               #

#                                                                 #

#       version:1.0               #

#                                 #

###################################

 
#清屏，好看点，不多说了哈
clear
#声明个数组存储数据
declare -A services

 

#数组的样例哈
services=(
  ["服务1"]="服务所在的路径（到启动脚本所在的那一级）,服务启动脚本所在路径（包含了启动脚本）,服务名（进程里显示的那个）,服务停止命令,打印日志所在路径（包含日志文件）"
  ["服务2"]="服务所在的路径（到启动脚本所在的那一级）,服务启动脚本所在路径（包含了启动脚本）,服务名（进程里显示的那个）,服务停止命令,打印日志所在路径（包含日志文件）"
  ["服务3"]="服务所在的路径（到启动脚本所在的那一级）,服务启动脚本所在路径（包含了启动脚本）,服务名（进程里显示的那个）,服务停止命令"

#后边可以放多个服务信息

)

#这块不多说了哈，就是去数组里获取启动脚本路径，调用启动命令启动服务
start_service() {
#声明一个变量从数组内拿相关信息，用于后边调用
  local service_path=$(echo ${services[$1]} | cut -d',' -f1)
  local start_script=$(echo ${services[$1]} | cut -d',' -f2)

  echo "Starting $1 at $service_path......"
  cd $service_path
  $start_script start || { echo "Failed to start $1"; return 1; }
}

 
#停止服务的模块，这部分有点特殊，我这边是因为有个服务的停止命令不是stop，是shutdown，所以做了个判断
#如果你们也有这样的情况，可以在这个地方做修改，没有的话，可以把中间的这个if拿掉
stop_service() {
#声明一个变量从数组内拿相关信息，用于后边调用
  local service_path=$(echo ${services[$1]} | cut -d',' -f1)
  local stop_script=$(echo ${services[$1]} | cut -d',' -f2)
  local service_name=$(echo ${services[$1]} | cut -d',' -f3)
  echo "Stopping $1 with: $service_path......"
  cd $service_path
#这个if模块不需要可以拿掉
  if [ "$service_name" == "特殊的服务" ];
  then
  $stop_script shutdown || { echo "Failed to stop $1"; return 1; }
else
  $stop_script stop || { echo "Faild to stop $1"; return 1; }
        fi
}

#重启模块了哈，这个就更简单， 我就是重复调用了一遍stop、start模块
restart_service() {

  local service_name=$1
#如果服务状态正常，就先停止，再启动，否则就直接启动
  if check_service_status $service_name; then
    stop_service $service_name
  fi
  start_service $service_name

}

#检查服务的状态，我这边用的是ps命令。直接查看进程里的服务是否存在，判断服务是否存活
check_service_status() {

  local process_name=$(echo ${services[$1]} | cut -d',' -f3)

  if pgrep -u $(whoami) -f "$process_name" > /dev/null; then
    echo "$1 is already running."

    return 0

  else

    echo "$1 is not running."

    return 1

  fi

}

 
#这块也是特殊服务用到的，有个服务启动的时间太长了所以必须要打印日志。。。。
#如果想让所有的服务都打印日志，可以在下边去掉判断，我会标注
#tail appserver&dbbackup log after them start

watch_log(){

   local service_name=$1

   local log_file=$(echo ${services[$service_name]} | cut -d',' -f5)

  if [ -n "$log_file" ];

   then

echo "Now Show log for $service_name: $log_file"

trap " echo 'Now Stop Show log for $service_name'; return" SIGINT

tail -f $log_file

else

  echo "No  log to show "

fi

}


#服务的list，用于check服务状态
service_processes_list() {

  for service in "${!services[@]}"; do
    echo "Showing the service processes for $service"
  check_service_status $service

  done

}

 
#主菜单展示，linux 的脚本就这样，丑点就丑点，好用就行
while true; do

  echo "#################################"

 

  echo "#****Services Control Menu******#"

 

  echo "#                               #"

 

  echo "#    1. Start Server            #"

 

  echo "#    2. Stop Server             #"

 

  echo "#    3. Restart Server          #"

 

  echo "#    4. Check Service status    #"

 

  echo "#    5. Exit menu               #"

 

  echo "#################################"

 

  read -p "Please select an option: " option

 
#进入实际的菜单功能模块了哈
  case $option in

    1)

      while true; do

        echo "Select a service to start:"

        select service in "${!services[@]}" "Back to main menu"; do

          case $service in

            "Back to main menu")

              break 2

              ;;

            *)

              if [ -n "$service" ]; then

                start_service $service

                sleep 3
#这个地方就是我说的特殊处理的，这两个特殊服务需要打印日志，不能仅通过进程判断是否启动正常
#如果想打印所有服务的日志，可以把这块的判断拿掉
                if [[ "$service" == "特殊服务1" || "$service" == "特殊服务2" ]];

                then

                 watch_log $service

                fi

                echo "Press any key to continue..."

                read -n 1 -s

                break

              else

                echo "Invalid option, please try again."


              fi

              ;;

          esac

        done

      done

      ;;

    2)

      while true; do

        echo "Select a service to stop:"

        select service in "${!services[@]}" "Back to main menu"; do

          case $service in

            "Back to main menu")

              break 2

              ;;

            *)

              if [ -n "$service" ]; then

                if check_service_status $service; then
#停止服务前，做了个提示，省的瞎搞误操作把生产环境服务停了。。。
                read -p "Are you sure want to stop $service ? (Y/N):" confirm

                if [ "$confirm" == "Y" ];

                then

                  stop_service $service

                sleep 3

                fi

                else

                  echo "$service is not running."

                fi

                echo "Press any key to continue..."

                read -n 1 -s

                break

              else

                echo "Invalid option, please try again."

              fi

              ;;

          esac

        done

      done

      ;;

    3)

    while true; do

        echo "Select a service to restart:"

        select service in "${!services[@]}" "Back to main menu"; do

          case $service in

            "Back to main menu")

              break 2

              ;;

            *)

              if [ -n "$service" ]; then

                restart_service $service

                sleep 3
#这个地方跟上边一样，就是专门给特殊服务做的，不需要可以拿掉
                if [[ "$service" == "特殊服务1" || "$service" == "特殊服务2" ]];

                then

                watch_log $service

                fi

                echo "Press any key to continue..."

                read -n 1 -s

                break

              else

                echo "Invalid option, please try again."

              fi

              ;;

          esac

        done

      done

      ;;

    4)

      service_processes_list

        sleep 3

      echo "Press any key to continue..."

      read -n 1 -s

      ;;

    5)

      exit 0

      ;;

    *)

      echo "Invalid option, please try again."

      ;;

  esac

done