run-sysbench

#!/bin/bash
set -u

usage()
{
cat << EOF
usage: $0 options

This script runs sysbench and iostat concurrently in iterations with each iteration using a higher 
number of threads than the last. The results are written to a new result directory each time the 
script is run. A file '.run_number' keeps track of the result directory numbering ensuring that 
each new run gets a new directory.

OPTIONS:
   -h   MySQL host            [localhost]
   -P   MySQL port            [3306]
   -s   MySQL socket             [/tmp/mysql.sock]
   -u   MySQL username           [root]
   -p   MySQL password
   -w   ReadWrite mode (on/off)          [on]
   -o   Timeout for each sysbench run     [60]
   -r   Directory to print results to     [./results]
   -t   List of thread counts to iterate [4 8 16 32 64 128 256]
EOF
}

host=localhost
port=3306
socket=/tmp/mysql.sock
user=root
password=
readwrite=on
readonly=off
timeout=60
threads="8,16,32,64,128"
resultsdir=./results
connparams=

while getopts h:P:s:u:p:w:o:r:t: name
do
     case $name in
       h) host="$OPTARG";;
       P) port="$OPTARG";;
       s) socket="$OPTARG";;
       u) user="$OPTARG";;
       p) password="$OPTARG";;
       w) readwrite="$OPTARG";;
       o) timeout="$OPTARG";;
       r) resultsdir="$OPTARG";;
       t) threads="$OPTARG";;
       ?) usage
          exit;;
     esac
done

threads=${threads//,/ }

if [ "$readwrite" == "off" ]; then
     readonly=on
fi
if [ ! "$socket" == "/tmp/mysql.sock" ]; then
  connparams="--mysql-socket=$socket --mysql-user=$user --mysql-password=$password"
else 
  connparams="--mysql-host=$host --mysql-port=$port --mysql-user=$user --mysql-password=$password"
fi

# Determine run number for selecting an output directory
mkdir -p $resultsdir
run_num=-1

if [ -f "$resultsdir/.run_number" ]; then
  read run_num < $resultsdir/.run_number
fi

if [ $run_num -eq -1 ]; then
     run_num=1
fi

if [ $run_num -lt 10 ]; then
     outdir=result-0$run_num
else 
     outdir=result-$run_num
fi

mkdir -p $resultsdir/$outdir

run_num=`expr $run_num + 1`
echo $run_num > $resultsdir/.run_number

for thread in $threads
do
     # Run iostat
     iostat -dx 10 $(($timeout/10+1))  >> $resultsdir/$outdir/iostat.$thread.res &
     
     # Run sysbench
     sysbench $connparams --test=oltp --init-rng=on --num-threads=$thread \
     --oltp-read-only=$readonly --oltp-dist-type=uniform --max-time=$timeout \
     --max-requests=0 --percentile=99 run | tee -a $resultsdir/$outdir/sysbench.$thread.res
     
     sleep 10

done