#!/bin/bash
P=linkbench-bench
DEFAULT_VERSION=0
. $SHELLPACK_INCLUDE/common.sh
TIME_CMD=`which time`
if [ "$TIME_CMD" = "" ]; then
        TIMEFORMAT="%2Uuser %2Ssystem %Relapsed %P%%CPU"
        TIME_CMD="time"
fi

FIRST_DB_INSTALL=false
MARIADB_VERSION=10.2.14
MARIADB_DBNAME=linkdb
MARIADB_ADMIN_USER=root
MARIADB_ADMIN_PASSWORD=mmtests-default
MYSQLCMD="${SHELLPACK_SOURCES}/mariadbbuild-${MARIADB_VERSION}-installed/bin/mysql -u $MARIADB_ADMIN_USER -p$MARIADB_ADMIN_PASSWORD"
MYSQLADMIN="${SHELLPACK_SOURCES}/mariadbbuild-${MARIADB_VERSION}-installed/bin/mysqladmin -u $MARIADB_ADMIN_USER -p$MARIADB_ADMIN_PASSWORD"
DBSTART_OPTIONS="--innodb_flush_method=nosync,--innodb_flush_log_at_trx_commit=0,--innodb_buffer_pool_size=${BUFF_POOL_SIZE},--innodb_log_file_size=512M,--max_allowed_packet=1G"

# Always scrub sources as the benchmark in question does not properly
# reinitialise itself if the data partitions gets scrubbed
if [ "$INSTALL_ONLY" = "yes" ]; then
	rm -rf $SHELLPACK_SOURCES/mariadb*
	rm -rf $SHELLPACK_SOURCES/maven*
	rm -rf $SHELLPACK_SOURCES/linkbench*
fi

# Basic argument parser
TASKSET_SERVER=
TASKSET_CLIENT=
SERVERSIDE_COMMAND=none
SERVERSIDE_NAME=`date +%Y%m%d-%H%M-%S`

while [ "$1" != "" ]; do
	case "$1" in
	-v)
		VERSION=$2
		shift 2
		;;
	--serverside-command)
		SERVERSIDE_COMMAND=$2
		shift 2
		;;
	--serverside-name)
		SERVERSIDE_NAME=$2
		shift 2
		;;
	--install-only)
		INSTALL_ONLY=yes
		shift
		;;
	--install-force)
		INSTALL_FORCE=yes
		shift
		;;
	--warmup-time)
		LINKBENCH_WARMUP_TIME=$2
		shift 2
		;;
	--min-threads)
		LINKBENCH_MIN_THREADS=$2
		shift 2
		;;
	--max-threads)
		LINKBENCH_MAX_THREADS=$2
		shift 2
		;;
	--iterations)
		LINKBENCH_ITERATIONS=$2
		shift 2
		;;
	*)
		echo Unrecognised option: $1
		shift
	esac
done
if [ "$TASKSET_SERVER" != "" ]; then
	echo TASKSET_SERVER: $TASKSET_SERVER
	echo TASKSET_CLIENT: $TASKSET_CLIENT
fi
if [ -z "$VERSION" ]; then
	VERSION=$DEFAULT_VERSION
fi

if [ "$INSTALL_FORCE" = "yes" ]; then
	rm -rf $SHELLPACK_SOURCES/linkbench-${VERSION}
fi
if [ ! -d $SHELLPACK_SOURCES/linkbench-${VERSION}-installed ]; then
	mmtests_activity source-install
	$SHELLPACK_INCLUDE/shellpack-install-linkbench -v ${VERSION}  || die linkbench install script returned error
	mmtests_activity source-installed
fi
cd $SHELLPACK_SOURCES/linkbench-${VERSION}-installed || die Failed to cd to linkbench install directory
if [ "$INSTALL_ONLY" = "yes" ]; then
	echo linkbench installed only as requested.
	exit $SHELLPACK_SUCCESS
