module Datadog::Tracing::Contrib::ActiveRecord::Events::Instantiation

def event_name
  self::EVENT_NAME
end

def process(span, event, _id, payload)
  span.resource = payload.fetch(:class_name)
  span.span_type = Ext::SPAN_TYPE_INSTANTIATION
  span.set_tag(Tracing::Metadata::Ext::TAG_COMPONENT, Ext::TAG_COMPONENT)
  span.set_tag(Tracing::Metadata::Ext::TAG_OPERATION, Ext::TAG_OPERATION_INSTANTIATION)
  # Set analytics sample rate
  if Contrib::Analytics.enabled?(configuration[:analytics_enabled])
    Contrib::Analytics.set_sample_rate(span, configuration[:analytics_sample_rate])
  end
  # Measure service stats
  Contrib::Analytics.set_measured(span)
  span.set_tag(Ext::TAG_INSTANTIATION_CLASS_NAME, payload.fetch(:class_name))
  span.set_tag(Ext::TAG_INSTANTIATION_RECORD_COUNT, payload.fetch(:record_count))
rescue StandardError => e
  Datadog.logger.debug(e.message)
end

def span_name
  Ext::SPAN_INSTANTIATION
end

def supported?
  Gem.loaded_specs['activerecord'] \
    && Gem.loaded_specs['activerecord'].version >= Gem::Version.new('4.2')
end