Contents

# frozen_string_literal: true

require 'elastic_apm/sql'

module ElasticAPM
  module Normalizers
    module ActiveRecord
      # @api private
      class SqlNormalizer < Normalizer
        register 'sql.active_record'

        TYPE = 'db'
        ACTION = 'sql'
        SKIP_NAMES = %w[SCHEMA CACHE].freeze
        UNKNOWN = 'unknown'

        def initialize(*args)
          super

          @summarizer = Sql.summarizer

          @adapters = {}
        end

        def normalize(_transaction, _name, payload)
          return :skip if SKIP_NAMES.include?(payload[:name])

          name = summarize(payload[:sql]) || payload[:name]
          subtype = subtype_for(payload)

          context =
            Span::Context.new(
              db: { statement: payload[:sql], type: 'sql' },
              destination: { name: subtype, resource: subtype, type: TYPE }
            )

          [name, TYPE, subtype, ACTION, context]
        end

        private

        def subtype_for(payload)
          cached_adapter_name(
            payload[:connection]&.adapter_name ||
              ::ActiveRecord::Base.connection_config[:adapter]
          )
        end

        def summarize(sql)
          @summarizer.summarize(sql)
        end

        def cached_adapter_name(adapter_name)
          return UNKNOWN if adapter_name.nil? || adapter_name.empty?
          @adapters[adapter_name] ||
            (@adapters[adapter_name] = adapter_name.downcase)
        rescue StandardError
          nil
        end
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
elastic-apm-3.6.0	lib/elastic_apm/normalizers/rails/active_record.rb
elastic-apm-3.5.0	lib/elastic_apm/normalizers/rails/active_record.rb
elastic-apm-3.4.0	lib/elastic_apm/normalizers/rails/active_record.rb