module NewRelic::Agent::Instrumentation::DynamoDB

Constants

DEFAULT_HOST
INSTRUMENTED_METHODS
PRODUCT

Public Instance Methods

build_request_with_new_relic(*args) { || ... } click to toggle source
# File lib/new_relic/agent/instrumentation/dynamodb/instrumentation.rb, line 48
def build_request_with_new_relic(*args)
  @nr_captured_request = yield
end
get_arn(params) click to toggle source
# File lib/new_relic/agent/instrumentation/dynamodb/instrumentation.rb, line 52
def get_arn(params)
  return unless params[:table_name]

  NewRelic::Agent::Aws.create_arn(PRODUCT.downcase, "table/#{params[:table_name]}", config&.region)
end
instrument_method_with_new_relic(method_name, *args) { || ... } click to toggle source
# 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