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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
module Sentry
module Rails
module ActionCableExtensions
class ErrorHandler
OP_NAME = "websocket.server".freeze
SPAN_ORIGIN = "auto.http.rails.actioncable"
class << self
def capture(connection, transaction_name:, extra_context: nil, &block)
return block.call unless Sentry.initialized?
# ActionCable's ConnectionStub (for testing) doesn't implement the exact same interfaces as Connection::Base.
# One thing that's missing is `env`. So calling `connection.env` direclty will fail in test environments when `stub_connection` is used.
# See https://github.com/getsentry/sentry-ruby/pull/1684 for more information.
env = connection.respond_to?(:env) ? connection.env : {}
Sentry.with_scope do |scope|
scope.set_rack_env(env)
scope.set_context("action_cable", extra_context) if extra_context
scope.set_transaction_name(transaction_name, source: :view)
transaction = start_transaction(env, scope)
scope.set_span(transaction) if transaction
begin
result = block.call
finish_transaction(transaction, 200)
result
rescue Exception => e # rubocop:disable Lint/RescueException
Sentry::Rails.capture_exception(e)
finish_transaction(transaction, 500)
raise
end
end
end
def start_transaction(env, scope)
options = {
name: scope.transaction_name,
source: scope.transaction_source,
op: OP_NAME,
origin: SPAN_ORIGIN
}
transaction = Sentry.continue_trace(env, **options)
Sentry.start_transaction(transaction: transaction, **options)
end
def finish_transaction(transaction, status_code)
return unless transaction
transaction.set_http_status(status_code)
transaction.finish
end
end
end
module Connection
private
def handle_open
ErrorHandler.capture(self, transaction_name: "#{self.class.name}#connect") do
super
end
end
def handle_close
ErrorHandler.capture(self, transaction_name: "#{self.class.name}#disconnect") do
super
end
end
end
module Channel
module Subscriptions
def self.included(base)
base.class_eval do
set_callback :subscribe, :around, ->(_, block) { sentry_capture(:subscribed, &block) }, prepend: true
set_callback :unsubscribe, :around, ->(_, block) { sentry_capture(:unsubscribed, &block) }, prepend: true
end
end
private
def sentry_capture(hook, &block)
extra_context = { params: params }
ErrorHandler.capture(connection, transaction_name: "#{self.class.name}##{hook}", extra_context: extra_context, &block)
end
end
module Actions
private
def dispatch_action(action, data)
extra_context = { params: params, data: data }
ErrorHandler.capture(connection, transaction_name: "#{self.class.name}##{action}", extra_context: extra_context) do
super
end
end
end
end
end
end
end
|
