#47 add iibench test script

scripts/run.iibench.bash

@@ -0,0 +1,172 @@
+#!/usr/bin/env bash
+
+function usage() {
+    echo "run iibench"
+    echo "--mysqlbuild=$mysqlbuild"
+    echo "[--max_row=$max_rows] [--rows_per_report=$rows_per_report] [--insert_only=$insert_only] [ --check=$check]"
+    echo "[--commit=$commit]"
+}
+
+function retry() {
+    local cmd=$*
+    local retries
+    local exitcode
+    let retries=0
+    while [ $retries -le 10 ] ; do
+        echo `date` $cmd
+        bash -c "$cmd"
+        exitcode=$?
+        echo `date` $cmd $exitcode $retries
+        let retries=retries+1
+        if [ $exitcode -eq 0 ] ; then break; fi
+        sleep 10
+    done
+    test $exitcode = 0
+}
+
+mysqlbuild=
+commit=0
+check=1
+mysqlserver=`hostname`
+mysqluser=`whoami`
+mysqlsocket=/tmp/mysql.sock
+svn_server=https://svn.tokutek.com/tokudb
+svn_branch=.
+svn_revision=HEAD
+basedir=$HOME/svn.build
+builddir=$basedir/mysql.build
+system=`uname -s | tr [:upper:] [:lower:]`
+instancetype=
+testinstance=
+arch=`uname -m | tr [:upper:] [:lower:]`
+tracefile=/tmp/run.iibench.trace
+cmd=iibench
+dbname=$cmd
+engine=tokudb
+tblname=testit
+max_rows=50000000
+rows_per_report=1000000
+insert_only=1
+
+# parse the command line
+while [ $# -gt 0 ] ; do
+    arg=$1; shift
+    if [ $arg = "--replace_into" ] ; then 
+        cmd=replace_into
+    elif [ $arg = "--insert_ignore" ] ; then
+        cmd=insert_ignore
+    elif [[ $arg =~ --(.*)=(.*) ]] ; then
+        eval ${BASH_REMATCH[1]}=${BASH_REMATCH[2]}
+    else
+        usage; exit 1
+    fi
+done
+
+if [[ $mysqlbuild =~ (.*)-(tokudb-.*)-(linux)-(x86_64) ]] ; then
+    mysql=${BASH_REMATCH[1]}
+    tokudb=${BASH_REMATCH[2]}
+    system=${BASH_REMATCH[3]}
+    arch=${BASH_REMATCH[4]}
+else
+    exit 1
+fi
+
+# setup the dbname
+if [ $dbname = "iibench" ] ; then dbname=${cmd}_${engine}; fi
+if [ "$testinstance" != "" ] ; then dbname=${dbname}_${testinstance}; fi
+
+if [ -d /usr/local/mysql ] ; then
+    export PATH=/usr/local/mysql/bin:$PATH
+fi
+
+if [ -d /usr/local/mysql/lib/mysql ] ; then
+    export LD_LIBRARY_PATH=/usr/local/mysql/lib/mysql:$PATH
+fi
+
+# goto the base directory
+if [ ! -d $basedir ] ; then mkdir $basedir; fi
+pushd $basedir
+
+# update the build directory
+if [ $commit != 0 ] ; then
+    if [ ! -d $builddir ] ; then mkdir $builddir; fi
+
+    date=`date +%Y%m%d`
+    testresultsdir=$builddir/$date
+    pushd $builddir
+        while [ ! -d $date ] ; do
+            svn mkdir $svn_server/mysql.build/$date -m ""
+            svn checkout -q $svn_server/mysql.build/$date
+            if [ $? -ne 0 ] ; then rm -rf $date; fi
+        done
+    popd
+else
+    testresultsdir=$PWD
+fi
+
+# checkout the code
+testdir=iibench-$mysqlbuild-$mysqlserver
+if [ "$testinstance" != "" ] ; then testdir=$testdir-$testinstance; fi
+rm -rf $testdir
+retry svn export -q -r $svn_revision $svn_server/$svn_branch/iibench $testdir
+exitcode=$?
+if [ $exitcode != 0 ] ; then 
+    retry svn export -q -r $svn_revision $svn_server/iibench $testdir
+    exitcode=$?
+fi
+if [ $exitcode != 0 ] ; then  exit 1; fi
+
+# create the iibench database
+mysql -S $mysqlsocket -u root -e "grant all on *.* to '$mysqluser'@'$mysqlserver'"
+exitcode=$?
+if [ $exitcode != 0 ] ; then exit 1; fi
+
+mysql -S $mysqlsocket -u $mysqluser -e "drop database if exists $dbname"
+exitcode=$?
+if [ $exitcode != 0 ] ; then exit 1; fi
+
+mysql -S $mysqlsocket -u $mysqluser -e "create database $dbname"
+exitcode=$?
+if [ $exitcode != 0 ] ; then exit 1; fi
+
+# run
+if [ $cmd = "iibench" -a $insert_only != 0 ] ; then
+    runfile=$testresultsdir/$dbname-insert_only-$max_rows-$mysqlbuild-$mysqlserver
+else
+    runfile=$testresultsdir/$dbname-$max_rows-$mysqlbuild-$mysqlserver
+fi
+if [ "$instancetype" != "" ] ; then runfile=$runfile-$instancetype; fi
+testresult="PASS"
+
+pushd $testdir/py
+    echo `date` $cmd start $mysql $svn_branch $svn_revision $max_rows $rows_per_report >>$runfile
+    runcmd=$cmd.py
+    args="--db_user=$mysqluser --db_name=$dbname --db_socket=$mysqlsocket --engine=$engine --setup --max_rows=$max_rows --rows_per_report=$rows_per_report --table_name=$tblname"
+    if [ $cmd = "iibench" -a $insert_only != 0 ] ; then runcmd="$runcmd --insert_only"; fi
+    if [ $cmd = "replace_into" ] ; then runcmd="replace_into.py --use_replace_into"; fi
+    if [ $cmd = "insert_ignore" ] ; then runcmd="replace_into.py"; fi
+    ./$runcmd $args >>$runfile 2>&1
+    exitcode=$?
+    echo `date` $cmd complete $exitcode >>$runfile
+    if [ $exitcode != 0 ] ; then testresult="FAIL"; fi
+popd
+
+if [ $check != 0 -a $testresult = "PASS" ] ; then
+    echo `date` check table $tblname >>$runfile
+    mysql -S $mysqlsocket -u $mysqluser -D $dbname -e "check table $tblname" >>$runfile 2>&1
+    exitcode=$?
+    echo `date` check table $tblname $exitcode >>$runfile
+    if [ $exitcode != 0 ] ; then testresult="FAIL"; fi
+fi
+
+# commit results
+if [ $commit != 0 ] ; then
+    if [ $cmd = "iibench" -a $insert_only != 0 ] ; then cmd="$cmd insert_only"; fi
+    svn add $runfile
+    retry svn commit -m \"$testresult $cmd $max_rows $dbname $mysqlbuild $mysqlserver `hostname`\" $runfile
+fi
+
+popd
+
+if [ $testresult = "PASS" ] ; then exitcode=0; else exitcode=1; fi
+exit $exitcode