diff options
Diffstat (limited to 'rel/files/env.sh')
| -rwxr-xr-x | rel/files/env.sh | 229 |
1 files changed, 229 insertions, 0 deletions
diff --git a/rel/files/env.sh b/rel/files/env.sh new file mode 100755 index 000000000..5a44a9919 --- /dev/null +++ b/rel/files/env.sh @@ -0,0 +1,229 @@ +#!/bin/sh +# -*- tab-width:4;indent-tabs-mode:nil -*- +# ex: ts=4 sw=4 et + +# installed by node_package (github.com/basho/node_package) + +# /bin/sh on Solaris is not a POSIX compatible shell, but /usr/bin/ksh is. +if [ `uname -s` = 'SunOS' -a "${POSIX_SHELL}" != "true" ]; then + POSIX_SHELL="true" + export POSIX_SHELL + # To support 'whoami' add /usr/ucb to path + PATH=/usr/ucb:$PATH + export PATH + exec /usr/bin/ksh $0 "$@" +fi +unset POSIX_SHELL # clear it so if we invoke other scripts, they run as ksh as well + +RUNNER_SCRIPT_DIR={{runner_script_dir}} +RUNNER_SCRIPT=${0##*/} + +RUNNER_BASE_DIR={{runner_base_dir}} +RUNNER_ETC_DIR={{runner_etc_dir}} +RUNNER_LOG_DIR={{runner_log_dir}} +RUNNER_LIB_DIR={{runner_lib_dir}} +RUNNER_PATCH_DIR={{runner_patch_dir}} +PIPE_DIR={{pipe_dir}} +RUNNER_USER={{runner_user}} +APP_VERSION={{app_version}} + +# Variables needed to support creation of .pid files +# PID directory and pid file name of this app +# ex: /var/run/riak & /var/run/riak/riak.pid +RUN_DIR="/var/run" # for now hard coded unless we find a platform that differs +PID_DIR=$RUN_DIR/$RUNNER_SCRIPT +PID_FILE=$PID_DIR/$RUNNER_SCRIPT.pid + +# Threshold where users will be warned of low ulimit file settings +# default it if it is not set +ULIMIT_WARN={{runner_ulimit_warn}} +if [ -z "$ULIMIT_WARN" ]; then + ULIMIT_WARN=4096 +fi + +# Registered process to wait for to consider start a success +WAIT_FOR_PROCESS={{runner_wait_process}} + +WHOAMI=`whoami` + +# Echo to stderr on errors +echoerr() { echo "$@" 1>&2; } + +# Extract the target node name from node.args +NAME_ARG=`egrep '^\-s?name' $RUNNER_ETC_DIR/vm.args` +if [ -z "$NAME_ARG" ]; then + echoerr "vm.args needs to have either -name or -sname parameter." + exit 1 +fi + +# Learn how to specify node name for connection from remote nodes +echo "$NAME_ARG" | grep '^-sname' > /dev/null 2>&1 +if [ "X$?" = "X0" ]; then + NAME_PARAM="-sname" + NAME_HOST="" +else + NAME_PARAM="-name" + echo "$NAME_ARG" | grep '@.*' > /dev/null 2>&1 + if [ "X$?" = "X0" ]; then + NAME_HOST=`echo "${NAME_ARG}" | sed -e 's/.*\(@.*\)$/\1/'` + else + NAME_HOST="" + fi +fi + +# Extract the target cookie +COOKIE_ARG=`grep '^\-setcookie' $RUNNER_ETC_DIR/vm.args` +if [ -z "$COOKIE_ARG" ]; then + echoerr "vm.args needs to have a -setcookie parameter." + exit 1 +fi + +# Parse out release and erts info +START_ERL=`cat $RUNNER_BASE_DIR/releases/start_erl.data` +ERTS_VSN=${START_ERL% *} +APP_VSN=${START_ERL#* } + +# Add ERTS bin dir to our path +ERTS_PATH=$RUNNER_BASE_DIR/erts-$ERTS_VSN/bin + +# Setup command to control the node +NODETOOL="$ERTS_PATH/escript $ERTS_PATH/nodetool $NAME_ARG $COOKIE_ARG" +NODETOOL_LITE="$ERTS_PATH/escript $ERTS_PATH/nodetool" + +# Ping node without stealing stdin +ping_node() { + $NODETOOL ping < /dev/null +} + +# Attempts to create a pid directory like /var/run/APPNAME and then +# changes the permissions on that directory so the $RUNNER_USER can +# read/write/delete .pid files during startup/shutdown +create_pid_dir() { + # Validate RUNNER_USER is set and they have permissions to write to /var/run + # Don't continue if we've already sudo'd to RUNNER_USER + if ([ "$RUNNER_USER" ] && [ "x$WHOAMI" != "x$RUNNER_USER" ]); then + if [ -w $RUN_DIR ]; then + mkdir -p $PID_DIR + ES=$? + if [ "$ES" -ne 0 ]; then + return 1 + else + # Change permissions on $PID_DIR + chown $RUNNER_USER $PID_DIR + ES=$? + if [ "$ES" -ne 0 ]; then + return 1 + else + return 0 + fi + fi + else + # If we don't have permissions, fail + return 1 + fi + fi + + # If RUNNER_USER is not set this is probably a test setup (devrel) and does + # not need a .pid file, so do not return error + return 0 +} + +# Attempt to create a pid file for the process +# This function assumes the process is already up and running and can +# respond to a getpid call. It also assumes that two processes +# with the same name will not be run on the machine +# Do not print any error messages as failure to create a pid file because +# pid files are strictly optional +# This function should really only be called in a "start" function +# you have been warned +create_pid_file() { + # Validate a pid directory even exists + if [ -w $PID_DIR ]; then + # Grab the proper pid from getpid + get_pid + ES=$? + if [ "$ES" -ne 0 ]; then + return $ES + else + # Remove pid file if it already exists since we do not + # plan for multiple identical runners on a single machine + rm -f $PID_FILE + echo $PID > $PID_FILE + return 0 + fi + else + return 1 + fi +} + +# Function to su into correct user +check_user() { + # Validate that the user running the script is the owner of the + # RUN_DIR. + if ([ "$RUNNER_USER" ] && [ "x$WHOAMI" != "x$RUNNER_USER" ]); then + type sudo > /dev/null 2>&1 + if [ "$?" -ne 0 ]; then + echoerr "sudo doesn't appear to be installed and your EUID isn't $RUNNER_USER" 1>&2 + exit 1 + fi + exec sudo -H -u $RUNNER_USER -i $RUNNER_SCRIPT_DIR/$RUNNER_SCRIPT $@ + fi +} + +# Function to validate the node is down +node_down_check() { + MUTE=`ping_node 2> /dev/null` + if [ "$?" -eq 0 ]; then + echoerr "Node is already running!" + exit 1 + fi +} + +# Function to validate the node is up +node_up_check() { + MUTE=`ping_node 2> /dev/null` + if [ "$?" -ne 0 ]; then + echoerr "Node is not running!" + exit 1 + fi +} + +# Function to check if the config file is valid +check_config() { + MUTE=`$NODETOOL_LITE chkconfig $RUNNER_ETC_DIR/app.config` + if [ "$?" -ne 0 ]; then + echoerr "Error reading $RUNNER_ETC_DIR/app.config" + exit 1 + fi + echo "config is OK" +} + +# Function to check if ulimit is properly set +check_ulimit() { + + # don't fail if this is unset + if [ ! -z "$ULIMIT_WARN" ]; then + ULIMIT_F=`ulimit -n` + if [ "$ULIMIT_F" -lt $ULIMIT_WARN ]; then + echo "!!!!" + echo "!!!! WARNING: ulimit -n is ${ULIMIT_F}; ${ULIMIT_WARN} is the recommended minimum." + echo "!!!!" + fi + fi +} + +# Set the PID global variable, return 1 on error +get_pid() { + PID=`$NODETOOL getpid < /dev/null` + if [ "$?" -ne 0 ]; then + echo "Node is not running!" + return 1 + fi + + # don't allow empty or init pid's + if [ -z $PID ] || [ "$PID" -le 1 ]; then + return 1 + fi + + return 0 +} |