Contents

module Cacheflow
  module Redis
    # redis 5 / redis-client
    module ClientNotifications
      def call(command, redis_config)
        payload = {
          commands: [command]
        }
        ActiveSupport::Notifications.instrument("query.redis", payload) do
          super
        end
      end

      def call_pipelined(commands, redis_config)
        payload = {
          commands: commands
        }
        ActiveSupport::Notifications.instrument("query.redis", payload) do
          super
        end
      end
    end

    # redis 4
    module Notifications
      def logging(commands)
        payload = {
          commands: commands
        }
        ActiveSupport::Notifications.instrument("query.redis", payload) do
          super
        end
      end
    end

    class Instrumenter < ActiveSupport::LogSubscriber
      def query(event)
        return if !logger.debug? || Cacheflow.silenced?

        name = "%s (%.2fms)" % ["Redis", event.duration]

        commands = []
        event.payload[:commands].map do |op, *args|
          commands << "#{op.to_s.upcase} #{args.join(" ")}".strip
        end

        debug "  #{color(name, RED, bold: true)} #{commands.join(" >> ")}"
      end
    end
  end
end

if defined?(RedisClient)
  RedisClient.register(Cacheflow::Redis::ClientNotifications)
elsif defined?(Redis)
  # redis < 5
  Redis::Client.prepend(Cacheflow::Redis::Notifications)
end

Cacheflow::Redis::Instrumenter.attach_to(:redis)

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
cacheflow-0.3.2	lib/cacheflow/redis.rb