mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-10-13 13:34:05 +00:00
179 lines
4.3 KiB
Bash
179 lines
4.3 KiB
Bash
#!/bin/bash
|
|
|
|
set -o nounset
|
|
|
|
# Make sure umask is sane
|
|
umask 022
|
|
|
|
# Check for uninitialized variables
|
|
NOUNSET=${NOUNSET:-}
|
|
if [[ -n "$NOUNSET" ]]; then
|
|
set -o nounset
|
|
fi
|
|
|
|
# TBD: Defaults to $HOME, otherwise if we want to address:
|
|
# pid path could be /etc/pid_path
|
|
# monit_script could be
|
|
# log_path /var/log/monit/
|
|
# exec path(for monitrc) could be /etc/monitrc, /usr/local/etc/monitrc
|
|
|
|
# Check if directory for monit logs exists
|
|
if [ ! -d "$HOME/.bigchaindb-monit" ]; then
|
|
mkdir -p "$HOME/.bigchaindb-monit"
|
|
fi
|
|
|
|
monit_pid_path=${MONIT_PID_PATH:=$HOME/.bigchaindb-monit/monit_processes}
|
|
monit_script_path=${MONIT_SCRIPT_PATH:=$HOME/.bigchaindb-monit/monit_script}
|
|
monit_log_path=${MONIT_LOG_PATH:=$HOME/.bigchaindb-monit/logs}
|
|
monit_exec_path=${MONIT_EXEC_PATH:=$HOME}
|
|
|
|
function usage() {
|
|
cat <<EOM
|
|
|
|
Usage: ${0##*/} [-h]
|
|
|
|
Configure Monit for BigchainDB and Tendermint process management.
|
|
|
|
ENV[MONIT_PID_PATH] || --monit-pid-path PATH
|
|
|
|
Absolute path to directory where the the program's pid-file will reside.
|
|
The pid-file contains the ID(s) of the process(es).
|
|
|
|
ENV[MONIT_SCRIPT_PATH] || --monit-script-path PATH
|
|
|
|
Absolute path to the directory where the executable program or
|
|
script is present.
|
|
|
|
ENV[MONIT_LOG_PATH] || --monit-log-path PATH
|
|
|
|
Absolute path to the directory where all the logs for processes
|
|
monitored by Monit are stored.
|
|
|
|
ENV[MONIT_EXEC_PATH] || --monit-exec-path PATH
|
|
|
|
Absolute path to the directory to run the script form.
|
|
|
|
-h|--help
|
|
Show this help and exit.
|
|
|
|
EOM
|
|
}
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
arg="$1"
|
|
case $arg in
|
|
--monit-pid-path)
|
|
monit_pid_path="$2"
|
|
shift
|
|
;;
|
|
--monit-script-path)
|
|
monit_script_path="$2"
|
|
shift
|
|
;;
|
|
--monit-log-path)
|
|
monit_log_path="$2"
|
|
shift
|
|
;;
|
|
--monit-exec-path)
|
|
monit_exec_path="$2"
|
|
shift
|
|
;;
|
|
-h|--help)
|
|
usage
|
|
exit
|
|
;;
|
|
*)
|
|
echo "Unknown option: $1"
|
|
usage
|
|
exit 1
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
# Check if directory for monit logs exists
|
|
if [ ! -d "$monit_log_path" ]; then
|
|
mkdir -p "$monit_log_path"
|
|
fi
|
|
|
|
# Check if directory for monit pid files exists
|
|
if [ ! -d "$monit_pid_path" ]; then
|
|
mkdir -p "$monit_pid_path"
|
|
fi
|
|
|
|
cat >${monit_script_path} <<EOF
|
|
#!/bin/bash
|
|
case \$1 in
|
|
|
|
start_bigchaindb)
|
|
|
|
pushd \$4
|
|
nohup bigchaindb start >> \$3/bigchaindb.out.log 2>> \$3/bigchaindb.err.log &
|
|
|
|
echo \$! > \$2
|
|
popd
|
|
|
|
;;
|
|
|
|
stop_bigchaindb)
|
|
|
|
kill -2 \`cat \$2\`
|
|
rm -f \$2
|
|
|
|
;;
|
|
|
|
start_tendermint)
|
|
|
|
pushd \$4
|
|
nohup tendermint node --consensus.create_empty_blocks=false >> \$3/tendermint.out.log 2>> \$3/tendermint.err.log &
|
|
|
|
echo \$! > \$2
|
|
popd
|
|
|
|
;;
|
|
|
|
stop_tendermint)
|
|
|
|
kill -2 \`cat \$2\`
|
|
rm -f \$2
|
|
|
|
;;
|
|
|
|
esac
|
|
exit 0
|
|
EOF
|
|
chmod +x ${monit_script_path}
|
|
|
|
# configure monitrc
|
|
cat >${monit_exec_path}/.monitrc <<EOF
|
|
set httpd
|
|
port 2812
|
|
allow localhost
|
|
|
|
check process bigchaindb
|
|
with pidfile ${monit_pid_path}/bigchaindb.pid
|
|
start program "${monit_script_path} start_bigchaindb $monit_pid_path/bigchaindb.pid ${monit_log_path} ${monit_exec_path}"
|
|
restart program "${monit_script_path} start_bigchaindb $monit_pid_path/bigchaindb.pid ${monit_log_path} ${monit_exec_path}"
|
|
stop program "${monit_script_path} stop_bigchaindb $monit_pid_path/bigchaindb.pid ${monit_log_path} ${monit_exec_path}"
|
|
|
|
check process tendermint
|
|
with pidfile ${monit_pid_path}/tendermint.pid
|
|
start program "${monit_script_path} start_tendermint ${monit_pid_path}/tendermint.pid ${monit_log_path} ${monit_exec_path}"
|
|
restart program "${monit_script_path} start_bigchaindb ${monit_pid_path}/bigchaindb.pid ${monit_log_path} ${monit_exec_path}"
|
|
stop program "${monit_script_path} stop_tendermint ${monit_pid_path}/tendermint.pid ${monit_log_path} ${monit_exec_path}"
|
|
depends on bigchaindb
|
|
EOF
|
|
|
|
# Setting permissions for control file
|
|
chmod 0700 ${monit_exec_path}/.monitrc
|
|
|
|
# Kill background processes on exit
|
|
trap exit_trap EXIT
|
|
function exit_trap {
|
|
exit $?
|
|
}
|
|
# Exit on any errors so that errors don't compound and kill if any services already started
|
|
|
|
echo -e "BigchainDB process manager configured!"
|
|
set -o errexit
|