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

feat(backend): only log pruned queue items / db freed space if > 0

Browse Source
This commit is contained in:
psychedelicious
2023-12-01 09:26:44 +11:00
committed by Kent Keirsey
parent 0228aba06f
commit 3f0e0af177
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/app/services/session_queue/session_queue_sqlite.py
View File

@ -42,6 +42,7 @@ class SqliteSessionQueue(SessionQueueBase):
self._set_in_progress_to_canceled() self._set_in_progress_to_canceled()
prune_result = self.prune(DEFAULT_QUEUE_ID) prune_result = self.prune(DEFAULT_QUEUE_ID)
local_handler.register(event_name=EventServiceBase.queue_event, _func=self._on_session_event) local_handler.register(event_name=EventServiceBase.queue_event, _func=self._on_session_event)
if prune_result.deleted > 0:
self.__invoker.services.logger.info(f"Pruned {prune_result.deleted} finished queue items") self.__invoker.services.logger.info(f"Pruned {prune_result.deleted} finished queue items")
def __init__(self, db: SqliteDatabase) -> None: def __init__(self, db: SqliteDatabase) -> None:

1
invokeai/app/services/shared/sqlite.py
View File

@ -41,6 +41,7 @@ class SqliteDatabase:
self.conn.commit() self.conn.commit()
final_db_size = Path(self.db_path).stat().st_size final_db_size = Path(self.db_path).stat().st_size
freed_space_in_mb = round((initial_db_size - final_db_size) / 1024 / 1024, 2) freed_space_in_mb = round((initial_db_size - final_db_size) / 1024 / 1024, 2)
if freed_space_in_mb > 0:
self._logger.info(f"Cleaned database (freed {freed_space_in_mb}MB)") self._logger.info(f"Cleaned database (freed {freed_space_in_mb}MB)")
except Exception as e: except Exception as e:
self._logger.error(f"Error cleaning database: {e}") self._logger.error(f"Error cleaning database: {e}")