#! /bin/bash

# 端口号
PORTS=(9054 9051 9056 9052 9055 9059 12010 9057 9060)
# 模块
MODULES=(DataMiniosService HealthyService IntegrationService ManagementService ProductService SystemService TheGateway ShowService AlgorithmService)
# 模块名称
MODULE_NAMES=(数据微服务 健康服务 集成服务 管理服务 产品服务 系统服务 网关服务 大屏服务 算法服务)
# jar包数组
JARS=(DataMiniosService-1.0-SNAPSHOT.jar HealthyService-1.0-SNAPSHOT.jar IntegrationService-1.0-SNAPSHOT.jar ManagementService-1.0-SNAPSHOT.jar ProductService-1.0-SNAPSHOT.jar SystemService-1.0-SNAPSHOT.jar egateway-1.0-SNAPSHOT.jar ShowService-1.0-SNAPSHOT.jar AlgorithmService-1.0-SNAPSHOT.jar)
# jar包路径
JAR_PATH='/opt/myapps/e01'
# 日志路径
LOG_PATH='/opt/myapps/e01/logs'

start() {
    local MODULE=
    local MODULE_NAME=
    local JAR_NAME=
    local command="$1"
    local commandOk=0
    local count=0
    local okCount=0
    local port=0
    for((i=0; i<${#MODULES[@]}; i++))
    do
        MODULE=${MODULES[$i]}
        MODULE_NAME=${MODULE_NAMES[$i]}
        JAR_NAME=${JARS[$i]}
        PORT=${PORTS[$i]}
        if [ "$command" == "all" ] || [ "$command" == "$MODULE" ]; then
            commandOk=1
            count=0
            PID=$(ps -ef | grep $(echo $JAR_NAME | awk -F/ '{print $NF}') | grep -v grep | awk '{print $2}')
            if [ -n "$PID" ]; then
                echo "$MODULE---$MODULE_NAME:已经运行,PID=$PID"
            else
                nohup java -Xms256m -Xmx512m -jar $JAR_PATH/$JAR_NAME > $LOG_PATH/$MODULE.log 2>&1 &
                PID=$(ps -ef | grep $(echo $JAR_NAME | awk -F/ '{print $NF}') | grep -v grep | awk '{print $2}')
                while [ -z "$PID" ]
                do
                    if [ $count -eq 30 ]; then
                        echo "$MODULE---$MODULE_NAME:$(($count * 10))秒内未启动,请检查!"
                        break
                    fi
                    count=$(($count+1))
                    echo "$MODULE_NAME启动中.................."
                    sleep 10
                    PID=$(ps -ef | grep $(echo $JAR_NAME | awk -F/ '{print $NF}') | grep -v grep | awk '{print $2}')
                done
                okCount=$(($okCount+1))
                echo "$MODULE---$MODULE_NAME:已经启动成功,PID=$PID"
            fi
        fi
    done
    if [ $commandOk -eq 0 ]; then
        echo "第二个参数输入错误"
    else
        echo "............本次共启动:$okCount个服务..........."
    fi
}

stop() {
    local MODULE=
    local MODULE_NAME=
    local JAR_NAME=
    local command="$1"
    local commandOk=0
    local okCount=0
    for((i=0; i<${#MODULES[@]}; i++))
    do
        MODULE=${MODULES[$i]}
        MODULE_NAME=${MODULE_NAMES[$i]}
        JAR_NAME=${JARS[$i]}
        if [ "$command" == "all" ] || [ "$command" == "$MODULE" ]; then
            commandOk=1
            PID=$(ps -ef | grep $(echo $JAR_NAME | awk -F/ '{print $NF}') | grep -v grep | awk '{print $2}')
            if [ -n "$PID" ]; then
                echo "$MODULE---$MODULE_NAME:准备结束,PID=$PID"
                kill -9 $PID
                PID=$(ps -ef | grep $(echo $JAR_NAME | awk -F/ '{print $NF}') | grep -v grep | awk '{print $2}')
                while [ -n "$PID" ]
                do
                    sleep 3
                    PID=$(ps -ef | grep $(echo $JAR_NAME | awk -F/ '{print $NF}') | grep -v grep | awk '{print $2}')
                done
                echo "$MODULE---$MODULE_NAME:成功结束"
                okCount=$((okCount + 1))
            else
                echo "$MODULE---$MODULE_NAME:未运行"
            fi
        fi
    done
    if [ $commandOk -eq 0 ]; then
        echo "第二个参数输入错误"
    else
        echo "............本次共停止:$okCount个服务............"
    fi
}

case "$1" in
    start)
        start "$2"
        ;;
    stop)
        stop "$2"
        ;;
    restart)
        stop "$2"
        sleep 3
        start "$2"
        ;;
    *)
        echo "第一个参数请输入:start|stop|restart"
        exit 1
        ;;
esac