fi

# Include monitor hooks
. $SHELLPACK_INCLUDE/include-monitor.sh

if [ ! -e $JAVA_HOME ]; then
	install-depends java-1_7_0-openjdk
	install-depends java-1_7_0-openjdk-devel

	if [ ! -e /usr/lib64/jvm/java ]; then
		echo ERROR: Java binary is not located at $JAVA_HOME as configured
		exit -1
	fi
	export JAVA_HOME=/usr/lib64/jvm
fi
export PATH=$JAVA_HOME:$JAVA_HOME/bin:$PATH

CONFIG_PATH=$SHELLPACK_SOURCES/linkbench-${VERSION}-installed/config/

EFFECTIVE_CACHESIZE=$((MEMTOTAL_BYTES*6/10))
SHARED_BUFFERS=$((MEMTOTAL_BYTES/4))
DATABASE_MEMORY_AVAILABLE=$((MEMTOTAL_BYTES-LINKBENCH_WORKLOAD_SIZE))
if [ $DATABASE_MEMORY_AVAILABLE -gt $((MEMTOTAL_BYTES/4)) ]; then
	EFFECTIVE_CACHESIZE=$((DATABASE_MEMORY_AVAILABLE*6/10))
	SHARED_BUFFERS=$((DATABASE_MEMORY_AVAILABLE/4))
fi

THREADS=
START_THREAD=$LINKBENCH_MIN_THREADS
END_THREAD=$LINKBENCH_MAX_THREADS
if [ $END_THREAD -gt 32 ]; then
	THREADS=`seq $START_THREAD 3 8`
	THREADS="$THREADS `seq 12 9 32`"
	THREADS="$THREADS `seq 48 31 $END_THREAD`"
elif [ $END_THREAD -gt 8 ]; then
	THREADS=`seq $START_THREAD 2 8`
	THREADS="$THREADS `seq 12 6 $END_THREAD`"
else
	THREADS=`seq $START_THREAD 2 $END_THREAD`
fi
if [ `echo $THREADS | awk '{print $NF}'` -ne $END_THREAD ]; then
	THREADS="$THREADS $END_THREAD"
fi

for NR_THREADS in $THREADS; do
	if [ $NR_THREADS -lt $LINKBENCH_MIN_THREADS ]; then
		continue
	fi
	mmtests_activity process $NR_THREADS/$END_THREAD
		monitor_post_hook $LOGDIR_RESULTS $NR_THREADS
		mmtests_activity $NR_THREADS

for ITERATION in `seq 1 $LINKBENCH_ITERATIONS`; do
	mmtests_activity iteration $ITERATION
				$SHELLPACK_INCLUDE/shellpack-bench-mariadbbuild --start \
				    --start_opts $DBSTART_OPTIONS \
				    --effective_cachesize $EFFECTIVE_CACHESIZE \
				    --shared_buffers $SHARED_BUFFERS \
				    --work_mem $((16*1048576)) || die Failed to get usable database installation

				# This is the data-request phase.
				echo Starting request run $ITERATION/$LINKBENCH_ITERATIONS, $NR_THREADS requester threads.
				echo Performing $LINKBENCH_WARMUP_TIME second warmup phase

				$SHELLPACK_SOURCES/linkbench-${VERSION}-installed/bin/linkbench -c $CONFIG_PATH/MyConfig.properties \
				    -D warmup_time=$LINKBENCH_WARMUP_TIME \
				    -D requesters=$NR_THREADS \
				    -L $LOGDIR_RESULTS/linkbench-request-${NR_THREADS}-${ITERATION}.log -r

				$SHELLPACK_INCLUDE/shellpack-bench-mariadbbuild --stop

sync
done
		monitor_post_hook $LOGDIR_RESULTS $NR_THREADS
done

exit $SHELLPACK_SUCCESS
#### Description linkbench
#### Details linkbench-bench 7
