class ElasticAPM::Injectors::RedisInjector

def call(command, &block)
  name = command[0].upcase
  statement =
    format('%s %s', name, command[1..command.length].join(' '))
  context = Span::Context.new(
    statement: statement,
    type: 'redis'
  )
  ElasticAPM.span(name.to_s, 'db.redis', context: context) do
    call_without_apm(command, &block)
  end
end

def install
  ::Redis::Client.class_eval do
    alias call_without_apm call
    def call(command, &block)
      name = command[0].upcase
      statement =
        format('%s %s', name, command[1..command.length].join(' '))
      context = Span::Context.new(
        statement: statement,
        type: 'redis'
      )
      ElasticAPM.span(name.to_s, 'db.redis', context: context) do
        call_without_apm(command, &block)
      end
    end
  end
end