Module: NewRelic::Agent

Extended by:
Forwardable, Agent, SupportabilityHelper
Included in:
Agent
Defined in:
lib/new_relic/agent.rb,
lib/new_relic/agent/method_tracer.rb,
lib/new_relic/agent/distributed_tracing.rb,
lib/new_relic/agent/external.rb,
lib/new_relic/agent/instrumentation/controller_instrumentation.rb,
lib/new_relic/agent/datastores.rb,
lib/new_relic/agent/messaging.rb,
lib/new_relic/agent/sql_sampler.rb,
lib/new_relic/agent/tracer.rb,
lib/new_relic/agent/transaction.rb,
lib/new_relic/agent/transaction_sampler.rb,
lib/new_relic/agent/transaction/external_request_segment.rb

Overview

This module contains most of the public API methods for the Ruby Agent.

For adding custom instrumentation to method invocations, see the docs for MethodTracer and MethodTracer::ClassMethods.

For information on how to trace transactions in non-Rack contexts, see Instrumentation::ControllerInstrumentation.

For general documentation about the Ruby agent, see: docs.newrelic.com/docs/agents/ruby-agent

Defined Under Namespace

Modules: Datastores, DistributedTracing, External, Instrumentation, Messaging, MethodTracer Classes: SqlSampler, Tracer, Transaction, TransactionSampler

Recording custom metrics collapse

Recording custom errors collapse

Recording custom Insights events collapse

LLM callbacks collapse

Manual agent configuration and startup/shutdown collapse

Ignoring or excluding data collapse

Adding custom attributes to traces collapse

Transaction naming collapse

Trace and Entity metadata collapse

Manual browser monitoring configuration collapse

Instance Method Details

#add_custom_attributes(params) ⇒ Object

Add attributes to the transaction trace, Insights Transaction event, and any traced errors recorded for the current transaction.

If Browser Monitoring is enabled, and the browser_monitoring.attributes.enabled configuration setting is true, these custom attributes will also be present in the script injected into the response body, making them available on Insights PageView events.

Parameters:

  • params (Hash)

    A Hash of attributes to be attached to the transaction event. Keys should be strings or symbols, and values may be strings, symbols, numeric values or booleans.



803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
# File 'lib/new_relic/agent.rb', line 803

def add_custom_attributes(params) # THREAD_LOCAL_ACCESS
  record_api_supportability_metric(:add_custom_attributes)

  unless params.is_a?(Hash)
    ::NewRelic::Agent.logger.warn("Bad argument passed to #add_custom_attributes. Expected Hash but got #{params.class}")
    return
  end

  if NewRelic::Agent.agent&.serverless?
    ::NewRelic::Agent.logger.warn('Custom attributes are not supported in serverless mode')
    return
  end

  Transaction.tl_current&.add_custom_attributes(params)
  segment = ::NewRelic::Agent::Tracer.current_segment
  add_new_segment_attributes(params, segment) if segment
end

#add_custom_log_attributes(params) ⇒ Object

Add global custom attributes to log events for the current agent instance. As these attributes are global to the agent instance, they will be attached to all log events generated by the agent, and this methods usage isn’t suitable for setting dynamic values.

Parameters:

  • params (Hash)

    A Hash of attributes to attach to log events. The agent accepts up to 240 custom log event attributes.

    Keys will be coerced into Strings and must be less than 256 characters. Keys longer than 255 characters will be truncated.

    Values may be Strings, Symbols, numeric values or Booleans and must be less than 4095 characters. If the value is a String or a Symbol, values longer than 4094 characters will be truncated. If the value exceeds 4094 characters and is of a different class, the attribute pair will be dropped.

    This API can be called multiple times. If the same key is passed more than once, the value associated with the last call will be preserved.

    Attribute pairs with empty or nil contents will be dropped.



883
884
885
886
887
888
889
890
891
# File 'lib/new_relic/agent.rb', line 883

def add_custom_log_attributes(params)
  record_api_supportability_metric(:add_custom_log_attributes)

  if params.is_a?(Hash)
    NewRelic::Agent.agent.log_event_aggregator.add_custom_attributes(params)
  else
    NewRelic::Agent.logger.warn("Bad argument passed to #add_custom_log_attributes. Expected Hash but got #{params.class}.")
  end
