| #!/usr/bin/env bash |
| # |
| # Copyright 2016 Yahoo Inc. |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # |
| |
| usage() { |
| cat <<EOF |
| Usage: pulsar-daemon (start|stop) <command> <args...> |
| where command is one of: |
| broker Run a broker server |
| bookie Run a bookie server |
| zookeeper Run a zookeeper server |
| global-zookeeper Run a global-zookeeper server |
| discovery Run a discovery server |
| websocket Run a websocket proxy server |
| standalone Run a standalone Pulsar service |
| |
| where argument is one of: |
| -force (accepted only with stop command): Decides whether to stop the server forcefully if not stopped by normal shutdown |
| EOF |
| } |
| |
| BINDIR=$(dirname "$0") |
| PULSAR_HOME=$(cd $BINDIR/..;pwd) |
| |
| if [ -f "$PULSAR_HOME/conf/pulsar_env.sh" ] |
| then |
| . "$PULSAR_HOME/conf/pulsar_env.sh" |
| fi |
| |
| PULSAR_LOG_DIR=${PULSAR_LOG_DIR:-"$PULSAR_HOME/logs"} |
| PULSAR_ROOT_LOGGER=${PULSAR_ROOT_LOGGER:-'INFO,ROLLINGFILE'} |
| PULSAR_STOP_TIMEOUT=${PULSAR_STOP_TIMEOUT:-30} |
| PULSAR_PID_DIR=${PULSAR_PID_DIR:-$PULSAR_HOME/bin} |
| |
| if [ $# -lt 2 ] |
| then |
| echo "Error: no enough arguments provided." |
| usage |
| exit 1 |
| fi |
| |
| startStop=$1 |
| shift |
| command=$1 |
| shift |
| |
| case $command in |
| (broker) |
| echo "doing $startStop $command ..." |
| ;; |
| (bookie) |
| echo "doing $startStop $command ..." |
| ;; |
| (zookeeper) |
| echo "doing $startStop $command ..." |
| ;; |
| (global-zookeeper) |
| echo "doing $startStop $command ..." |
| ;; |
| (discovery) |
| echo "doing $startStop $command ..." |
| ;; |
| (websocket) |
| echo "doing $startStop $command ..." |
| ;; |
| (standalone) |
| echo "doing $startStop $command ..." |
| ;; |
| (*) |
| echo "Error: unknown service name $command" |
| usage |
| exit 1 |
| ;; |
| esac |
| |
| export PULSAR_LOG_DIR=$PULSAR_LOG_DIR |
| export PULSAR_ROOT_LOGGER=$PULSAR_ROOT_LOGGER |
| export PULSAR_LOG_FILE=pulsar-$command-$HOSTNAME.log |
| |
| pid=$PULSAR_PID_DIR/pulsar-$command.pid |
| out=$PULSAR_LOG_DIR/pulsar-$command-$HOSTNAME.out |
| logfile=$PULSAR_LOG_DIR/$PULSAR_LOG_FILE |
| |
| rotate_out_log () |
| { |
| log=$1; |
| num=5; |
| if [ -n "$2" ]; then |
| num=$2 |
| fi |
| if [ -f "$log" ]; then # rotate logs |
| while [ $num -gt 1 ]; do |
| prev=`expr $num - 1` |
| [ -f "$log.$prev" ] && mv "$log.$prev" "$log.$num" |
| num=$prev |
| done |
| mv "$log" "$log.$num"; |
| fi |
| } |
| |
| mkdir -p "$PULSAR_LOG_DIR" |
| |
| case $startStop in |
| (start) |
| if [ -f $pid ]; then |
| if kill -0 `cat $pid` > /dev/null 2>&1; then |
| echo $command running as process `cat $pid`. Stop it first. |
| exit 1 |
| fi |
| fi |
| |
| rotate_out_log $out |
| echo starting $command, logging to $logfile |
| pulsar=$PULSAR_HOME/bin/pulsar |
| nohup $pulsar $command "$@" > "$out" 2>&1 < /dev/null & |
| echo $! > $pid |
| sleep 1; head $out |
| sleep 2; |
| if ! ps -p $! > /dev/null ; then |
| exit 1 |
| fi |
| ;; |
| |
| (stop) |
| if [ -f $pid ]; then |
| TARGET_PID=$(cat $pid) |
| if kill -0 $TARGET_PID > /dev/null 2>&1; then |
| echo "stopping $command" |
| kill $TARGET_PID |
| |
| count=0 |
| location=$PULSAR_LOG_DIR |
| while ps -p $TARGET_PID > /dev/null; |
| do |
| echo "Shutdown is in progress... Please wait..." |
| sleep 1 |
| count=`expr $count + 1` |
| |
| if [ "$count" = "$PULSAR_STOP_TIMEOUT" ]; then |
| break |
| fi |
| done |
| |
| if [ "$count" != "$PULSAR_STOP_TIMEOUT" ]; then |
| echo "Shutdown completed." |
| fi |
| |
| if kill -0 $TARGET_PID > /dev/null 2>&1; then |
| fileName=$location/$command.out |
| $JAVA_HOME/bin/jstack $TARGET_PID > $fileName |
| echo "Thread dumps are taken for analysis at $fileName" |
| if [ "$1" == "-force" ] |
| then |
| echo "forcefully stopping $command" |
| kill -9 $TARGET_PID >/dev/null 2>&1 |
| echo Successfully stopped the process |
| else |
| echo "WARNNING : $command is not stopped completely." |
| exit 1 |
| fi |
| fi |
| else |
| echo "no $command to stop" |
| fi |
| rm $pid |
| else |
| echo no "$command to stop" |
| fi |
| ;; |
| |
| (*) |
| usage |
| exit 1 |
| ;; |
| esac |