Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(db): fix bug with releasing without lock in db.clean()

Browse Source
This commit is contained in:
psychedelicious 2023-12-05 06:48:53 +11:00
parent 7d4a78e470
commit 10cf10c16c
1 changed files with 15 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/app/services/shared/sqlite/sqlite_database.py
View File

@ -31,6 +31,7 @@ class SqliteDatabase:
self.conn.execute("PRAGMA foreign_keys = ON;") self.conn.execute("PRAGMA foreign_keys = ON;")
def clean(self) -> None: def clean(self) -> None:
with self.lock:
try: try:
if self.db_path == sqlite_memory: if self.db_path == sqlite_memory:
return return
@ -45,5 +46,3 @@ class SqliteDatabase:
except Exception as e: except Exception as e:
self._logger.error(f"Error cleaning database: {e}") self._logger.error(f"Error cleaning database: {e}")
raise raise
finally:
self.lock.release()