add sqlite db reset threshold (#868)

* add sqlite db reset threshold * refresh session --------- Co-authored-by: noO0ob <[email protected]>
Meituan-Dianping · Jul 4, 2024 · 76791cc · 76791cc
1 parent ec1633e
commit 76791cc
Showing 1 changed file with 12 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,12 +164,19 @@ 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()
+                    session = self._scoped_session()
+                else:
+                    session.rollback()
             except Exception:
                 logger.error(f'Save event failed. {traceback.format_exc()}')
+                session.rollback()
 
     def stop(self):
         super().stop()