fix(db): missing rollback on execute_batch method
Arguably more important on a batch function to have rollback.
This commit is contained in:
@@ -38,9 +38,13 @@ class PostgresConnector:
|
|||||||
raise
|
raise
|
||||||
|
|
||||||
def execute_batch(self, query, values):
|
def execute_batch(self, query, values):
|
||||||
|
try:
|
||||||
with self.connection.cursor(cursor_factory=RealDictCursor) as cursor:
|
with self.connection.cursor(cursor_factory=RealDictCursor) as cursor:
|
||||||
execute_batch(cursor, query, values)
|
execute_batch(cursor, query, values)
|
||||||
self.connection.commit()
|
self.connection.commit()
|
||||||
|
except Exception:
|
||||||
|
self.connection.rollback()
|
||||||
|
raise
|
||||||
|
|
||||||
def close(self):
|
def close(self):
|
||||||
if self.connection:
|
if self.connection:
|
||||||
|
|||||||
Reference in New Issue
Block a user