From 538f930c89a2e458ccb5301dafc363f25e9a3458 Mon Sep 17 00:00:00 2001 From: Geoffrey Yu Date: Sun, 12 Nov 2023 03:18:39 +0000 Subject: [PATCH] Workload adjustments --- experiments/15-e2e-scenarios-v2/scale_up/run_workload.sh | 1 + src/brad/front_end/front_end.py | 9 +++++++-- 2 files changed, 8 insertions(+), 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 index 063fd631..3d81b335 100755 --- 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 index c36aef9f..f36b06f8 100644 --- 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,