debian-mirror-gitlab/lib/gitlab/instrumentation/redis_interceptor.rb

108 lines
3.5 KiB
Ruby
Raw Normal View History

2020-06-23 00:09:42 +05:30
# frozen_string_literal: true
module Gitlab
module Instrumentation
module RedisInterceptor
2020-07-28 23:09:34 +05:30
APDEX_EXCLUDE = %w[brpop blpop brpoplpush bzpopmin bzpopmax xread xreadgroup].freeze
2021-10-27 15:23:28 +05:30
class MysteryRedisDurationError < StandardError
attr_reader :backtrace
def initialize(backtrace)
@backtrace = backtrace
end
end
2020-06-23 00:09:42 +05:30
def call(*args, &block)
2021-09-30 23:02:18 +05:30
start = Gitlab::Metrics::System.monotonic_time # must come first so that 'start' is always defined
2020-07-28 23:09:34 +05:30
instrumentation_class.instance_count_request
instrumentation_class.redis_cluster_validate!(args.first)
2020-06-23 00:09:42 +05:30
super(*args, &block)
2020-07-28 23:09:34 +05:30
rescue ::Redis::BaseError => ex
instrumentation_class.instance_count_exception(ex)
raise ex
2020-06-23 00:09:42 +05:30
ensure
2021-09-30 23:02:18 +05:30
duration = Gitlab::Metrics::System.monotonic_time - start
2020-07-28 23:09:34 +05:30
unless APDEX_EXCLUDE.include?(command_from_args(args))
instrumentation_class.instance_observe_duration(duration)
end
2020-06-23 00:09:42 +05:30
if ::RequestStore.active?
2020-07-28 23:09:34 +05:30
# These metrics measure total Redis usage per Rails request / job.
2020-06-23 00:09:42 +05:30
instrumentation_class.increment_request_count
instrumentation_class.add_duration(duration)
instrumentation_class.add_call_details(duration, args)
end
end
def write(command)
measure_write_size(command) if ::RequestStore.active?
super
end
def read
result = super
measure_read_size(result) if ::RequestStore.active?
result
end
private
def measure_write_size(command)
size = 0
# Mimic what happens in
# https://github.com/redis/redis-rb/blob/f597f21a6b954b685cf939febbc638f6c803e3a7/lib/redis/connection/command_helper.rb#L8.
# This count is an approximation that omits the Redis protocol overhead
# of type prefixes, length prefixes and line endings.
command.each do |x|
size += begin
if x.is_a? Array
x.inject(0) { |sum, y| sum + y.to_s.bytesize }
else
x.to_s.bytesize
end
end
end
instrumentation_class.increment_write_bytes(size)
end
def measure_read_size(result)
# The Connection::Ruby#read class can return one of four types of results from read:
# https://github.com/redis/redis-rb/blob/f597f21a6b954b685cf939febbc638f6c803e3a7/lib/redis/connection/ruby.rb#L406
#
# 1. Error (exception, will not reach this line)
# 2. Status (string)
# 3. Integer (will be converted to string by to_s.bytesize and thrown away)
# 4. "Binary" string (i.e. may contain zero byte)
# 5. Array of binary string
if result.is_a? Array
# Redis can return nested arrays, e.g. from XRANGE or GEOPOS, so we use recursion here.
result.each { |x| measure_read_size(x) }
else
# This count is an approximation that omits the Redis protocol overhead
# of type prefixes, length prefixes and line endings.
instrumentation_class.increment_read_bytes(result.to_s.bytesize)
end
end
# That's required so it knows which GitLab Redis instance
# it's interacting with in order to categorize accordingly.
#
def instrumentation_class
@options[:instrumentation_class] # rubocop:disable Gitlab/ModuleWithInstanceVariables
end
2020-07-28 23:09:34 +05:30
def command_from_args(args)
command = args[0]
command = command[0] if command.is_a?(Array)
command.to_s.downcase
end
2020-06-23 00:09:42 +05:30
end
end
end