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