fix: failed to recover log consumer

RockChinQ · Oct 17, 2023 · e54f974 · e54f974
1 parent 4e34668
commit e54f974
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/free_one_api/impls/log.py b/free_one_api/impls/log.py
@@ -1,5 +1,6 @@
 import logging
 import asyncio
+import traceback
 
 from ..models.database import db
 
@@ -17,9 +18,12 @@ def __init__(self, dbmgr: db.DatabaseInterface, level=logging.NOTSET):
 
     async def _consumer(self):
         while True:
-            record: logging.LogRecord = await self.queue.get()
-            await self.dbmgr.insert_log(record.created, record.getMessage())
-            self.queue.task_done()
+            try:
+                record: logging.LogRecord = await self.queue.get()
+                await self.dbmgr.insert_log(record.created, record.getMessage())
+                self.queue.task_done()
+            except Exception as e:
+                print(traceback.format_exc())
 
     def emit(self, record: logging.LogRecord):
         """Emit a record."""