From fcef003df2ec26e0d15446f1870b16b027a44207 Mon Sep 17 00:00:00 2001
From: Andy Arthur <andy@dolthub.com>
Date: Wed, 6 Jul 2022 16:09:41 -0700
Subject: [PATCH] added tpcc_local script

---
 go/performance/scripts/.gitignore    |   1 +
 go/performance/scripts/tpcc_local.sh | 116 +++++++++++++++++++++++++++
 2 files changed, 117 insertions(+)
 create mode 100755 go/performance/scripts/tpcc_local.sh

diff --git a/go/performance/scripts/.gitignore b/go/performance/scripts/.gitignore
index 1905089d0a..779032f0a1 100644
--- a/go/performance/scripts/.gitignore
+++ b/go/performance/scripts/.gitignore
@@ -1 +1,2 @@
 sysbench-lua-scripts/
+sysbench-tpcc/
diff --git a/go/performance/scripts/tpcc_local.sh b/go/performance/scripts/tpcc_local.sh
new file mode 100755
index 0000000000..0544ea3436
--- /dev/null
+++ b/go/performance/scripts/tpcc_local.sh
@@ -0,0 +1,116 @@
+#!/bin/bash
+set -e
+set -o pipefail
+
+WORKING_DIR=`mktemp -d`
+PPROF=0
+PORT=3366
+
+# parse options
+# superuser.com/questions/186272/
+while test $# -gt 0
+do
+    case "$1" in
+
+        # benchmark with new NomsBinFmt
+        --new-nbf) export DOLT_DEFAULT_BIN_FORMAT="__DOLT_1__"
+            ;;
+
+        --new-new) export DOLT_DEFAULT_BIN_FORMAT="__DOLT_1__" &&
+            export ENABLE_ROW_ITER_2=true
+            ;;
+
+        --no-exchange) export SINGLE_THREAD_FEATURE_FLAG=true
+            ;;
+
+        # benchmark with pprof profiling
+        --pprof) PPROF=1
+            ;;
+
+        # run dolt single threaded
+        --single) export GOMAXPROCS=1
+            ;;
+
+    esac
+    shift
+done
+
+if [ ! -d "./sysbench-tpcc" ]; then
+  git clone https://github.com/Percona-Lab/sysbench-tpcc.git
+fi
+
+# collect custom sysbench scripts
+cp ./sysbench-tpcc/*.lua "$WORKING_DIR"
+cd "$WORKING_DIR"
+
+# make a sql-server config file
+cat <<YAML > dolt-config.yaml
+log_level: "debug"
+
+behavior:
+  read_only: false
+  autocommit: true
+
+user:
+  name: "user"
+  password: "pass"
+
+listener:
+  host: "0.0.0.0"
+  port: $PORT
+  max_connections: 128
+  read_timeout_millis: 28800000
+  write_timeout_millis: 28800000
+
+databases:
+  - name: "sbtest"
+    path: "."
+YAML
+
+# start a server
+dolt init
+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 TPC-C bootstrapping at $WORKING_DIR"
+
+sleep 1
+
+./tpcc.lua --db-driver="mysql" --mysql-db="sbtest" --mysql-host="0.0.0.0" --mysql-port="$PORT" --mysql-user="user" --mysql-password="pass" --time=10 --report_interval=1 --threads=2 --tables=1 --scale=1 --trx_level="RR" prepare
+
+# restart server to isolate bench run
+kill -15 "$SERVER_PID"
+
+# stomp conflicts on tx commit
+export DOLT_TRANSACTION_MERGE_STOMP=1
+
+# 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"
+
+./tpcc.lua --db-driver="mysql" --mysql-db="sbtest" --mysql-host="0.0.0.0" --mysql-port="$PORT" --mysql-user="user" --mysql-password="pass" --time=10 --report_interval=1 --threads=2 --tables=1 --scale=1 --trx_level="RR" run
+
+unset DOLT_TRANSACTION_MERGE_STOMP
+unset DOLT_DEFAULT_BIN_FORMAT
+unset ENABLE_ROW_ITER_2
+unset SINGLE_THREAD_FEATURE_FLAG
+unset GOMAXPROCS
+
+echo "benchmark TPC-C complete at $WORKING_DIR"
+echo "DOLT_DEFAULT_BIN_FORMAT='$DOLT_DEFAULT_BIN_FORMAT'"
+echo ""