前置知识

  1. $0:这个程式的执行名字
  2. $n:这个程式的第n个参数，n=1...9
  3. $*:这个程式的所有参数，此选项参数可超过9个
  4. $#:这个程式的参数个数
  5. $$:这个程式的PID（脚本运行的当前进程ID号）
  6. $!:执行上一个背景指令的PID（后台运行的最后一个进程的进程ID号）
  7. $?:执行上一个指令的返回值（显示最后命令的退出状态，0表示没有错误，其他任何值表明有错误）
  8. $-:显示shell使用的当前选项，与set命令功能相同
  9. @跟*类似，但是可以当做数组用
  1. #!/bin/bash
  2. echo "######starting run service.sh######"
  3. # 下面判断程式如果没有参数，退出
  4. if [ $# -lt 1 ]; then
  5.     echo "USAGE: $0 classname opts"
  6.     exit 1
  7. fi
  9. BASE_DIR=$(dirname $0)
  10. LIB_DIR="$BASE_DIR/lib"
  11. CONF_DIR="$BASE_DIR/conf"
  12. #请输入应用名，注意不要有空格
  13. SERVICE_NAME="appname"
  14. if [ "$SERVICE_NAME" == "" ]; then
  15.     echo "变量SERVICE_NAME值为空，有可能会影响程序停服和启服"
  16. fi
  18. if [ ! -d $LIB_DIR ]; then
  19.     ## 本段适用于springboot打包的应用
  20.     JAR_PATH=`ls *.jar`
  21.     KEYWORD="$JAR_PATH"
  22.     PID_FILE="$BASE_DIR/../$JAR_PATH.pid"
  23. else
  24.   echo "您的应用可能不属于springboot类型，如已自定义，请忽略此提示"
  25.   ## 下面一段脚本适用于jar包打在工程目录lib文件夹的应用##
  26.   #请填写关键词（main方法所在的类路径）
  27.   KEYWORD="你的服务包名"
  28.   PID_FILE="$BASE_DIR/../$SERVICE_NAME.pid"
  29. fi
  30. #请指定jdk版本
  31. JAVA_VERSION=1.8
  32. if [ "$JAVA_VERSION" ]; then
  33.     JDK_PATH=/usr/local/jdk8
  34. else
  35.   JDK_PATH=/usr/local/jdk
  36. fi
  38. function test_java_version() {
  39.     java_version=`$1/java -version 2>&1 |awk 'NR==1{print $3}' |sed 's/\''//g'`
  40.     if [[ ! "$java_version" =~ "$JAVA_VERSION" ]]; then
  41.         return 1
  42.     fi
  43.     return 0
  44. }
  46. if [ -z "$JAVA_HOME" ]; then
  47.     JAVA_HOME=$JDK_PATH
  48. fi
  50. if ! test_java_version "$JAVA_HOME/bin" ; then
  51.     if [[ ! -d "$JDK_PATH" ]]; then
  52.         echo "$JDK_PATH not exist!"
  53.         exit 2
  54.     fi
  56.     if test_java_version "$JDK_PATH/bin" ; then
  57.         export JAVA_HOME="$JDK_PATH"
  58.     else
  59.       echo "java version not match"
  60.       exit 3
  61.     fi
  62. fi
  64. JAVA_HOME=$JDK_PATH
  65. #请设置应用的JVM环境参数，此处仅为参考，注意点：MaxMetaspaceSize是jdk1.8版本的参数，1.7的改成permsize
  66. XMS=${JVM_XMS:-4g}
  67. XMX=${JVM_XMX:-4g}
  68. XMN=${JVM_XMN:-2g}
  69. JAVA_OPTS="$JAVA_OPTS -server -Xms$XMS -Xmx$XMX -Xmn$XMN -XX:MaxMetaspaceSize=512m -XX:SurvivorRatio=8"
  70. JAVA_OPTS="$JAVA_OPTS -XX:+UseConcMarkSweepGC -XX:CMSInitiatingOccupancyFraction=70 -XX:+ScavengeBeforeFullGC -XX:CMSScavengeBeforeRemark"
  71. JAVA_OPTS="$JAVA_OPTS -XX:DisableExplicitGC -XX:OmitStackTraceInFastThrow"
  72. JAVA_OPTS="$JAVA_OPTS -XX:PrintGCDateStamps -verbose:gc -XX:+PrintGCDetails -xloggc:${HOME}/service_gc_`date +%Y%m%d%H%M%S`.log"
  73. JAVA_OPTS="$JAVA_OPTS -XX:+HeapDumpOnOutOfMemoryError -XX:HeadDumpPath=${HOME}/service_heap_dump_`date +%Y%m%d%H%M%S`.hprof"
  74. JAVA="$JAVA_HOME/bin/java"
  75. ###skyWalking hook ine ###
  76. if [ ! -d $LIB_DIR ]; then
  77.     ## 本段适用于springboot打包的uberjar应用
  78.     JAVA_OPTS="$JAVA_OPTS -Djava.ext.dirs=${JAVA_HOME}/jre/lib/ext:./ext:./external"
  79.     JAVA_OPTS="$JAVA_OPTS -jar" # only applies to springboot's uberjar
  80. else
  81.   #############################################################################################
  82.   ## 下面一段脚本适用于jar包打在工程根目录lib文件夹的应用
  83.   JAVA_OPTS="$JAVA_OPTS -Djava.ext.dirs=${JAVA_HOME}/jre/lib/ext::${LIB_DIR}"
  84.   CLASSPATH=${BASE_DIR}/conf:${BASE_DIR}/lib/\*
  85.   JAVA_OPTS="${JAVA_OPTS} -cp ${CLASSPATH}"
  86. fi
  88. ## 设置起服停服检查最大时间设置
  89. startchecktimeM="120"
  90. stopchecktimeM="120"
  91. if [ "$startchecktimeM" == "" ]; then
  92.     echo "起服时间默认最大是120s，也可以在模板自定义"
  93.     startchecktimeM="120"
  94. fi
  95. if [ "$startchecktimeM" == "" ]; then
  96.     echo "停服时间默认最大是120s，也可以在模板自定义"
  97.     stopchecktimeM="120"
  98. fi
  100. function startService() {
  101.     local server_type="$1"
  102.     getServicePID
  103.     if [ $? -eq 0 ]; then echo "$SERVICE_NAME is already running."; RETVAL=0; return 0; fi
  104.     echo -n "Starting $SERVICE_NAME..."
  105.     startServiceProcess
  106.     if [ $? -ne 0 ]; then RETVAL=1; echo "$SERVICE_NAME start failed,see nohup.log."; return 1; fi
  107.     COUNT=0
  108.     if [ "$server_type" == "provider" ]; then
  109.         the_result=1
  110.         while [ $COUNT -1t 1 ]; do
  111.             #provider起服轮询时间最大值默认为120s
  112.             for (( i = 0; i < $startchecktimeM; i++ )); do
  113.                 startPortCheck $pid
  114.                 if [ $? -ne 0 ]; then
  115.                     echo -e ".\c"
  116.                     sleep 1;
  117.                 else
  118.                   the_result=0
  119.                   echo "start cost $i s"
  120.                   break;
  121.                 fi
  122.             done
  123.           break
  124.         done
  125.     else
  126.       ## assuming standdalone dubbo server,we will check if
  127.       the_result=1
  128.       while [ $COUNT -1t 1 ]; do
  129.           # consume起服轮询时间最大值默认为120s
  130.           for (( i = 0; i < $startchecktimeM; i++ )); do
  131.               STR=`grep "server started" nohup.log`
  132.               if [ ! -z "$STR" ]; then
  133.                   echo "PID=$pid"
  134.                   echo "Server start OK in $i seconds"
  135.                   the_result=0
  136.                   break;
  137.               fi
  138.               echo -e ".\c"
  139.               sleep 1
  140.           done
  141.         break
  142.       done
  143.     fi
  144.     echo $the_result
  145.     RETVAL=$the_result
  146.     if [ $the_result -eq 1 ]; then echo "The service is gone or port is unnormal,please check!" fi
  147.     return $RETVAL;
  148. }
  150. function startServiceProcess() {
  151.     touch $PID_FILE
  152.     rm -rf nohup.log
  153.     nohup $JAVA $JAVA_OPTS $KEYWORD >> nohup.log 2>&1 & echo $! > $PID_FILE
  154.     sleep 1
  155.     pid="$(<$PID_FILE)"
  156.     if checkProcessIsRunning $pid; then :; else
  157.       echo "$SERVICE_NAME start failed,see nohup.log"
  158.       return 1
  159.     fi
  160.     return 0;
  161. }
  163. # return 0 if port is released successfully when starting
  164. function startPortCheck() {
  165.     if [ ! -f $PID_FILE ]; then
  166.         echo "Error:can't find $PID_FILE when starting port check!";
  167.         return 1;
  168.     fi
  169.     pid="$(<$PID_FILE)"
  170.     app_ports=`netstat -lntp 2> /dev/null|grep $pid |awk '{print $4}' |awk -F ':' '{print $NF}'`
  171.     for m in ${app_ports}
  172.     do
  173.       echo "$SERVICE_NAME is ok and ports as $m"
  174.       result=`/opt/nc/bin/nc -zw 2 127.0.0.1 $m | grep successd |wc -l`
  175.       if [[ $result == 1 ]]; then
  176.           echo "$m port check success"
  177.           continue
  178.       else
  179.         break
  180.       fi
  181.     done
  182.     if [[ $result == 1 ]]; then
  183.         echo "please wait for some seconds,it's checking the status of dubbo service"
  184.         return 0;
  185.     fi
  186.     echo "$SERVICE_NAME dubbo provider check fail"
  187.     return 1;
  188. }
  190. #returns 0 if process with PID $1 id running
  191. function checkProcessIsRunning() {
  192.     local pid="$1"
  193.     ps -ef | grep java | grep $pid | grep "$KEYWORD" | grep -q --binary -F java
  194.     if [ $? -ne 0 ]; then
  195.         echo "Process $pid is running or could not be found."
  196.         return 1;
  197.     fi
  198.     return 0;
  199. }
  201. # Returns 0 when the service is running and sets the variable $pid to the PID
  202. function getServicePID() {
  203.     if [ ! -f $PID_FILE ]; then
  204.         return 1;
  205.     fi
  206.     pid="$(<$PID_FILE)"
  207.     checkProcessIsRunning $pid || return 1
  208.     return 0;
  209. }
  211. function stopService() {
  212.     getServicePID
  213.     if [ $? -ne 0 ]; then
  214.         echo "$SERVICE_NAME is not running.";
  215.         RETVAL=0;
  216.         return 0;
  217.     fi
  218.     processcheck
  219.     getPortFile
  220.     if [ $? -ne 0 ]; then
  221.         echo "can't find portFile."
  222.         return 1;
  223.     fi
  224.     echo "Stopping $SERVICE_NAME..."
  225.     stopServiceProcess
  226.     if [ $? -ne 0 ]; then
  227.         RETVAL=1;
  228.         echo "failed."
  229.         return 1;
  230.     fi
  231.     for (( i = 0; i < 10; i++ )); do
  232.         portReleaseCheck
  233.         if [ $? -ne 0 ]; then
  234.             echo "the port is not released successfully"
  235.             sleep 1
  236.         else
  237.           echo "all ports are released successfully"
  238.           return 0
  239.           break;
  240.         fi
  241.     done
  242.     RETVAL=0
  243.     return 0;
  244. }
  246. function stopServiceProcess() {
  247.     STOP_DATE=`date +%Y%m%d%H%M%S`
  248.     kill $pid || return 1
  249.     # 停服最大时间默认为120s
  250.     for (( i = 0; i < stopchecktimeM; i++ )); do
  251.         checkProcessIsRunning $pid
  252.         if [ $? -ne 0 ]; then
  253.             rm -f $PID_FILE
  254.             return 0
  255.         fi
  256.         sleep 1
  257.     done
  258.     echo "\n $SERVICE_NAME did not terminate within 120 seconds,sending SIGKILL..."
  259.     kill -s KILL $pid
  260.     local killWaitTime=15
  261.     for (( i = 0; i < 10; i++ )); do
  262.         checkProcessIsRunning $pid
  263.         if [ $? -ne 0 ]; then
  264.             rm -f $PID_FILE
  265.             return 0
  266.         fi
  267.         sleep 1
  268.     done
  269.     echo "Error: $SERVICE_NAME could not be stopped within 120 + 10 seconds!"
  270.     return 1;
  271. }
  273. function getPortFile() {
  274.     port=`netstat -lntp 2> /dev/null |egrep -w "$pid\/" |awk '{print $4}' |awk -F ":" '{print $NF}'`
  275.     if [ $? -ne 0 ]; then
  276.         echo "WARNING: can't find service port when getPortFile"
  277.         return 1;
  278.     fi
  279.     echo $port
  280.     return 0;
  281. }
  283. #进程数目以及起服账号检查
  284. function processcheck() {
  285.     piduser=`ps -ef | grep $pid | awk '{print $1}'`
  286.     pidnum=`ps -ef | grep $pid | grep java | awk '{print $2}'`
  287.     echo -e "=============================piduser=$piduser pidnum=$pidnum================\n"
  288.     if [ "$piduser" == "root" ]; then
  289.         echo -e "WARNING: process using root,please contact SCM!\n"
  290.         exit -1;
  291.     fi
  292. }
  294. #return 0 if all ports are released successfully when stopping
  295. function portReleaseCheck() {
  296.     sleep 3
  297.     port_check_result=0
  298.     for n in $port;do
  299.       echo "port as $n,wait to release"
  300.       port_check=`netstat -nltup 2> /dev/null | awk '{print $4}' | awk -F ":" '{if ($NF == "'$n''")print $NF}' | sort -u`
  301.       echo "the number of unreleased ports:$port_check"
  302.       if [[ $prot_check -eq 0 ]]; then
  303.         echo "this port:$n is released successfully"
  304.       else
  305.         echo "this port is not released"
  306.         port_check_result=1
  307.         return 1
  308.       fi
  309.     done
  310.     return 0;
  311. }
  313. function checkServiceStatus() {
  314.     echo -n "checking for $SERVICE_NAME:"
  315.     if getServicePID; then
  316.       echo "running PID=$pid"
  317.       RETVAL=0
  318.     else
  319.       echo "stopped"
  320.       RETVAL=3
  321.     fi
  322.     return 0;
  323. }
  325. function main() {
  326.     RETVAL=0
  327.     case "$1" in
  328.     start)
  329.       startService "$2"
  330.       ;;
  331.     stop)
  332.       stopService
  333.       ;;
  334.     restart)
  335.       stopService && startService "$2"
  336.       ;;
  337.     status)
  338.       checkServiceStatus
  339.       ;;
  340.     *)
  341.       echo "Usage: $0 {start|stop|restart|status} {provider}"
  342.       exit 1
  343.       ;;
  344.     esac
  345.     exit $RETVAL
  346. }
  348. main "$1" "$2"