Techsystech
/
server-tools
mirror of https://github.com/OCA/server-tools.git
3
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

session_db: improve resiliency to database errors 

Retry on OperationalError exception, which we receive on database restart.
Return cursor to pool when reconnecting.
pull/2531/head
Stéphane Bidoul 2023-01-21 13:27:21 +01:00
parent 061a51c354
commit 018b664940
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
session_db/pg_session_store.py
View File

@ -23,7 +23,7 @@ def with_cursor(func):
tries += 1 tries += 1
try: try:
return func(self, *args, **kwargs) return func(self, *args, **kwargs)
except psycopg2.InterfaceError as e: except (psycopg2.InterfaceError, psycopg2.OperationalError) as e:
_logger.info("Session in DB connection Retry %s/5" % tries) _logger.info("Session in DB connection Retry %s/5" % tries)
if tries > 4: if tries > 4:
raise e raise e
@ -49,6 +49,13 @@ class PGSessionStore(sessions.SessionStore):
def _open_connection(self): def _open_connection(self):
cnx = odoo.sql_db.db_connect(self._uri, allow_uri=True) cnx = odoo.sql_db.db_connect(self._uri, allow_uri=True)
try:
# return cursor to the pool
if self._cr is not None:
self._cr.close()
self._cr = None
except Exception: # pylint: disable=except-pass
pass
self._cr = cnx.cursor() self._cr = cnx.cursor()
self._cr._cnx.autocommit = True self._cr._cnx.autocommit = True