Skip to content

Commit 4a8209a

Browse files
committed
add transaction logging
1 parent 8db18d9 commit 4a8209a

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

django_mongodb_backend/base.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33

44
from django.core.exceptions import ImproperlyConfigured
55
from django.db.backends.base.base import BaseDatabaseWrapper
6+
from django.db.backends.utils import debug_transaction
67
from django.utils.asyncio import async_unsafe
78
from django.utils.functional import cached_property
89
from pymongo.collection import Collection
@@ -194,21 +195,24 @@ def _commit(self):
194195
if not self.features.supports_transactions:
195196
return
196197
if self.session:
197-
self.session.commit_transaction()
198+
with debug_transaction(self, "session.commit_transaction()"):
199+
self.session.commit_transaction()
198200
self.session.end_session()
199201
self.session = None
200202

201203
def _rollback(self):
202204
if not self.features.supports_transactions:
203205
return
204206
if self.session:
205-
self.session.abort_transaction()
207+
with debug_transaction(self, "session.abort_transaction()"):
208+
self.session.abort_transaction()
206209
self.session = None
207210

208211
def _start_session(self):
209212
if self.session is None:
210213
self.session = self.connection.start_session()
211-
self.session.start_transaction()
214+
with debug_transaction(self, "session.start_transaction()"):
215+
self.session.start_transaction()
212216

213217
def _start_transaction_under_autocommit(self):
214218
if not self.features.supports_transactions:

0 commit comments

Comments
 (0)