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

Delay the initialization of asyncio objects to on_start

Browse source
This commit is contained in:
Gavin M. Roy 2020-07-08 14:33:04 -04:00
parent 70cf5c005b
commit 4f60c7bf0a
2 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
VERSION
View file

@ -1 +1 @@
1.3.1 1.3.2

7
sprockets_postgres.py
View file

@ -290,8 +290,8 @@ class ApplicationMixin:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self._postgres_pool: typing.Optional[pool.Pool] = None self._postgres_pool: typing.Optional[pool.Pool] = None
self._postgres_connected = asyncio.Event() self._postgres_connected: typing.Optional[asyncio.Event] = None
self._postgres_reconnect = asyncio.Lock() self._postgres_reconnect: typing.Optional[asyncio.Lock] = None
self._postgres_srv: bool = False self._postgres_srv: bool = False
self.runner_callbacks['on_start'].append(self._postgres_on_start) self.runner_callbacks['on_start'].append(self._postgres_on_start)
self.runner_callbacks['shutdown'].append(self._postgres_shutdown) self.runner_callbacks['shutdown'].append(self._postgres_shutdown)
@ -480,6 +480,9 @@ class ApplicationMixin:
callback mechanism. callback mechanism.
""" """
self._postgres_connected = asyncio.Event()
self._postgres_reconnect = asyncio.Lock()
if 'POSTGRES_URL' not in os.environ: if 'POSTGRES_URL' not in os.environ:
LOGGER.critical('Missing POSTGRES_URL environment variable') LOGGER.critical('Missing POSTGRES_URL environment variable')
return self.stop(loop) return self.stop(loop)