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

Merge pull request #5 from nvllsvm/on_error

Browse source 
Make on_error optional for PostgresConnector
This commit is contained in:
Gavin M. Roy 2020-09-16 19:28:52 -04:00 committed by GitHub
commit b7ddd137b6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
sprockets_postgres.py
View file

@ -110,7 +110,7 @@ class PostgresConnector:
""" """
def __init__(self, def __init__(self,
cursor: aiopg.Cursor, cursor: aiopg.Cursor,
on_error: typing.Callable, on_error: typing.Optional[typing.Callable] = None,
on_duration: typing.Optional[typing.Callable] = None, on_duration: typing.Optional[typing.Callable] = None,
timeout: Timeout = None): timeout: Timeout = None):
self.cursor = cursor self.cursor = cursor
@ -246,9 +246,10 @@ class PostgresConnector:
try: try:
await method(**kwargs) await method(**kwargs)
except (asyncio.TimeoutError, psycopg2.Error) as err: except (asyncio.TimeoutError, psycopg2.Error) as err:
exc = self._on_error(metric_name, err) if self._on_error:
if exc: err = self._on_error(metric_name, err)
raise exc if err:
raise err
else: else:
results = await self._query_results() results = await self._query_results()
if self._on_duration: if self._on_duration:
@ -313,7 +314,8 @@ class ApplicationMixin:
@contextlib.asynccontextmanager @contextlib.asynccontextmanager
async def postgres_connector(self, async def postgres_connector(self,
on_error: typing.Callable, on_error: typing.Optional[
typing.Callable] = None,
on_duration: typing.Optional[ on_duration: typing.Optional[
typing.Callable] = None, typing.Callable] = None,
timeout: Timeout = None, timeout: Timeout = None,