Contents

# frozen_string_literal: true

module Sentry
  # @api private
  class Redis
    OP_NAME ||= "db.redis.command"
    LOGGER_NAME ||= :redis_logger

    def initialize(commands, host, port, db)
      @commands, @host, @port, @db = commands, host, port, db
    end

    def instrument
      return yield unless Sentry.initialized?

      record_span do
        yield.tap do
          record_breadcrumb
        end
      end
    end

    private

    attr_reader :commands, :host, :port, :db

    def record_span
      return yield unless (transaction = Sentry.get_current_scope.get_transaction) && transaction.sampled

      sentry_span = transaction.start_child(op: OP_NAME, start_timestamp: Sentry.utc_now.to_f)

      yield.tap do
        sentry_span.set_description(commands_description)
        sentry_span.set_data(:server, server_description)
        sentry_span.set_timestamp(Sentry.utc_now.to_f)
      end
    end

    def record_breadcrumb
      return unless Sentry.configuration.breadcrumbs_logger.include?(LOGGER_NAME)

      Sentry.add_breadcrumb(
        Sentry::Breadcrumb.new(
          level: :info,
          category: OP_NAME,
          type: :info,
          data: {
            commands: parsed_commands,
            server: server_description
          }
        )
      )
    end

    def commands_description
      parsed_commands.map do |statement|
        statement.values.join(" ").strip
      end.join(", ")
    end

    def parsed_commands
      commands.map do |statement|
        command, key, *_values = statement

        { command: command.to_s.upcase, key: key }
      end
    end

    def server_description
      "#{host}:#{port}/#{db}"
    end

    module Client
      def logging(commands, &block)
        Sentry::Redis.new(commands, host, port, db).instrument do
          super
        end
      end
    end
  end
end

if defined?(::Redis::Client)
  Sentry.register_patch do
    patch = Sentry::Redis::Client
    Redis::Client.prepend(patch) unless Redis::Client.ancestors.include?(patch)
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
sentry-ruby-core-5.1.1	lib/sentry/redis.rb
sentry-ruby-core-5.1.0	lib/sentry/redis.rb