end

#add_custom_span_attributes(params) ⇒ Object

Add custom attributes to the span event for the current span. Attributes will be visible on spans in the New Relic Distributed Tracing UI and on span events in New Relic Insights.

Custom attributes will not be transmitted when high_security setting is enabled or custom_attributes setting is disabled.

Parameters:

  • params (Hash)

    A Hash of attributes to be attached to the span event. Keys should be strings or symbols, and values may be strings, symbols, numeric values or booleans.

See Also:



842
843
844
845
846
847
848
849
850
851
852
# File 'lib/new_relic/agent.rb', line 842

def add_custom_span_attributes(params)
  record_api_supportability_metric(:add_custom_span_attributes)

  if params.is_a?(Hash)
    if segment = NewRelic::Agent::Tracer.current_segment
      segment.add_custom_attributes(params)
    end
  else
    ::NewRelic::Agent.logger.warn("Bad argument passed to #add_custom_span_attributes. Expected Hash but got #{params.class}")
  end
end

#add_instrumentation(file_pattern) ⇒ Object

Add instrumentation files to the agent. The argument should be a glob matching ruby scripts which will be executed at the time instrumentation is loaded. Since instrumentation is not loaded when the agent is not running it’s better to use this method to register instrumentation than just loading the files directly, although that probably also works.



665
666
667
668
# File 'lib/new_relic/agent.rb', line 665

def add_instrumentation(file_pattern)
  record_api_supportability_metric(:add_instrumentation)
  NewRelic::Control.instance.add_instrumentation(file_pattern)
end

#after_fork(options = {}) ⇒ Object

Register this method as a callback for processes that fork jobs.

If the master/parent connects to the agent prior to forking the agent in the forked process will use that agent_run. Otherwise the forked process will establish a new connection with the server.

Use this especially when you fork the process to run background jobs or other work. If you are doing this with a web dispatcher that forks worker processes then you will need to force the agent to reconnect, which it won’t do by default. Passenger and Unicorn are already handled, nothing special needed for them.

Options:

  • :force_reconnect => true to force the spawned process to establish a new connection, such as when forking a long running process. The default is false–it will only connect to the server if the parent had not connected.

  • :keep_retrying => false if we try to initiate a new connection, this tells me to only try it once so this method returns quickly if there is some kind of latency with the server.



628
629
630
631
632
# File 'lib/new_relic/agent.rb', line 628

def after_fork(options = {})
  record_api_supportability_metric(:after_fork)
  # the following line needs else branch coverage
  agent.after_fork(options) if agent # rubocop:disable Style/SafeNavigation
end

#browser_timing_header(nonce = nil) ⇒ Object

This method returns a string suitable for inclusion in a page - known as ‘manual instrumentation’ for Real User Monitoring. Can return either a script tag with associated javascript, or in the case of disabled Real User Monitoring, an empty string

This is the header string - it should be placed as high in the page as is reasonably possible - that is, before any style or javascript inclusions, but after any header-related meta tags

In previous agents there was a corresponding footer required, but all the work is now done by this single method.

Parameters:

  • nonce (String) (defaults to: nil)

    The nonce to use in the javascript tag for browser instrumentation



1049
1050
1051
1052
1053
1054
1055
# File 'lib/new_relic/agent.rb', line 1049

def browser_timing_header(nonce = nil)
  record_api_supportability_metric(:browser_timing_header)

  return EMPTY_STR unless agent

  agent.javascript_instrumentor.browser_timing_header(nonce)
end

#disable_all_tracingObject

Yield to the block without collecting any metrics or traces in any of the subsequent calls. If executed recursively, will keep track of the first entry point and turn on tracing again after leaving that block. This uses the thread local Tracer::State.



741
742
743
744
745
746
747
748
749
750
751
752
# File 'lib/new_relic/agent.rb', line 741

def disable_all_tracing
  record_api_supportability_metric(:disable_all_tracing)

  return yield unless agent

  begin
    agent.push_trace_execution_flag(false)
    yield
  ensure
    agent.pop_trace_execution_flag
  end
end

#disable_sql_recordingObject

This method sets the state of sql recording in the transaction sampler feature. Within the given block, no sql will be recorded

usage:

