Croc/croccybot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix (data): Add reconnection on trans fail.

Browse Source
This commit is contained in:
Conatum
2023-08-19 13:10:53 +03:00
parent 01a4f7f2bf
commit e9c879da7b
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/data/connector.py
View File

@@ -2,6 +2,7 @@ from typing import Protocol, runtime_checkable, Callable, Awaitable
import logging import logging
import psycopg as psq import psycopg as psq
from psycopg.pq import TransactionStatus
from .cursor import AsyncLoggingCursor from .cursor import AsyncLoggingCursor
@@ -27,6 +28,11 @@ class Connector:
# TODO: Reconnection logic? # TODO: Reconnection logic?
if not self.conn: if not self.conn:
raise ValueError("Attempting to get connection before initialisation!") raise ValueError("Attempting to get connection before initialisation!")
if self.conn.info.transaction_status is TransactionStatus.INERROR:
await self.connect()
logger.error(
"Database connection transaction failed!! This should not happen. Reconnecting."
)
return self.conn return self.conn
async def connect(self) -> psq.AsyncConnection: async def connect(self) -> psq.AsyncConnection: