Tomcat 안전 셧다운 및 재시작 스크립트

#!/bin/bash 
 
###################################################################
# Script Name  : Tomcat Restart
# Description  : 톰캣 프로세스를 자동으로 안전하게 킬하고 다시 띄운다. 먼저 SIGTERM 5회를 시도하고, 그래도 종료되지 않으면 SIGKILL (-9)을 보낸다.
###################################################################
 
check_root() {
    # root 사용자로 톰캣을 띄우는 보안 취약점 방지
    if [ "${UID}" -eq 0 ]
    then
        echo "You are root. Can not run tomcat as root user."
        exit 1
    fi
}
 
tomcat_pid() {
    # 시스템에 여러 톰캣이 존재할 경우를 대비한 유니크 키 추출. (예시: /web 디렉토리를 포함한 프로세스 식별)
    echo `ps aux | grep org.apache.catalina.startup.Bootstrap | grep /web | grep -v grep | awk '{ print $2 }'`
}
 
start() {
    echo -e "\e[92mStarting Tomcat\e[0m"
    "$TOMCAT_PATH"/startup.sh >& /dev/null
    sleep 2
    PID=$(tomcat_pid)
    echo -e "Tomcat is running with pid: \e[96m$PID\e[0m"
}
 
stop() {
    echo -e "\e[93mShutdown Tomcat\e[0m"
    "$TOMCAT_PATH"/shutdown.sh >& /dev/null
    sleep 2 # 기본 2초 대기
 
    # 1차: 정상 종료 확인 (5회 반복)
    for i in {1..5}
    do
        PID=$(tomcat_pid)
        if [ -z "$PID" ] ; then
            echo -e "\e[91mShutdown completed.\e[0m"
            return
        else
            sleep 2
        fi
    done
 
    # 2차: 프로세스가 안 죽었으면 일반 kill (SIGTERM)
    PID=$(tomcat_pid)
    if [ -n "$PID" ] ; then
        echo "Kill process..."
        for i in {1..2}
        do
            PID=$(tomcat_pid)
            if [ -z "$PID" ] ; then
                echo -e "\e[91mShutdown completed.\e[0m"
                return
            else
                killProc
                sleep 2
            fi
        done
    fi
        
    # 3차: 그래도 살아있으면 강제 킬 (SIGKILL -9)
    if [ -n "$PID" ] ; then
        echo "Force kill process..."
        killProc 9
    fi
 
    sleep 2
 
    PID=$(tomcat_pid)
    if [ -z "$PID" ] ; then
        echo -e "\e[91mShutdown completed.\e[0m"
        return
    fi
}
 
# 프로세스 종료 래퍼 함수
killProc() {
    local signal="TERM"
 
    if [ "$#" = 1 ] ; then
        signal=$1
    fi
 
    if [ ! -z ${PID} ] && [ ${PID} -gt 0 ] ; then
        echo "kill -${signal} ${PID}"
        kill -${signal} ${PID};
    fi
}
 
# 메인 로직 시작
PID=$(tomcat_pid)
TOMCAT_PATH="$(cd "$(dirname "$0")" && pwd)"
 
if [ -n "$PID" ] ; then
    echo -e "Tomcat is running with pid: \e[96m$PID\e[0m"
    echo -e "Tomcat bin location is \e[94m$TOMCAT_PATH\e[0m"
    stop
    start
else
    echo "Tomcat is not running"
    echo -e "Tomcat bin location is \e[94m$TOMCAT_PATH\e[0m"
    start
fi