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
with DatastoreTrace(transaction,
product='Elasticsearch',
target=index,
operation=operation):
return wrapped(*args, **kwargs)
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(transaction, 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_Redis_method_(wrapped, instance, args, kwargs):
transaction = current_transaction()
if transaction is None:
return wrapped(*args, **kwargs)
dt = DatastoreTrace(transaction,
product='Redis',
target=None,
operation=operation)
transaction._nr_datastore_instance_info = (None, None, None)
with dt:
result = 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_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
def _nr_wrapper_Redis_method_(wrapped, instance, args, kwargs):
transaction = current_transaction()
if transaction is None:
return wrapped(*args, **kwargs)
dt = DatastoreTrace(
transaction,
product='Redis',
target=None,
operation=operation
)
transaction._nr_datastore_instance_info = (None, None, None)
with dt:
result = 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_Connection_send_command_wrapper_(wrapped, instance, args, kwargs):
transaction = current_transaction()
if transaction is None or not args:
return wrapped(*args, **kwargs)
host, port_path_or_id, db = (None, None, None)
try:
dt = transaction.settings.datastore_tracer
if (dt.instance_reporting.enabled
or dt.database_name_reporting.enabled):
conn_kwargs = _conn_attrs_to_dict(instance)
host, port_path_or_id, db = _instance_info(conn_kwargs)
except:
pass
transaction._nr_datastore_instance_info = (host, port_path_or_id, db)
# Older Redis clients would when sending multi part commands pass
# them in as separate arguments to send_command(). Need to therefore
# detect those and grab the next argument from the set of arguments.
operation = args[0].strip().lower()
# If it's not a multi part command, there's no need to trace it, so
# we can return early.
if operation.split()[0] not in _redis_multipart_commands:
return wrapped(*args, **kwargs)
# Convert multi args to single arg string
if operation in _redis_multipart_commands and len(args) > 1:
operation = '%s %s' % (operation, args[1].strip().lower())
operation = _redis_operation_re.sub('_', operation)
with DatastoreTrace(transaction,
product='Redis',
target=None,
operation=operation,
host=host,
port_path_or_id=port_path_or_id,
database_name=db):
return wrapped(*args, **kwargs)
def _nr_Connection_send_command_wrapper_(wrapped, instance, args, kwargs):
transaction = current_transaction()
if transaction is None or not args:
return wrapped(*args, **kwargs)
operation = args[0].strip().lower()
# Older Redis clients would when sending multi part commands pass
# them in as separate arguments to send_command(). Need to therefore
# detect those and grab the next argument from the set of arguments.
if operation in _redis_multipart_commands and len(args) > 1:
operation = '%s %s' % (operation, args[1].strip().lower())
operation = _redis_operation_re.sub('_', operation)
with DatastoreTrace(transaction,
product='Redis',
target=None,
operation=operation):
return wrapped(*args, **kwargs)
