diff --git a/src/risotto/dispatcher.py b/src/risotto/dispatcher.py index e9244c7..2572a88 100644 --- a/src/risotto/dispatcher.py +++ b/src/risotto/dispatcher.py @@ -147,40 +147,39 @@ class PublishDispatcher: check_role, kwargs, function_objs) - else: - try: - async with self.pool.acquire() as connection: - await connection.set_type_codec( - 'json', - encoder=dumps, - decoder=loads, - schema='pg_catalog' - ) - risotto_context.connection = connection - async with connection.transaction(): - return await self.launch(version, - message, - risotto_context, - check_role, - kwargs, - function_objs) - except CallError as err: - raise err - except Exception as err: - # if there is a problem with arguments, just send an error and do nothing - if get_config()['global']['debug']: - print_exc() - async with self.pool.acquire() as connection: - await connection.set_type_codec( - 'json', - encoder=dumps, - decoder=loads, - schema='pg_catalog' - ) - risotto_context.connection = connection - async with connection.transaction(): - await log.error_msg(risotto_context, kwargs, err) - raise err + try: + async with self.pool.acquire() as connection: + await connection.set_type_codec( + 'json', + encoder=dumps, + decoder=loads, + schema='pg_catalog' + ) + risotto_context.connection = connection + async with connection.transaction(): + return await self.launch(version, + message, + risotto_context, + check_role, + kwargs, + function_objs) + except CallError as err: + raise err + except Exception as err: + # if there is a problem with arguments, just send an error and do nothing + if get_config()['global']['debug']: + print_exc() + async with self.pool.acquire() as connection: + await connection.set_type_codec( + 'json', + encoder=dumps, + decoder=loads, + schema='pg_catalog' + ) + risotto_context.connection = connection + async with connection.transaction(): + await log.error_msg(risotto_context, kwargs, err) + raise err class Dispatcher(register.RegisterDispatcher, @@ -329,8 +328,6 @@ class Dispatcher(register.RegisterDispatcher, if key in function_obj['arguments']: kw[key] = value - - kw['risotto_context'] = risotto_context returns = await function(self.injected_self[function_obj['module']], **kw) if risotto_context.type == 'rpc':