Run helper
Change-Id: I77ae32408a3241f224b73dfb8f1d6402334d95c9
This commit is contained in:
Endre Karlson
parent
84450e6d22
commit
facdc71b27
4
.gitignore
vendored
4
.gitignore
vendored
@ -47,3 +47,7 @@ etc/billingstack/*.ini
|
|||||||
etc/billingstack/*.conf
|
etc/billingstack/*.conf
|
||||||
billingstack/versioninfo
|
billingstack/versioninfo
|
||||||
*.sqlite
|
*.sqlite
|
||||||
|
|
||||||
|
|
||||||
|
billingstack-screenrc
|
||||||
|
run/
|
||||||
|
|||||||
217
tools/control.sh
Executable file
217
tools/control.sh
Executable file
@ -0,0 +1,217 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
|
||||||
|
# script to help with BS
|
||||||
|
|
||||||
|
# Dependencies:
|
||||||
|
# - functions
|
||||||
|
|
||||||
|
# Save trace setting
|
||||||
|
XTRACE=$(set +o | grep xtrace)
|
||||||
|
set -x
|
||||||
|
|
||||||
|
# Keep track of this directory
|
||||||
|
TOOL_DIR=$(cd $(dirname "$0") && pwd)
|
||||||
|
TOP_DIR=$TOOL_DIR/..
|
||||||
|
|
||||||
|
RUN_DIR=$TOP_DIR/run
|
||||||
|
|
||||||
|
SCREEN_NAME=billingstack
|
||||||
|
SCREEN_LOGDIR=$RUN_DIR
|
||||||
|
|
||||||
|
CONF_DIR=$TOP_DIR/etc/billingstack
|
||||||
|
CONFIG=${CONFIG:-$CONF_DIR/billingstack.conf}
|
||||||
|
|
||||||
|
|
||||||
|
function ensure_dir() {
|
||||||
|
local dir=$1
|
||||||
|
[ ! -d "$dir" ] && {
|
||||||
|
echo "Attempting to create $dir"
|
||||||
|
mkdir -p $dir
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# Normalize config values to True or False
|
||||||
|
# Accepts as False: 0 no false False FALSE
|
||||||
|
# Accepts as True: 1 yes true True TRUE
|
||||||
|
# VAR=$(trueorfalse default-value test-value)
|
||||||
|
function trueorfalse() {
|
||||||
|
local default=$1
|
||||||
|
local testval=$2
|
||||||
|
|
||||||
|
[[ -z "$testval" ]] && { echo "$default"; return; }
|
||||||
|
[[ "0 no false False FALSE" =~ "$testval" ]] && { echo "False"; return; }
|
||||||
|
[[ "1 yes true True TRUE" =~ "$testval" ]] && { echo "True"; return; }
|
||||||
|
echo "$default"
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# _run_process() is designed to be backgrounded by run_process() to simulate a
|
||||||
|
# fork. It includes the dirty work of closing extra filehandles and preparing log
|
||||||
|
# files to produce the same logs as screen_it(). The log filename is derived
|
||||||
|
# from the service name and global-and-now-misnamed SCREEN_LOGDIR
|
||||||
|
# _run_process service "command-line"
|
||||||
|
function _run_process() {
|
||||||
|
local service=$1
|
||||||
|
local command="$2"
|
||||||
|
|
||||||
|
# Undo logging redirections and close the extra descriptors
|
||||||
|
exec 1>&3
|
||||||
|
exec 2>&3
|
||||||
|
exec 3>&-
|
||||||
|
exec 6>&-
|
||||||
|
|
||||||
|
if [[ -n ${SCREEN_LOGDIR} ]]; then
|
||||||
|
exec 1>&${SCREEN_LOGDIR}/screen-${1}.${CURRENT_LOG_TIME}.log 2>&1
|
||||||
|
ln -sf ${SCREEN_LOGDIR}/screen-${1}.${CURRENT_LOG_TIME}.log ${SCREEN_LOGDIR}/screen-${1}.log
|
||||||
|
|
||||||
|
# TODO(dtroyer): Hack to get stdout from the Python interpreter for the logs.
|
||||||
|
export PYTHONUNBUFFERED=1
|
||||||
|
fi
|
||||||
|
|
||||||
|
exec /bin/bash -c "$command"
|
||||||
|
die "$service exec failure: $command"
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# run_process() launches a child process that closes all file descriptors and
|
||||||
|
# then exec's the passed in command. This is meant to duplicate the semantics
|
||||||
|
# of screen_it() without screen. PIDs are written to
|
||||||
|
# $SERVICE_DIR/$SCREEN_NAME/$service.pid
|
||||||
|
# run_process service "command-line"
|
||||||
|
function run_process() {
|
||||||
|
local service=$1
|
||||||
|
local command="$2"
|
||||||
|
|
||||||
|
# Spawn the child process
|
||||||
|
_run_process "$service" "$command" &
|
||||||
|
echo $!
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# Helper to launch a service in a named screen
|
||||||
|
# screen_it service "command-line"
|
||||||
|
function screen_it {
|
||||||
|
SCREEN_NAME=${SCREEN_NAME:-stack}
|
||||||
|
SERVICE_DIR=${SERVICE_DIR:-${RUN_DIR}/status}
|
||||||
|
USE_SCREEN=$(trueorfalse True $USE_SCREEN)
|
||||||
|
|
||||||
|
if is_service_enabled $1; then
|
||||||
|
# Append the service to the screen rc file
|
||||||
|
screen_rc "$1" "$2"
|
||||||
|
|
||||||
|
if [[ "$USE_SCREEN" = "True" ]]; then
|
||||||
|
screen -S $SCREEN_NAME -X screen -t $1
|
||||||
|
|
||||||
|
if [[ -n ${SCREEN_LOGDIR} ]]; then
|
||||||
|
screen -S $SCREEN_NAME -p $1 -X logfile ${SCREEN_LOGDIR}/screen-${1}.${CURRENT_LOG_TIME}.log
|
||||||
|
screen -S $SCREEN_NAME -p $1 -X log on
|
||||||
|
ln -sf ${SCREEN_LOGDIR}/screen-${1}.${CURRENT_LOG_TIME}.log ${SCREEN_LOGDIR}/screen-${1}.log
|
||||||
|
fi
|
||||||
|
|
||||||
|
# sleep to allow bash to be ready to be send the command - we are
|
||||||
|
# creating a new window in screen and then sends characters, so if
|
||||||
|
# bash isn't running by the time we send the command, nothing happens
|
||||||
|
sleep 1.5
|
||||||
|
|
||||||
|
NL=`echo -ne '\015'`
|
||||||
|
screen -S $SCREEN_NAME -p $1 -X stuff "$2 || touch \"$SERVICE_DIR/$SCREEN_NAME/$1.failure\"$NL"
|
||||||
|
else
|
||||||
|
# Spawn directly without screen
|
||||||
|
run_process "$1" "$2" >$SERVICE_DIR/$SCREEN_NAME/$service.pid
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# Screen rc file builder
|
||||||
|
# screen_rc service "command-line"
|
||||||
|
function screen_rc {
|
||||||
|
SCREEN_NAME=${SCREEN_NAME:-stack}
|
||||||
|
SCREENRC=$TOP_DIR/$SCREEN_NAME-screenrc
|
||||||
|
if [[ ! -e $SCREENRC ]]; then
|
||||||
|
# Name the screen session
|
||||||
|
echo "sessionname $SCREEN_NAME" > $SCREENRC
|
||||||
|
# Set a reasonable statusbar
|
||||||
|
echo "hardstatus alwayslastline '$SCREEN_HARDSTATUS'" >> $SCREENRC
|
||||||
|
echo "screen -t shell bash" >> $SCREENRC
|
||||||
|
fi
|
||||||
|
# If this service doesn't already exist in the screenrc file
|
||||||
|
if ! grep $1 $SCREENRC 2>&1 > /dev/null; then
|
||||||
|
NL=`echo -ne '\015'`
|
||||||
|
echo "screen -t $1 bash" >> $SCREENRC
|
||||||
|
echo "stuff \"$2$NL\"" >> $SCREENRC
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# Uses global ``ENABLED_SERVICES``
|
||||||
|
# is_service_enabled service [service ...]
|
||||||
|
function is_service_enabled() {
|
||||||
|
services=$@
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
function screen_setup() {
|
||||||
|
ensure_dir $SCREEN_LOGDIR
|
||||||
|
|
||||||
|
# Check to see if we are already running DevStack
|
||||||
|
# Note that this may fail if USE_SCREEN=False
|
||||||
|
if type -p screen >/dev/null && screen -ls | egrep -q "[0-9].$SCREEN_NAME"; then
|
||||||
|
echo "You are already running a stack.sh session."
|
||||||
|
echo "To rejoin this session type 'screen -x stack'."
|
||||||
|
echo "To destroy this session, type './unstack.sh'."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
USE_SCREEN=$(trueorfalse True $USE_SCREEN)
|
||||||
|
echo $USE_SCREEN
|
||||||
|
if [[ "$USE_SCREEN" == "True" ]]; then
|
||||||
|
# Create a new named screen to run processes in
|
||||||
|
screen -d -m -S $SCREEN_NAME -t shell -s /bin/bash
|
||||||
|
sleep 1
|
||||||
|
|
||||||
|
# Set a reasonable status bar
|
||||||
|
if [ -z "$SCREEN_HARDSTATUS" ]; then
|
||||||
|
SCREEN_HARDSTATUS='%{= .} %-Lw%{= .}%> %n%f %t*%{= .}%+Lw%< %-=%{g}(%{d}%H/%l%{g})'
|
||||||
|
fi
|
||||||
|
screen -r $SCREEN_NAME -X hardstatus alwayslastline "$SCREEN_HARDSTATUS"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Clear screen rc file
|
||||||
|
SCREENRC=$TOP_DIR/$SCREEN_NAME-screenrc
|
||||||
|
if [[ -e $SCREENRC ]]; then
|
||||||
|
echo -n > $SCREENRC
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
function screen_destroy() {
|
||||||
|
SCREEN=$(which screen)
|
||||||
|
if [[ -n "$SCREEN" ]]; then
|
||||||
|
SESSION=$(screen -ls | awk '/[0-9].billingstack/ { print $1 }')
|
||||||
|
if [[ -n "$SESSION" ]]; then
|
||||||
|
screen -X -S $SESSION quit
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
function prereq_setup() {
|
||||||
|
ensure_dir $RUN_DIR
|
||||||
|
ensure_dir $SCREEN_DIR
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
case $1 in
|
||||||
|
start)
|
||||||
|
prereq_setup
|
||||||
|
screen_setup
|
||||||
|
screen_it bs-central "$TOOL_DIR/with_venv.sh billingstack-central --config-file $CONFIG"
|
||||||
|
screen_it bs-api "$TOOL_DIR/with_venv.sh billingstack-api --config-file $CONFIG"
|
||||||
|
;;
|
||||||
|
stop)
|
||||||
|
screen_destroy
|
||||||
|
;;
|
||||||
|
esac
|
||||||
Loading…
x
Reference in New Issue
Block a user