38 lines
956 B
Ruby
38 lines
956 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab
|
||
|
|
module Instrumentation
|
||
|
|
module RedisPayload
|
||
|
|
include ::Gitlab::Utils::StrongMemoize
|
||
|
|
|
||
|
|
# Fetches payload keys from the lazy payload (this avoids
|
||
|
|
# unnecessary processing of the values).
|
||
|
|
def known_payload_keys
|
||
|
|
to_lazy_payload.keys
|
||
|
|
end
|
||
|
|
|
||
|
|
def payload
|
||
|
|
to_lazy_payload.transform_values do |value|
|
||
|
|
result = value.call
|
||
|
|
result if result > 0
|
||
|
|
end.compact
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def to_lazy_payload
|
||
|
|
strong_memoize(:to_lazy_payload) do
|
||
|
|
key_prefix = storage_key ? "redis_#{storage_key}" : 'redis'
|
||
|
|
|
||
|
|
{
|
||
|
|
"#{key_prefix}_calls": -> { get_request_count },
|
||
|
|
"#{key_prefix}_duration_s": -> { query_time },
|
||
|
|
"#{key_prefix}_read_bytes": -> { read_bytes },
|
||
|
|
"#{key_prefix}_write_bytes": -> { write_bytes }
|
||
|
|
}.symbolize_keys
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|