htwkmooc/codeocean
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Correctly set current Sentry span for WebSocket requests

Browse Source
This commit is contained in:
Sebastian Serth
2023-11-30 22:06:25 +01:00
committed by Sebastian Serth
parent fc1b5bfa6b
commit 05adde58ce
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lib/runner/connection.rb
View File

@ -22,9 +22,13 @@ class Runner::Connection
def initialize(url, strategy, event_loop, locale = I18n.locale)
Rails.logger.debug { "#{Time.zone.now.getutc.inspect}: Opening connection to #{url}" }
sentry_transaction = Sentry.get_current_scope&.get_span
sentry_span = sentry_transaction&.start_child(op: SENTRY_OP_NAME, start_timestamp: Sentry.utc_now.to_f)
sentry_parent_transaction = Sentry.get_current_scope&.get_span
sentry_span = sentry_parent_transaction&.start_child(op: SENTRY_OP_NAME, start_timestamp: Sentry.utc_now.to_f)
# Set the span as the current span in the scope for correct trace headers.
Sentry.get_current_scope&.set_span(sentry_span) if sentry_span
http_headers = strategy.class.websocket_header.deep_merge sentry_trace_header(sentry_span)
Sentry.get_current_scope&.set_span(sentry_parent_transaction) if sentry_parent_transaction
@socket = Faye::WebSocket::Client.new(url, [], http_headers)
@strategy = strategy