bash-com/master

#!/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;