mirror of
https://github.com/dolthub/dolt.git
synced 2026-01-29 10:41:05 -06:00
146 lines
2.8 KiB
Bash
Executable File
146 lines
2.8 KiB
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
set -o pipefail
|
|
|
|
SYSBENCH_TEST="types_table_scan"
|
|
WORKING_DIR=`mktemp -d`
|
|
PPROF=0
|
|
PORT=3366
|
|
|
|
USER="root"
|
|
PASS=""
|
|
|
|
# parse options
|
|
# superuser.com/questions/186272/
|
|
while test $# -gt 0
|
|
do
|
|
case "$1" in
|
|
|
|
--new-new) export DOLT_DEFAULT_BIN_FORMAT="__DOLT__"
|
|
;;
|
|
|
|
--no-exchange) export SINGLE_THREAD_FEATURE_FLAG=true
|
|
;;
|
|
|
|
# benchmark with pprof profiling
|
|
--pprof) PPROF=1
|
|
;;
|
|
|
|
# run dolt single threaded
|
|
--single) export GOMAXPROCS=1
|
|
;;
|
|
|
|
--journal) export DOLT_ENABLE_CHUNK_JOURNAL=true
|
|
;;
|
|
|
|
# specify sysbench benchmark
|
|
*) SYSBENCH_TEST="$1"
|
|
;;
|
|
|
|
esac
|
|
shift
|
|
done
|
|
|
|
if [ ! -d "./sysbench-lua-scripts" ]; then
|
|
git clone https://github.com/dolthub/sysbench-lua-scripts.git
|
|
fi
|
|
|
|
# collect custom sysbench scripts
|
|
cp ./sysbench-lua-scripts/*.lua "$WORKING_DIR"
|
|
|
|
# grab testing SSL pems
|
|
cp ../../libraries/doltcore/servercfg/testdata/chain* "$WORKING_DIR"
|
|
|
|
cd "$WORKING_DIR"
|
|
|
|
# make a sql-server config file
|
|
cat <<YAML > dolt-config.yaml
|
|
log_level: info
|
|
|
|
listener:
|
|
host: 127.0.0.1
|
|
port: $PORT
|
|
tls_key: "./chain_key.pem"
|
|
tls_cert: "./chain_cert.pem"
|
|
require_secure_transport: true
|
|
|
|
system_variables: {
|
|
sql_mode: ""
|
|
}
|
|
|
|
YAML
|
|
|
|
# start a server
|
|
mkdir sbtest
|
|
cd sbtest
|
|
dolt init
|
|
cd ..
|
|
dolt sql-server --config="dolt-config.yaml" 2> prepare.log &
|
|
SERVER_PID="$!"
|
|
|
|
# stop it if it crashes
|
|
cleanup() {
|
|
kill -15 "$SERVER_PID"
|
|
}
|
|
trap cleanup EXIT
|
|
|
|
# setup benchmark
|
|
echo "benchmark $SYSBENCH_TEST bootstrapping at $WORKING_DIR"
|
|
|
|
sleep 1
|
|
sysbench \
|
|
--db-driver="mysql" \
|
|
--mysql-host="127.0.0.1" \
|
|
--mysql-port="$PORT" \
|
|
--mysql-user="$USER" \
|
|
--mysql-password="$PASS" \
|
|
--db-ps-mode=disable \
|
|
--time=120 \
|
|
--table-size=10000 \
|
|
--percentile=50 \
|
|
--rand-type=uniform \
|
|
--rand-seed=1 \
|
|
"$SYSBENCH_TEST" prepare
|
|
|
|
# restart server to isolate bench run
|
|
kill -15 "$SERVER_PID"
|
|
|
|
# maybe run with pprof
|
|
if [ "$PPROF" -eq 1 ]; then
|
|
dolt --prof cpu sql-server --config="dolt-config.yaml" 2> run.log &
|
|
else
|
|
dolt sql-server --config="dolt-config.yaml" 2> run.log &
|
|
fi
|
|
SERVER_PID="$!"
|
|
sleep 1
|
|
|
|
|
|
# run benchmark
|
|
echo "benchmark $SYSBENCH_TEST starting at $WORKING_DIR"
|
|
|
|
sysbench \
|
|
--db-driver="mysql" \
|
|
--mysql-host="127.0.0.1" \
|
|
--mysql-port="$PORT" \
|
|
--mysql-user="$USER" \
|
|
--mysql-password="$PASS" \
|
|
--db-ps-mode=disable \
|
|
--time=120 \
|
|
--table-size=10000 \
|
|
--percentile=50 \
|
|
--rand-type=uniform \
|
|
--rand-seed=1 \
|
|
"$SYSBENCH_TEST" run
|
|
|
|
unset DOLT_ENABLE_CHUNK_JOURNAL
|
|
unset DOLT_DEFAULT_BIN_FORMAT
|
|
unset SINGLE_THREAD_FEATURE_FLAG
|
|
unset GOMAXPROCS
|
|
|
|
echo "benchmark $SYSBENCH_TEST complete at $WORKING_DIR"
|
|
if [ "$PPROF" -eq 1 ]; then
|
|
# parse run.log to output the profile location
|
|
head -n1 "$WORKING_DIR/run.log" | cut -d ":" -f 4
|
|
fi
|
|
echo ""
|