bin/chaos-daemon.sh

#!/usr/bin/env bash
#
#/**
# * Licensed to the Apache Software Foundation (ASF) under one
# * or more contributor license agreements.  See the NOTICE file
# * distributed with this work for additional information
# * regarding copyright ownership.  The ASF licenses this file
# * to you 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="Usage: chaos-daemon.sh (start|stop) chaosagent"

# if no args specified, show usage
if [ $# -le 1 ]; then
  echo "$usage"
  exit 1
fi

# get arguments
startStop=$1
shift

command=$1
shift

check_before_start(){
    #ckeck if the process is not running
    mkdir -p "$HBASE_PID_DIR"
    if [ -f "$CHAOS_PID" ]; then
      if kill -0 "$(cat "$CHAOS_PID")" > /dev/null 2>&1; then
        echo "$command" running as process "$(cat "$CHAOS_PID")".  Stop it first.
        exit 1
      fi
    fi
}

bin=`dirname "${BASH_SOURCE-$0}"`
bin=$(cd "$bin">/dev/null || exit; pwd)

. "$bin"/hbase-config.sh
. "$bin"/hbase-common.sh

CLASSPATH=$HBASE_CONF_DIR
for f in ../lib/*.jar; do
  CLASSPATH=${CLASSPATH}:$f
done

# get log directory
if [ "$HBASE_LOG_DIR" = "" ]; then
  export HBASE_LOG_DIR="$HBASE_HOME/logs"
fi

if [ "$HBASE_PID_DIR" = "" ]; then
  HBASE_PID_DIR=/tmp
fi

if [ "$HBASE_IDENT_STRING" = "" ]; then
  export HBASE_IDENT_STRING="$USER"
fi

if [ "$JAVA_HOME" != "" ]; then
  #echo "run java in $JAVA_HOME"
  JAVA_HOME=$JAVA_HOME
fi
if [ "$JAVA_HOME" = "" ]; then
  echo "Error: JAVA_HOME is not set."
  exit 1
fi

export HBASE_LOG_PREFIX=hbase-$HBASE_IDENT_STRING-$command-$HOSTNAME
export CHAOS_LOGFILE=$HBASE_LOG_PREFIX.log

if [ -z "${HBASE_ROOT_LOGGER}" ]; then
export HBASE_ROOT_LOGGER=${HBASE_ROOT_LOGGER:-"INFO,RFA"}
fi

if [ -z "${HBASE_SECURITY_LOGGER}" ]; then
export HBASE_SECURITY_LOGGER=${HBASE_SECURITY_LOGGER:-"INFO,RFAS"}
fi

CHAOS_LOGLOG=${CHAOS_LOGLOG:-"${HBASE_LOG_DIR}/${CHAOS_LOGFILE}"}
CHAOS_PID=$HBASE_PID_DIR/hbase-$HBASE_IDENT_STRING-$command.pid

if [ -z "$CHAOS_JAVA_OPTS" ]; then
  CHAOS_JAVA_OPTS="-Xms1024m -Xmx4096m"
fi

case $startStop in

(start)
    check_before_start
    echo running $command
    CMD="${JAVA_HOME}/bin/java -Dapp.home=${HBASE_CONF_DIR}/../  ${CHAOS_JAVA_OPTS} -cp ${CLASSPATH} org.apache.hadoop.hbase.chaos.ChaosService -$command start &>> ${CHAOS_LOGLOG} &"

    eval $CMD
    PID=$(echo $!)
    echo ${PID} >${CHAOS_PID}

    echo "Chaos ${1} process Started with ${PID} !"
    now=$(date)
    echo "${now} Chaos ${1} process Started with ${PID} !" >>${CHAOS_LOGLOG}
    ;;

(stop)
    echo stopping $command
    if [ -f $CHAOS_PID ]; then
      pidToKill=`cat $CHAOS_PID`
      # kill -0 == see if the PID exists
      if kill -0 $pidToKill > /dev/null 2>&1; then
        echo -n stopping $command
        echo "`date` Terminating $command" >> $CHAOS_LOGLOG
        kill $pidToKill > /dev/null 2>&1
        waitForProcessEnd $pidToKill $command
      else
        retval=$?
        echo no $command to stop because kill -0 of pid $pidToKill failed with status $retval
      fi
    else
      echo no $command to stop because no pid file $CHAOS_PID
    fi
    rm -f $CHAOS_PID
  ;;

(*)
  echo $usage
  exit 1
  ;;

esac