1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
# 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, *arguments = statement
{ command: command.to_s.upcase, key: key }.tap do |command_set|
command_set[:arguments] = arguments.join(" ") if Sentry.configuration.send_default_pii
end
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
|
