add sqlite db reset threshold

Meituan-Dianping · Jun 28, 2024 · cf3ad13 · cf3ad13
1 parent 73d35cb
commit cf3ad13
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/lyrebird/db/database_server.py b/lyrebird/db/database_server.py
@@ -34,6 +34,8 @@
 class LyrebirdDatabaseServer(ThreadServer):
     def __init__(self, path=None):
         self.database_uri = None
+        self.error_log = []
+        self.error_log_threshold = application.config.get('event.db_connection_recover_threshold', 0)
         super().__init__()
 
         if not path or path.isspace():
@@ -162,10 +164,15 @@ def run(self):
                 session.add(event)
                 session.commit()
                 context.emit('db_action', 'add event log')
-            except OperationalError:
+            except OperationalError as e:
                 logger.error(f'Save event failed. {traceback.format_exc()}')
-                logger.warning(f'DB would be reset: {self.database_uri}')
-                self.reset()
+                self.error_log.append(e)
+                if len(self.error_log) > self.error_log_threshold:
+                    logger.warning(f'DB would be reset: {self.database_uri}')
+                    self.error_log = []
+                    self.reset()
+                else:
+                    session.rollback()
             except Exception:
                 logger.error(f'Save event failed. {traceback.format_exc()}')