chore: move check_implicit_commit() call back to its original location

Signed-off-by: Akhil Narang <me@akhilnarang.dev>
This commit is contained in:
Akhil Narang 2023-10-20 12:35:55 +05:30
parent 806748b063
commit 6b49b5f32a
No known key found for this signature in database
GPG key ID: 9DCC61E211BF645F

View file

@ -375,12 +375,11 @@ class Database:
"""Raises exception if more than 200,000 `INSERT`, `UPDATE` queries are
executed in one transaction. This is to ensure that writes are always flushed otherwise this
could cause the system to hang."""
self.check_implicit_commit(query)
if query and is_query_type(query, ("commit", "rollback")):
self.transaction_writes = 0
self.check_implicit_commit(query)
if query[:6].lower() in ("update", "insert", "delete"):
self.transaction_writes += 1
if self.transaction_writes > self.MAX_WRITES_PER_TRANSACTION: