Merge pull request #298 from zhanghaofei/begin_transaction_reconnect_…

…by_lost If mysql begin a transaction and lost connection, reconnect
sdispater · Jul 26, 2019 · e5d48d6 · e5d48d6
2 parents 8aaeafe + 5dd2702
commit e5d48d6
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/orator/connections/mysql_connection.py b/orator/connections/mysql_connection.py
@@ -38,7 +38,16 @@ def get_schema_manager(self):
         return MySQLSchemaManager(self)
 
     def begin_transaction(self):
-        self._connection.autocommit(False)
+        self._reconnect_if_missing_connection()
+
+        try:
+            self._connection.autocommit(False)
+        except Exception as e:
+            if self._caused_by_lost_connection(e):
+                self.reconnect()
+                self._connection.autocommit(False)
+            else:
+                raise
 
         super(MySQLConnection, self).begin_transaction()