p4broker_base #3

#!/bin/bash
#==============================================================================
# Copyright and license info is available in the LICENSE file included with
# the Server Deployment Package (SDP), and also available online:
# https://swarm.workshop.perforce.com/projects/perforce-software-sdp/view/main/LICENSE
#------------------------------------------------------------------------------

# Helix Broker base init script

# Do nothing unless $SDP_INSTANCE is defined.
export SDP_INSTANCE=${1:-Unset}

if [[ $SDP_INSTANCE == Unset ]]; then
   echo -e "\nError: The SDP_INSTANCE is not defined.\n."
   exit 1
fi

# Load SDP controlled shell environment.
source /p4/common/bin/p4_vars "$SDP_INSTANCE"
if [[ $? -ne 0 ]]; then
   echo -e "\nError: Failed to load SDP environment for instance $SDP_INSTANCE.\n"
   exit 1
fi

source $P4CBIN/ps_functions.sh
if [[ $? -ne 0 ]]; then
   echo -e "\nError: Failed to load SDP ps_functions.sh lib.\n"
   exit 1
fi

if [[ $(id -u) -eq 0 ]]; then
   exec su - $OSUSER -c "$0 $*"
elif [[ $(id -u -n) != $OSUSER ]]; then
   echo "$0 can only be run by root or $OSUSER"
   exit 1
fi

if [[ ! -x $P4BROKERBIN ]]; then
   echo -e "\nError: $P4BROKERBIN is not executable."
   exit 2;
fi

declare OpMode=${2:-usage}
declare ConfigTag=${3:-default}
declare ConfigTagList=
declare -i ExitStatus=0

if [[ "$ConfigTag" == all ]]; then
   ConfigTagList="default $(cd ${P4CCFG}; ls ${P4SERVER}.broker.*.cfg 2>/dev/null | cut -d '.' -f 3)"
   ConfigTagList=$(echo $ConfigTagList)
else
   ConfigTagList=$ConfigTag
fi

for tag in $ConfigTagList; do
   if [[ $tag == default ]]; then
      if [[ -r /p4/common/config/${P4SERVER}.broker.${HOSTNAME%%.*}.cfg ]]; then
         export P4BROKERCFG=${P4CCFG}/${P4SERVER}.broker.${HOSTNAME%%.*}.cfg
      else
         export P4BROKERCFG=${P4CCFG}/${P4SERVER}.broker.cfg
      fi
   else
      export P4BROKERCFG=${P4CCFG}/${P4SERVER}.broker.${tag}.cfg
   fi

   if [[ ! -r "$P4BROKERCFG" ]]; then
      echo -e "\nError: Missing broker config file $P4BROKERCFG.\n"
      ExitStatus=1
      continue
   fi

   case "$OpMode" in
      start)
         echo "Starting ${P4BROKERBIN} ($tag config) using ${P4BROKERCFG}."
         ${P4BROKERBIN} -d -c ${P4BROKERCFG}
      ;;

      status)
         p4broker_pids=$(get_pids "$P4BROKERCFG")
         if [[ -n "$p4broker_pids" ]]; then
            echo "${P4BROKERBIN} ($tag config) is running as pids: $p4broker_pids."
            ExitStatus=0
         else
            echo "${P4BROKERBIN} ($tag config) is NOT running."
            ExitStatus=1
         fi
      ;;

      stop)
         p4broker_pids=$(get_pids "$P4BROKERCFG")

         if [[ -n "$p4broker_pids" ]]; then
            while [[ -n "$p4broker_pids" ]]; do
               echo "Shutting down ${P4BROKERBIN} ($tag config) pids $p4broker_pids."
               kill $p4broker_pids 2>&1 | $GREP -v -i "No such"
               sleep 1
               p4broker_pids=$(get_pids "$P4BROKERCFG")
            done
            echo "${P4BROKERBIN} ($tag config) stopped."
            ExitStatus=0
         else
            echo "${P4BROKERBIN} ($tag config) was NOT running."
            ExitStatus=1
         fi
      ;;
      restart)
         $0 $SDP_INSTANCE stop $tag
         $0 $SDP_INSTANCE start $tag
      ;;
      *)
         echo -e "\nUsage: $0 SDP_INSTANCE {start|stop|status|restart} [<ConfigTag>|all]\n"
         ExitStatus=1
      ;;
   esac
done

exit $ExitStatus