Method: NewRelic::Agent::Instrumentation::DynamoDB#instrument_method_with_new_relic

Defined in:
lib/new_relic/agent/instrumentation/dynamodb/instrumentation.rb

#instrument_method_with_new_relic(method_name, *args) ⇒ Object

[View source]

21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
 
# File 'lib/new_relic/agent/instrumentation/dynamodb/instrumentation.rb', line 21

def instrument_method_with_new_relic(method_name, *args)
  return yield unless NewRelic::Agent::Tracer.tracing_enabled?

  NewRelic::Agent.record_instrumentation_invocation(PRODUCT)

  segment = NewRelic::Agent::Tracer.start_datastore_segment(
    product: PRODUCT,
    operation: method_name,
    host: config&.endpoint&.host || DEFAULT_HOST,
    port_path_or_id: config&.endpoint&.port,
    collection: args[0][:table_name]
  )

  arn = get_arn(args[0])
  segment&.add_agent_attribute('cloud.resource_id', arn) if arn

  @nr_captured_request = nil # clear request just in case
  begin
    NewRelic::Agent::Tracer.capture_segment_error(segment) { yield }
  ensure
    segment&.add_agent_attribute('aws.operation', method_name)
    segment&.add_agent_attribute('aws.requestId', @nr_captured_request&.context&.http_response&.headers&.[]('x-amzn-requestid'))
    segment&.add_agent_attribute('aws.region', config&.region)
    segment&.finish
  end
end