highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-06-11 00:20:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

always finalize the serialized transaction so that it doesn't leak ou… (#2843)

Browse Source 
* always finalize the serialized transaction so that it doesn't leak outside the function

* re-raise the exception and log it
This commit is contained in:
rkuo-danswer 2024-10-17 16:13:57 -07:00 committed by GitHub
parent b169f78699
commit 4c2cf8b132
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 30 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
backend/danswer/background/indexing/run_indexing.py
View File

@ -387,6 +387,7 @@ def _prepare_index_attempt(
# after the next commit:
# https://docs.sqlalchemy.org/en/20/orm/session_transaction.html#setting-isolation-for-individual-transactions
db_session.connection(execution_options={"isolation_level": "SERIALIZABLE"}) # type: ignore
try:
if tenant_id is not None:
# Explicitly set the search path for the given tenant
db_session.execute(text(f'SET search_path TO "{tenant_id}"'))
@ -401,7 +402,9 @@ def _prepare_index_attempt(
)
if attempt is None:
raise RuntimeError(f"Unable to find IndexAttempt for ID '{index_attempt_id}'")
raise RuntimeError(
f"Unable to find IndexAttempt for ID '{index_attempt_id}'"
)
if attempt.status != IndexingStatus.NOT_STARTED:
raise RuntimeError(
@ -409,9 +412,15 @@ def _prepare_index_attempt(
f"Current status is '{attempt.status}'."
)
# only commit once, to make sure this all happens in a single transaction
mark_attempt_in_progress(attempt, db_session)
# only commit once, to make sure this all happens in a single transaction
db_session.commit()
except Exception:
db_session.rollback()
logger.exception("_prepare_index_attempt exceptioned.")
raise
return attempt