NewRelic::Agent.disable_sql_recording do
  ...
end


765
766
767
768
769
770
771
772
773
774
775
776
# File 'lib/new_relic/agent.rb', line 765

def disable_sql_recording
  record_api_supportability_metric(:disable_sql_recording)

  return yield unless agent

  state = agent.set_record_sql(false)
  begin
    yield
  ensure
    agent.set_record_sql(state)
  end
end

#drop_buffered_dataObject

Clear out any data the agent has buffered but has not yet transmitted to the collector.



650
651
652
653
654
# File 'lib/new_relic/agent.rb', line 650

def drop_buffered_data
  # the following line needs else branch coverage
  agent.drop_buffered_data if agent # rubocop:disable Style/SafeNavigation
  record_api_supportability_metric(:drop_buffered_data)
end

#get_transaction_nameObject

Get the name of the current running transaction. This is useful if you want to modify the default name.



952
953
954
955
956
957
958
959
960
# File 'lib/new_relic/agent.rb', line 952

def get_transaction_name # THREAD_LOCAL_ACCESS
  record_api_supportability_metric(:get_transaction_name)

  txn = Transaction.tl_current
  if txn
    namer = Instrumentation::ControllerInstrumentation::TransactionNamer
    txn.best_name.sub(Regexp.new("\\A#{Regexp.escape(namer.prefix_for_category(txn))}"), '')
  end
end

#ignore_apdexObject

This method disables the recording of Apdex metrics in the current transaction.



719
720
721
722
# File 'lib/new_relic/agent.rb', line 719

def ignore_apdex
  record_api_supportability_metric(:ignore_apdex)
  NewRelic::Agent::Transaction.tl_current&.ignore_apdex!
end

#ignore_enduserObject

This method disables browser monitoring javascript injection in the current transaction.



729
730
731
732
# File 'lib/new_relic/agent.rb', line 729

def ignore_enduser
  record_api_supportability_metric(:ignore_enduser)
  NewRelic::Agent::Transaction.tl_current&.ignore_enduser!
end

#ignore_error_filter(&block) ⇒ Object

Set a filter to be applied to errors that the Ruby agent will track. The block should evaluate to the exception to track (which could be different from the original exception) or nil to ignore this exception.

The block is yielded to with the exception to filter.

Return the new block or the existing filter Proc if no block is passed.



365
366
367
368
369
370
371
372
373
# File 'lib/new_relic/agent.rb', line 365

def ignore_error_filter(&block)
  record_api_supportability_metric(:ignore_error_filter)

  if block
    NewRelic::Agent::ErrorCollector.ignore_error_filter = block
  else
    NewRelic::Agent::ErrorCollector.ignore_error_filter
  end
end

#ignore_transactionObject

This method disables the recording of the current transaction. No metrics, traced errors, transaction traces, Insights events, slow SQL traces, or RUM injection will happen for this transaction.



709
710
711
712
# File 'lib/new_relic/agent.rb', line 709

def ignore_transaction
  record_api_supportability_metric(:ignore_transaction)
  NewRelic::Agent::Transaction.tl_current&.ignore!
end

#increment_metric(metric_name, amount = 1) ⇒ Object

Increment a simple counter metric.

metric_name should follow a slash separated path convention. Application specific metrics should begin with “Custom/”.

This method is safe to use from any thread.



338
339
340
341
342
343
344
345
346
347
348
# File 'lib/new_relic/agent.rb', line 338

def increment_metric(metric_name, amount = 1) # THREAD_LOCAL_ACCESS
  return unless agent

  if amount == 1
    metrics = [metric_name, SUPPORTABILITY_INCREMENT_METRIC]
    agent.stats_engine.tl_record_unscoped_metrics(metrics) { |stats| stats.increment_count }
  else
    agent.stats_engine.tl_record_unscoped_metrics(metric_name) { |stats| stats.increment_count(amount) }
    agent.stats_engine.tl_record_unscoped_metrics(SUPPORTABILITY_INCREMENT_METRIC) { |stats| stats.increment_count }
  end
end

#linking_metadataObject

Returns a new hash containing trace and entity metadata that can be used to relate data to a trace or to an entity in APM.

This hash includes:

  • trace.id - The current trace id, if there is a current trace id. This value may be omitted.

  • span.id - The current span id, if there is a current span. This value may be omitted.

  • entity.name - The name of the current application. This is read from the app_name key in your config. If there are multiple application names, the first one is used.

  • entity.type - The entity type is hardcoded to the string ‘SERVICE’.

  • entity.guid - The guid of the current entity.

  • hostname - The fully qualified hostname.



1022
1023
1024
1025
1026
1027
# File 'lib/new_relic/agent.rb', line 1022

def 
   = Hash.new
  LinkingMetadata.()
  LinkingMetadata.()
  
end

#manual_start(options = {}) ⇒ Object

Call this to manually start the agent in situations where the agent does not auto-start.

When the app environment loads, so does the agent. However, the agent will only connect to the service if a web front-end is found. If you want to selectively monitor ruby processes that don’t use web plugins, then call this method in your code and the agent will fire up and start reporting to the service.

Options are passed in as overrides for values in the newrelic.yml, such as app_name. In addition, the option log will take a logger that will be used instead of the standard file logger. The setting for the newrelic.yml section to use (ie, RAILS_ENV) can be overridden with an :env argument.



596
597
598
599
600
601
# File 'lib/new_relic/agent.rb', line 596

def manual_start(options = {})
  raise 'Options must be a hash' unless Hash === options

  NewRelic::Control.instance.init_plugin({:agent_enabled => true, :sync_startup => true}.merge(options))
  record_api_supportability_metric(:manual_start)
end

#notice_error(exception, options = {}) ⇒ Object

Send an error to New Relic.

Any option keys other than the ones listed here are treated as :custom_params.

Note: Previous versions of the agent allowed passing :request_params, but those are now ignored. If you need to record the request parameters, call this method inside a transaction or pass the information in :custom_params.

Most of the time, you do not need to specify the :uri or :metric options; only pass them if you are calling notice_error outside a transaction.

Parameters:

  • exception (Exception)

    Error you wish to send

  • options (Hash) (defaults to: {})

    Modify how New Relic processes the error

Options Hash (options):

  • :custom_params (Hash)

    Custom parameters to attach to the trace

  • :expected (Boolean)

    Only record the error trace (do not affect error rate or Apdex status)

  • :uri (String)

    Request path, minus request params or query string (usually not needed)

  • :metric (String)

    Metric name associated with the transaction (usually not needed)



403
404
405
406
407
408
# File 'lib/new_relic/agent.rb', line 403

def notice_error(exception, options = {})
  record_api_supportability_metric(:notice_error)

  Transaction.notice_error(exception, options)
  nil # don't return a noticed error data structure. it can only hurt.
end

#record_custom_event(event_type, event_attrs) ⇒ Object

Record a custom event to be sent to New Relic Insights. The recorded event will be buffered in memory until the next time the agent sends data to New Relic’s servers.

If you want to be able to tie the information recorded via this call back to the web request or background job that it happened in, you may want to instead use the add_custom_attributes API call to attach attributes to the Transaction event that will automatically be generated for the request.

A timestamp will be automatically added to the recorded event when this method is called.

Parameters:

  • event_type (Symbol or String)

    The name of the event type to record. Event types must consist of only alphanumeric characters, ‘_’, ‘:’, or ‘ ’.

  • event_attrs (Hash)

    A Hash of attributes to be attached to the event. Keys should be strings or symbols, and values may be strings, symbols, numeric values or booleans.



479
480
481
482
483
484
485
486
487
# File 'lib/new_relic/agent.rb', line 479

def record_custom_event(event_type, event_attrs)
  record_api_supportability_metric(:record_custom_event)

  if agent && NewRelic::Agent.config[:'custom_insights_events.enabled']
    agent.custom_event_aggregator.record(event_type, event_attrs)
  end

  nil
end

#record_llm_feedback_event(trace_id:, rating:, category: nil, message: nil, metadata: NewRelic::EMPTY_HASH) ⇒ Object

Records user feedback events for LLM applications. This API must pass the current trace id as a parameter, which can be obtained using:

NewRelic::Agent::Tracer.current_trace_id

Parameters:

  • ID (String)

    of the trace where the chat completion(s) related to the feedback occurred.

  • Rating (String or Integer)

    provided by an end user (ex: “Good“, ”Bad”, 1, 2, 5, 8, 10).

  • Category (optional, String)

    of the feedback as provided by the end user (ex: “informative”, “inaccurate”).

  • start_time (optional, String)

    Freeform text feedback from an end user.

  • Set (optional, Hash)

    of key-value pairs to store any other desired data to submit with the feedback event.



511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
# File 'lib/new_relic/agent.rb', line 511

def record_llm_feedback_event(trace_id:,
  rating:,
  category: nil,
  message: nil,
  metadata: NewRelic::EMPTY_HASH)

  record_api_supportability_metric(:record_llm_feedback_event)
  unless NewRelic::Agent.config[:'distributed_tracing.enabled']
    return NewRelic::Agent.logger.error('Distributed tracing must be enabled to record LLM feedback')
  end

  feedback_message_event = {
    'trace_id': trace_id,
    'rating': rating,
    'category': category,
    'message': message,
    'id': NewRelic::Agent::GuidGenerator.generate_guid,
    'ingest_source': NewRelic::Agent::Llm::LlmEvent::INGEST_SOURCE
  }
  feedback_message_event.merge!() unless .empty?

  NewRelic::Agent.record_custom_event(LLM_FEEDBACK_MESSAGE, feedback_message_event)
rescue ArgumentError
  raise
rescue => exception
  NewRelic::Agent.logger.error('record_llm_feedback_event', exception)
end

#record_metric(metric_name, value) ⇒ Object

Record a value for the given metric name.

This method should be used to record event-based metrics such as method calls that are associated with a specific duration or magnitude.

metric_name should follow a slash separated path convention. Application specific metrics should begin with “Custom/”.

value should be either a single Numeric value representing the duration/ magnitude of the event being recorded, or a Hash containing :count, :total, :min, :max, and :sum_of_squares keys. The latter form is useful for recording pre-aggregated metrics collected externally.

This method is safe to use from any thread.



297
298
299
300
301
302
303
304
305
306
307
308
# File 'lib/new_relic/agent.rb', line 297

def record_metric(metric_name, value) # THREAD_LOCAL_ACCESS
  record_api_supportability_metric(:record_metric)

  return unless agent

  if value.is_a?(Hash)
    stats = NewRelic::Agent::Stats.new
    value = stats.hash_merge(value)
  end

  agent.stats_engine.tl_record_unscoped_metrics(metric_name, value)
end

#require_test_helperObject

Require agent testing helper methods



673
674
675
676
# File 'lib/new_relic/agent.rb', line 673

def require_test_helper
  record_api_supportability_metric(:require_test_helper)
  require File.expand_path('../../../test/agent_helper', __FILE__)
end

#set_error_group_callback(callback_proc) ⇒ Object

Set a callback proc for determining an error’s error group name

Typically this method should be called only once to set a callback for use with all noticed errors. If it is called multiple times, each new callback given will replace the old one.

The proc will be called with a single hash as its input argument and is expected to return a string representing the desired error group.

see docs.newrelic.com/docs/errors-inbox/errors-inbox/#groups

The hash passed to the customer defined callback proc has the following keys:

:error => The Ruby error class instance, likely inheriting from

StandardError. Call `#class`, `#message`, and `#backtrace` on
the error object to retrieve the error's class, message, and
backtrace.

:customAttributes => Any customer defined custom attributes that have been

associated with the current transaction.

:‘request.uri’ => The current request URI if available :‘http.statusCode’ => The HTTP status code (200, 404, etc.) if available :‘http.method’ => The HTTP method (GET, PUT, etc.) if available :‘error.expected’ => Whether (true) or not (false) the error was expected :options => The options hash passed to ‘NewRelic::Agent.notice_error`

Parameters:

  • callback_proc (Proc)

    the callback proc



440
441
442
443
444
445
446
447
448
449
# File 'lib/new_relic/agent.rb', line 440

def set_error_group_callback(callback_proc)
  unless callback_proc.is_a?(Proc)
    NewRelic::Agent.logger.error("#{self}.#{__method__}: expected an argument of type Proc, " \
                                 "got #{callback_proc.class}")
    return
  end

  record_api_supportability_metric(:set_error_group_callback)
  @error_group_callback = callback_proc
end

#set_llm_token_count_callback(callback_proc) ⇒ Object

Set a callback proc for calculating ‘token_count` attributes for LlmEmbedding and LlmChatCompletionMessage events

