github/sprockets-postgres
1
0
Fork 0
mirror of https://github.com/sprockets/sprockets-postgres.git synced 2024-12-28 03:00:18 +00:00
Code Issues Projects Releases Packages Wiki Activity

Actually wait while the first in reconnects

Browse source
This commit is contained in:
Gavin M. Roy 2020-07-07 16:21:01 -04:00
parent ecc070e974
commit a1736bccae
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
sprockets_postgres.py
View file

@ -344,14 +344,17 @@ class ApplicationMixin:
message = str(err)
if isinstance(err, psycopg2.OperationalError) and _attempt == 1:
LOGGER.critical('Disconnected from Postgres: %s', err)
retry = True
if not self._postgres_reconnect.locked():
async with self._postgres_reconnect:
if await self._postgres_connect():
async with self.postgres_connector(
on_error, on_duration, timeout,
_attempt + 1) as connector:
yield connector
return
retry = await self._postgres_connect()
if retry:
await self._postgres_connected.wait()
async with self.postgres_connector(
on_error, on_duration, timeout,
_attempt + 1) as connector:
yield connector
return
message = 'disconnected'
exc = on_error('postgres_connector', ConnectionException(message))
if exc: