async def _nr_wrapper_Aredis_method_(wrapped, instance, args, kwargs):
transaction = current_transaction()
if transaction is None:
return await wrapped(*args, **kwargs)
dt = DatastoreTrace(product="Redis", target=None, operation=operation)
transaction._nr_datastore_instance_info = (None, None, None)
with dt:
result = await wrapped(*args, **kwargs)
host, port_path_or_id, db = transaction._nr_datastore_instance_info
dt.host = host
dt.port_path_or_id = port_path_or_id
dt.database_name = db
return result
def _nr_datastore_trace_wrapper_(wrapped, instance, args, kwargs):
transaction = current_transaction()
if transaction is None:
return wrapped(*args, **kwargs)
transaction._nr_datastore_instance_info = (None, None, None)
dt = DatastoreTrace(product, target, operation)
with dt:
result = wrapped(*args, **kwargs)
instance_info = transaction._nr_datastore_instance_info
(host, port_path_or_id, db) = instance_info
dt.host = host
dt.port_path_or_id = port_path_or_id
return result
def _nr_wrapper_Elasticsearch_method_(wrapped, instance, args, kwargs):
transaction = current_transaction()
if transaction is None:
return wrapped(*args, **kwargs)
# When arg_extractor is None, it means there is no target field
# associated with this method. Hence this method will only
# create an operation metric and no statement metric. This is
# handled by setting the target to None when calling the
# DatastoreTrace.
if arg_extractor is None:
index = None
else:
index = arg_extractor(*args, **kwargs)
if prefix:
operation = '%s.%s' % (prefix, name)
else:
operation = name
transaction._nr_datastore_instance_info = (None, None, None)
dt = DatastoreTrace(transaction=transaction,
product='Elasticsearch',
target=index,
operation=operation)
with dt:
result = wrapped(*args, **kwargs)
instance_info = transaction._nr_datastore_instance_info
host, port_path_or_id, _ = instance_info
dt.host = host
dt.port_path_or_id = port_path_or_id
return result