Workload adjustments

mitdbg · Nov 12, 2023 · 538f930 · 538f930
1 parent 580638c
commit 538f930
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/experiments/15-e2e-scenarios-v2/scale_up/run_workload.sh b/experiments/15-e2e-scenarios-v2/scale_up/run_workload.sh
@@ -65,6 +65,7 @@ sleep 30
 log_workload_point "start_rana_8"
 start_repeating_olap_runner 8 15 5 $initial_queries "ra_8"
 rana_pid=$runner_pid
+sleep 2
 
 # Scale up to 8 transactional clients and hold for 15 minutes.
 log_workload_point "start_increase_txn_4_to_8"

diff --git a/src/brad/front_end/front_end.py b/src/brad/front_end/front_end.py
@@ -306,8 +306,13 @@ async def _run_query_impl(
             query_rep = QueryRep(query)
             if query_rep.is_transaction_start():
                 session.set_in_transaction(True)
-            assert self._router is not None
-            engine_to_use = await self._router.engine_for(query_rep, session)
+
+            if query.startswith("SET SESSION"):
+                # Support for setting transaction isolation level (temporary).
+                engine_to_use = Engine.Aurora
+            else:
+                assert self._router is not None
+                engine_to_use = await self._router.engine_for(query_rep, session)
 
             log_verbose(
                 logger,