This method should be called only once to set a callback for use with all LLM token calculations. If it is called multiple times, each new callback will replace the old one.

The proc will be called with a single hash as its input argument and must return an Integer representing the number of tokens used for that particular prompt, completion message, or embedding. Values less than or equal to 0 will not be attached to an event.

The hash has the following keys:

:model => [String] The name of the LLM model :content => [String] The message content or prompt

Parameters:

  • callback_proc (Proc)

    the callback proc



564
565
566
567
568
569
570
571
572
573
# File 'lib/new_relic/agent.rb', line 564

def set_llm_token_count_callback(callback_proc)
  unless callback_proc.is_a?(Proc)
    NewRelic::Agent.logger.error("#{self}.#{__method__}: expected an argument of type Proc, " \
                                 "got #{callback_proc.class}")
    return
  end

  record_api_supportability_metric(:set_llm_token_count_callback)
  @llm_token_count_callback = callback_proc
end

#set_sql_obfuscator(type = :replace, &block) ⇒ Object

This method sets the block sent to this method as a sql obfuscator. The block will be called with a single String SQL statement to obfuscate. The method must return the obfuscated String SQL. If chaining of obfuscators is required, use type = :before or :after

type = :before, :replace, :after

Example:

NewRelic::Agent.set_sql_obfuscator(:replace) do |sql|
   my_obfuscator(sql)
end


694
695
696
697
# File 'lib/new_relic/agent.rb', line 694

def set_sql_obfuscator(type = :replace, &block)
  record_api_supportability_metric(:set_sql_obfuscator)
  NewRelic::Agent::Database.set_sql_obfuscator(type, &block)
end

#set_transaction_name(name, options = {}) ⇒ Object

Set the name of the current running transaction. The agent will apply a reasonable default based on framework routing, but in cases where this is insufficient, this can be used to manually control the name of the transaction.

The category of transaction can be specified via the :category option. The following are the only valid categories:

  • :category => :controller indicates that this is a controller action and will appear with all the other actions.

  • :category => :task indicates that this is a background task and will show up in New Relic with other background tasks instead of in the controllers list

  • :category => :middleware if you are instrumenting a rack middleware call. The :name is optional, useful if you have more than one potential transaction in the #call.

  • :category => :uri indicates that this is a web transaction whose name is a normalized URI, where ‘normalized’ means the URI does not have any elements with data in them such as in many REST URIs.

The default category is the same as the running transaction.



942
943
944
945
# File 'lib/new_relic/agent.rb', line 942

def set_transaction_name(name, options = {})
  record_api_supportability_metric(:set_transaction_name)
  Transaction.set_overriding_transaction_name(name, options[:category])
end

#set_user_id(user_id) ⇒ Object

Set the user id for the current transaction. When present, this value will be included in the agent attributes for transaction and error events as ‘enduser.id’.

Parameters:

  • user_id (String)

    The user id to add to the current transaction attributes



898
899
900
901
902
903
904
905
906
907
908
909
910
911
# File 'lib/new_relic/agent.rb', line 898

def set_user_id(user_id)
  record_api_supportability_metric(:set_user_id)

  if user_id.nil? || user_id.empty?
    ::NewRelic::Agent.logger.warn('NewRelic::Agent.set_user_id called with a nil or empty user id.')
    return
  end

  default_destinations = NewRelic::Agent::AttributeFilter::DST_TRANSACTION_TRACER |
    NewRelic::Agent::AttributeFilter::DST_TRANSACTION_EVENTS |
    NewRelic::Agent::AttributeFilter::DST_ERROR_COLLECTOR

  NewRelic::Agent::Transaction.add_agent_attribute(:'enduser.id', user_id, default_destinations)
end

#shutdown(options = {}) ⇒ Object

Shutdown the agent. Call this before exiting. Sends any queued data and kills the background thread.

Parameters:

  • options (Hash) (defaults to: {})

    Unused options Hash, for back compatibility only



641
642
643
644
# File 'lib/new_relic/agent.rb', line 641

def shutdown(options = {})
  record_api_supportability_metric(:shutdown)
  agent&.shutdown
end