Handle exceptions when attempting rollback

mitdbg · May 12, 2024 · 24fb2c1 · 24fb2c1
1 parent a081664
commit 24fb2c1
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/workloads/chbenchmark/py-tpcc/pytpcc/runtime/executor.py b/workloads/chbenchmark/py-tpcc/pytpcc/runtime/executor.py
@@ -178,7 +178,14 @@ def execute(
                 if self.stop_on_error:
                     raise
                 r.abortTransaction(txn_id)
-                self.driver.ensureRollback()
+
+                try:
+                    self.driver.ensureRollback()
+                except:  # pylint: disable=bare-except
+                    # This may happen if we try to issue a rollback when the connection has dropped.
+                    verbose_logger.exception(
+                        "[T %d] Ran into error when running rollback.", worker_index
+                    )
 
                 # Back off slightly.
                 if backoff is None: