From f805c64b434cdea13a390db7aba1e759a32840a8 Mon Sep 17 00:00:00 2001
From: Michael Maurer <maurer.mi@northeastern.edu>
Date: Sat, 23 Mar 2024 17:31:36 -0400
Subject: [PATCH] Usability updates for lua bench testing

Add in script for running lua bench, and add
configurable logging to the lua bench test.

Also updates .gitignore to ignore installed
lua files but not files in a 'lua/' subdirectory.

Signed-off-by: Michael Maurer <maurer.mi@northeastern.edu>
---
 .gitignore                           |  2 +-
 scripts/lua_bench.sh                 | 36 ++++++++++++++++++++++++++++
 tools/bench/parsec/lua/lua_bench.cpp |  1 +
 3 files changed, 38 insertions(+), 1 deletion(-)
 create mode 100755 scripts/lua_bench.sh

diff --git a/.gitignore b/.gitignore
index ca0adf668..e76ac3e60 100644
--- a/.gitignore
+++ b/.gitignore
@@ -43,7 +43,7 @@ NuRaft*/
 curl*/
 jsoncpp*/
 ethash*/
-lua*/
+lua-*/
 benchmark-results/
 CMakeFiles/
 plots/
diff --git a/scripts/lua_bench.sh b/scripts/lua_bench.sh
new file mode 100755
index 000000000..d2ab0ced0
--- /dev/null
+++ b/scripts/lua_bench.sh
@@ -0,0 +1,36 @@
+#!/bin/bash
+
+IP="localhost"
+PORT="8889"
+N_WALLETS=5
+LOGLEVEL="WARN"
+
+function print_help() {
+    echo "Usage: lua_bench.sh [OPTIONS]"
+    echo ""
+    echo "OPTIONS:"
+    echo "  --ip           The IP address to use. Default is localhost."
+    echo "  --port         The port number to use. Default is 8888."
+    echo "  --loglevel     The log level to use. Default is WARN."
+    echo "  -h, --help     Show this help message and exit."
+    echo ""
+}
+
+for arg in "$@"; do
+    if [[ "$arg" == "-h" || "$arg" == "--help" ]]; then
+        print_help
+        exit 0
+    elif [[ "$arg" == "--ip"* ]]; then
+        IP="${arg#--ip=}"
+    elif [[ "$arg" == "--port"* ]]; then
+        PORT="${arg#--port=}"
+    elif [[ "$arg" == "--loglevel"* ]]; then
+        LOGLEVEL="${arg#--loglevel=}"
+    fi
+done
+./build/tools/bench/parsec/lua/lua_bench --component_id=0 \
+    --ticket_machine0_endpoint=$IP:7777 --ticket_machine_count=1 \
+    --shard_count=1 --shard0_count=1 --shard00_endpoint=$IP:5556 \
+    --agent_count=1 --agent0_endpoint=$IP:$PORT \
+    --loglevel=$LOGLEVEL scripts/gen_bytecode.lua $N_WALLETS
+echo done
diff --git a/tools/bench/parsec/lua/lua_bench.cpp b/tools/bench/parsec/lua/lua_bench.cpp
index 94972e04e..6a9b53676 100644
--- a/tools/bench/parsec/lua/lua_bench.cpp
+++ b/tools/bench/parsec/lua/lua_bench.cpp
@@ -32,6 +32,7 @@ auto main(int argc, char** argv) -> int {
         log->error("Error parsing options");
         return 1;
     }
+    log->set_loglevel(cfg->m_loglevel);
 
     auto args = cbdc::config::get_args(argc, argv);
     auto n_wallets = std::stoull(args.back());