github/sprockets-postgres
1
0
Fork 0
mirror of https://github.com/sprockets/sprockets-postgres.git synced 2024-12-27 20:24:44 +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,
cursor: aiopg.Cursor,
on_error: typing.Callable,
on_error: typing.Optional[typing.Callable] = None,
on_duration: typing.Optional[typing.Callable] = None,
timeout: Timeout = None):
self.cursor = cursor
@ -246,9 +246,10 @@ class PostgresConnector:
try:
await method(**kwargs)
except (asyncio.TimeoutError, psycopg2.Error) as err:
exc = self._on_error(metric_name, err)
if exc:
raise exc
if self._on_error:
err = self._on_error(metric_name, err)
if err:
raise err
else:
results = await self._query_results()
if self._on_duration:
@ -313,7 +314,8 @@ class ApplicationMixin:
@contextlib.asynccontextmanager
async def postgres_connector(self,
on_error: typing.Callable,
on_error: typing.Optional[
typing.Callable] = None,
on_duration: typing.Optional[
typing.Callable] = None,
timeout: Timeout = None,