151 lines
3.7 KiB
Bash
Executable File
151 lines
3.7 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
|
|
# (1) Init.
|
|
#--------------------------------------------------------#
|
|
# (1) Get current absolute dir
|
|
ROOT=$(dirname `realpath $0`);
|
|
|
|
# (2) Check argc
|
|
test $# -lt 2 && echo -e "error: too few arguments\n\n\e[1mUSAGE\e[0m\n\tmaster <key> <host> <port> <dir>\n\n\e[1mARGUMENTS\e[0m\n\t<key>\tThe key of the pool\n\t<host>\tThe host where the pool is hosted\n\t<port>\tTo port where the pool is bound\n\t<dir>\tThe folder containing ONLY the logger daemons\n" && exit 1;
|
|
|
|
# (3) Check @PORT range #
|
|
MIN_PORT=1024;
|
|
MAX_PORT=49151;
|
|
test "$3" -gt "$MAX_PORT" -o "$3" -lt "$MIN_PORT" && echo "error: <port> must be between $MIN_PORT and $MAX_PORT" && exit 1;
|
|
|
|
# (4) Set argument explicit names #
|
|
POOL_KEY="$1";
|
|
POOL_HOST="$2";
|
|
POOL_PORT="$3";
|
|
LOGGER_DIR="`realpath $4`";
|
|
|
|
# (5) Check @LOGGER_DIR #
|
|
test ! -d "$LOGGER_DIR" && echo "error: <dir> is not a valid folder." && exit 1;
|
|
|
|
# (6) init logger variables #
|
|
declare -A OUTPUT_PID; # will contain each bind-output PIDs
|
|
declare -A LOGGER_PID; # will contain each logger PIDs
|
|
|
|
# (7) get logger list #
|
|
LOGGERS=(`ls $LOGGER_DIR`);
|
|
|
|
# (8) If no logger -> exit #
|
|
test "${#LOGGERS[@]}" -eq 0 && echo "error: no logger found in <dir>." && exit 1;
|
|
|
|
# (9) Get local IP address #
|
|
LOCAL_IP="`ip addr | grep 'state UP' -A2 | tail -n1 | awk '{print $2}' | cut -f1 -d'/'`";
|
|
|
|
test -z "$LOCAL_IP" && echo "error: cannot get local IP address." && exit 1;
|
|
|
|
|
|
|
|
|
|
# (2) Set exit management
|
|
#--------------------------------------------------------#
|
|
# (1) On-Exit routine #
|
|
on_exit(){
|
|
|
|
echo "killing 'bind-output' sub processes";
|
|
|
|
for logger_name in "${!OUTPUT_PID[@]}"; do
|
|
|
|
# kill each logger OUTPUT
|
|
echo " > killing output@$logger_name";
|
|
kill -INT ${OUTPUT_PID[$logger_name]} 2>/dev/null;
|
|
|
|
done;
|
|
|
|
for logger_name in "${!LOGGER_PID[@]}"; do
|
|
|
|
# kill each logger OUTPUT
|
|
echo " > killing logger@$logger_name";
|
|
kill -INT ${LOGGER_PID[$logger_name]} 2>/dev/null;
|
|
|
|
done;
|
|
|
|
echo "killing 'master' output socket";
|
|
kill -INT $MASTER_PID 2>/dev/null;
|
|
|
|
echo "killing 'master' loop";
|
|
# kill -INT $LOOP_PID 2>/dev/null;
|
|
|
|
}
|
|
|
|
trap "on_exit;" HUP INT KILL TERM;
|
|
|
|
|
|
|
|
|
|
|
|
# (3) Connect to pool
|
|
#--------------------------------------------------------#
|
|
# (1) Bind-output #
|
|
echo "(.) connect to $POOL_HOST:$POOL_PORT";
|
|
( $ROOT/bind-output master $POOL_HOST $POOL_PORT )&
|
|
MASTER_PID=$?;
|
|
|
|
|
|
# (4) Launch master manager
|
|
#--------------------------------------------------------#
|
|
# (1) For each logger
|
|
PORT="$MIN_PORT";
|
|
for logger_name in "${LOGGERS[@]}"; do
|
|
|
|
# Find an available port
|
|
while true; do
|
|
|
|
# if port not in use -> use it
|
|
ss -tl4 "( sport = :$PORT )" | grep "$PORT" >/dev/null 2>&1 || break;
|
|
|
|
# else try next port (+1)
|
|
PORT="`expr $PORT + 1`";
|
|
|
|
done;
|
|
|
|
# send logger port request
|
|
echo " (.) Sending logger@$logger_name request";
|
|
$ROOT/write master "$POOL_KEY:$logger_name:$LOCAL_IP:$PORT";
|
|
|
|
# check pool response
|
|
RESP="`nc -w 2 -lp $PORT`";
|
|
|
|
# empty response -> ignore
|
|
test -z "$RESP" && continue;
|
|
|
|
# not a number -> ignore
|
|
echo -n "$RESP" | grep -P '^\d+$' >/dev/null || continue;
|
|
|
|
|
|
# Create output bound
|
|
echo " (.) Connecting '$logger_name' to $POOL_HOST:$RESP";
|
|
( $ROOT/bind-output master_$logger_name $POOL_HOST $RESP )&
|
|
OUTPUT_PID[$logger_name]=$!;
|
|
|
|
echo " (.) launching logger@$logger_name";
|
|
# launch logger bound to output
|
|
( $LOGGER_DIR/$logger_name | cat > /tmp/outbuf_master_$logger_name )&
|
|
LOGGER_PID[$logger_name]=$!;
|
|
|
|
|
|
|
|
# increment port for next logger
|
|
PORT="`expr $PORT + 1`";
|
|
|
|
done;
|
|
|
|
|
|
# When all loggers are bound, destroy pool connection
|
|
# kill -INT $MASTER_PID 2>/dev/null;
|
|
|
|
echo "waiting for ${#LOGGER_PID[@]} + ${#OUTPUT_PID[@]} ..";
|
|
for logger_name in "${!LOGGER_PID[@]}"; do
|
|
|
|
echo "waiting for $logger_name";
|
|
wait ${LOGGER_PID[$logger_name]};
|
|
echo "done with $logger_name";
|
|
|
|
done;
|
|
|
|
|
|
# on_exit; |