#!/bin/bash

function bail () { echo -e "Error: ${1:-Unknown Error}\n"; exit ${2:-1}; }

instance=${1:-Unset}
[[ $instance == "Unset" ]] && bail "Specify instance as parameter"
[[ ! -d /p4/$instance/logs ]] && bail "Instance $instance not found: /p4/$instance/logs"

P4BENCH_SCRIPT=${2:-Unset}
[[ $P4BENCH_SCRIPT == "Unset" ]] && bail "Specify P4BENCH_SCRIPT as second parameter"
[[ ! -f locust_files/p4_$P4BENCH_SCRIPT.py ]] && bail "Benchmark script $P4BENCH_SCRIPT not found: locust_files/p4_$P4BENCH_SCRIPT.py"

export P4BENCH_HOST=`hostname`
export P4BENCH_SCRIPT
export P4BENCH_NUM_SLAVES=$(grep "num_slaves" hosts | awk '{print $2}')

echo "Running p4_${P4BENCH_SCRIPT} on instance ${instance}"

# Remove existing logs to make sure they don't clutter up the measurements
sudo rm /p4/$instance/logs/log
# Remove shared logs on other servers if appropriate
#sudo rm /remote/p4/rep/h02_$instance/logs/log
#sudo rm /remote/p4/rep/h03_$instance/logs/log
#sudo rm /remote/p4/rep/h04_$instance/logs/log

config_file="config_p4_${P4BENCH_SCRIPT}.yml"
sed -e "s/:1666/:${instance}666/" < locust_files/$config_file > $config_file

rm logs/*slave*.out
ansible-playbook -i hosts ansible/copy_logs.yml
ansible-playbook -i hosts ansible/rm_logs.yml
ansible-playbook -i hosts ansible/post_bench.yml
ansible-playbook -i hosts ansible/pre_bench.yml

./run_master.sh
ansible-playbook -i hosts ansible/bench.yml
# ansible-playbook -i hosts ansible/replicas.yml
bash run_monitor.sh