def _nr_aiohttp_request_wrapper_(wrapped, instance, args, kwargs):
transaction = current_transaction()
if transaction is None:
return wrapped(*args, **kwargs)
method, url = _bind_request(*args, **kwargs)
trace = ExternalTrace('aiohttp', str(url), method)
@asyncio.coroutine
def _coro():
try:
response = yield from wrapped(*args, **kwargs)
try:
trace.process_response_headers(response.headers.items())
except:
pass
return response
except Exception as e:
try:
trace.process_response_headers(e.headers.items())
except:
pass
raise
return async_wrapper(wrapped)(_coro, trace)()
def dynamic_wrapper(wrapped, instance, args, kwargs):
wrapper = async_wrapper(wrapped)
if not wrapper:
parent = current_trace()
if not parent:
return wrapped(*args, **kwargs)
else:
parent = None
if callable(url):
if instance is not None:
_url = url(instance, *args, **kwargs)
else:
_url = url(*args, **kwargs)
else:
_url = url
if callable(method):
if instance is not None:
_method = method(instance, *args, **kwargs)
else:
_method = method(*args, **kwargs)
else:
_method = method
trace = ExternalTrace(library, _url, _method, parent=parent)
if wrapper:
return wrapper(wrapped, trace)(*args, **kwargs)
with trace:
return wrapped(*args, **kwargs)
def _nr_database_trace_wrapper_(wrapped, instance, args, kwargs):
wrapper = async_wrapper(wrapped)
if not wrapper:
parent = current_trace()
if not parent:
return wrapped(*args, **kwargs)
else:
parent = None
if callable(sql):
if instance is not None:
_sql = sql(instance, *args, **kwargs)
else:
_sql = sql(*args, **kwargs)
else:
_sql = sql
trace = DatabaseTrace(_sql,
dbapi2_module,
parent=parent,
source=wrapped)
if wrapper: # pylint: disable=W0125,W0126
return wrapper(wrapped, trace)(*args, **kwargs)
with trace:
return wrapped(*args, **kwargs)
def literal_wrapper(wrapped, instance, args, kwargs):
wrapper = async_wrapper(wrapped)
if not wrapper:
parent = current_trace()
if not parent:
return wrapped(*args, **kwargs)
else:
parent = None
trace = ExternalTrace(library, url, method, parent=parent)
wrapper = async_wrapper(wrapped)
if wrapper:
return wrapper(wrapped, trace)(*args, **kwargs)
with trace:
return wrapped(*args, **kwargs)
def dynamic_wrapper(wrapped, instance, args, kwargs):
wrapper = async_wrapper(wrapped)
if not wrapper:
parent = current_trace()
if not parent:
return wrapped(*args, **kwargs)
else:
parent = None
if callable(name):
if instance is not None:
_name = name(instance, *args, **kwargs)
else:
_name = name(*args, **kwargs)
elif name is None:
_name = callable_name(wrapped)
else:
_name = name
if callable(group):
if instance is not None:
_group = group(instance, *args, **kwargs)
else:
_group = group(*args, **kwargs)
else:
_group = group
if callable(label):
if instance is not None:
_label = label(instance, *args, **kwargs)
else:
_label = label(*args, **kwargs)
else:
_label = label
if callable(params):
if instance is not None:
_params = params(instance, *args, **kwargs)
else:
_params = params(*args, **kwargs)
else:
_params = params
trace = FunctionTrace(_name, _group, _label, _params, terminal, rollup, parent=parent, source=wrapped)
if wrapper: # pylint: disable=W0125,W0126
return wrapper(wrapped, trace)(*args, **kwargs)
with trace:
return wrapped(*args, **kwargs)
def _nr_message_trace_wrapper_(wrapped, instance, args, kwargs):
wrapper = async_wrapper(wrapped)
if not wrapper:
parent = current_trace()
if not parent:
return wrapped(*args, **kwargs)
else:
parent = None
if callable(library):
if instance is not None:
_library = library(instance, *args, **kwargs)
else:
_library = library(*args, **kwargs)
else:
_library = library
if callable(operation):
if instance is not None:
_operation = operation(instance, *args, **kwargs)
else:
_operation = operation(*args, **kwargs)
else:
_operation = operation
if callable(destination_type):
if instance is not None:
_destination_type = destination_type(instance, *args, **kwargs)
else:
_destination_type = destination_type(*args, **kwargs)
else:
_destination_type = destination_type
if callable(destination_name):
if instance is not None:
_destination_name = destination_name(instance, *args, **kwargs)
else:
_destination_name = destination_name(*args, **kwargs)
else:
_destination_name = destination_name
trace = MessageTrace(_library,
_operation,
_destination_type,
_destination_name,
params={},
parent=parent,
source=wrapped)
if wrapper: # pylint: disable=W0125,W0126
return wrapper(wrapped, trace)(*args, **kwargs)
with trace:
return wrapped(*args, **kwargs)
def _nr_graphql_trace_wrapper_(wrapped, instance, args, kwargs):
wrapper = async_wrapper(wrapped)
if not wrapper:
parent = current_trace()
if not parent:
return wrapped(*args, **kwargs)
else:
parent = None
trace = GraphQLResolverTrace(parent=parent, source=wrapped)
if wrapper: # pylint: disable=W0125,W0126
return wrapper(wrapped, trace)(*args, **kwargs)
with trace:
return wrapped(*args, **kwargs)
def literal_wrapper(wrapped, instance, args, kwargs):
wrapper = async_wrapper(wrapped)
if not wrapper:
parent = current_trace()
if not parent:
return wrapped(*args, **kwargs)
else:
parent = None
_name = name or callable_name(wrapped)
trace = FunctionTrace(_name, group, label, params, terminal, rollup, parent=parent, source=wrapped)
if wrapper: # pylint: disable=W0125,W0126
return wrapper(wrapped, trace)(*args, **kwargs)
with trace:
return wrapped(*args, **kwargs)
def _nr_datastore_trace_wrapper_(wrapped, instance, args, kwargs):
wrapper = async_wrapper(wrapped)
if not wrapper:
parent = current_trace()
if not parent:
return wrapped(*args, **kwargs)
else:
parent = None
if callable(product):
if instance is not None:
_product = product(instance, *args, **kwargs)
else:
_product = product(*args, **kwargs)
else:
_product = product
if callable(target):
if instance is not None:
_target = target(instance, *args, **kwargs)
else:
_target = target(*args, **kwargs)
else:
_target = target
if callable(operation):
if instance is not None:
_operation = operation(instance, *args, **kwargs)
else:
_operation = operation(*args, **kwargs)
else:
_operation = operation
trace = DatastoreTrace(_product,
_target,
_operation,
parent=parent,
source=wrapped)
if wrapper: # pylint: disable=W0125,W0126
return wrapper(wrapped, trace)(*args, **kwargs)
with trace:
return wrapped(*args, **kwargs)
def _nr_wrapper_memcache_trace_(wrapped, instance, args, kwargs):
wrapper = async_wrapper(wrapped)
if not wrapper:
parent = current_trace()
if not parent:
return wrapped(*args, **kwargs)
else:
parent = None
if callable(command):
if instance is not None:
_command = command(instance, *args, **kwargs)
else:
_command = command(*args, **kwargs)
else:
_command = command
trace = MemcacheTrace(_command, parent=parent)
if wrapper:
return wrapper(wrapped, trace)(*args, **kwargs)
with trace:
return wrapped(*args, **kwargs)
