44from sentry_sdk .consts import OP , SPANDATA
55from sentry_sdk .integrations import DidNotEnable , Integration
66from sentry_sdk .scope import should_send_default_pii
7- from sentry_sdk .tracing import BAGGAGE_HEADER_NAME
87from sentry_sdk .tracing_utils import (
98 add_http_request_source ,
10- add_sentry_baggage_to_headers ,
119 has_span_streaming_enabled ,
12- should_propagate_trace ,
10+ propagate_trace_headers ,
1311)
1412from sentry_sdk .utils import (
1513 SENSITIVE_DATA_SUBSTITUTE ,
1614 capture_internal_exceptions ,
1715 ensure_integration_enabled ,
18- logger ,
1916 parse_url ,
2017)
2118
@@ -61,6 +58,8 @@ def send(self: "Client", request: "Request", **kwargs: "Any") -> "Response":
6158
6259 if is_span_streaming_enabled :
6360 if sentry_sdk .traces .get_current_span () is None :
61+ propagate_trace_headers (client , request )
62+
6463 return real_send (self , request , ** kwargs )
6564
6665 with sentry_sdk .traces .start_span (
@@ -84,21 +83,7 @@ def send(self: "Client", request: "Request", **kwargs: "Any") -> "Response":
8483 if parsed_url .fragment :
8584 attributes ["url.fragment" ] = parsed_url .fragment
8685
87- if should_propagate_trace (client , str (request .url )):
88- for (
89- key ,
90- value ,
91- ) in (
92- sentry_sdk .get_current_scope ().iter_trace_propagation_headers ()
93- ):
94- logger .debug (
95- f"[Tracing] Adding `{ key } ` header { value } to outgoing request to { request .url } ."
96- )
97-
98- if key == BAGGAGE_HEADER_NAME :
99- add_sentry_baggage_to_headers (request .headers , value )
100- else :
101- request .headers [key ] = value
86+ propagate_trace_headers (client , request )
10287
10388 try :
10489 rv = real_send (self , request , ** kwargs )
@@ -128,21 +113,7 @@ def send(self: "Client", request: "Request", **kwargs: "Any") -> "Response":
128113 span .set_data (SPANDATA .HTTP_QUERY , parsed_url .query )
129114 span .set_data (SPANDATA .HTTP_FRAGMENT , parsed_url .fragment )
130115
131- if should_propagate_trace (client , str (request .url )):
132- for (
133- key ,
134- value ,
135- ) in (
136- sentry_sdk .get_current_scope ().iter_trace_propagation_headers ()
137- ):
138- logger .debug (
139- f"[Tracing] Adding `{ key } ` header { value } to outgoing request to { request .url } ."
140- )
141-
142- if key == BAGGAGE_HEADER_NAME :
143- add_sentry_baggage_to_headers (request .headers , value )
144- else :
145- request .headers [key ] = value
116+ propagate_trace_headers (client , request )
146117
147118 rv = real_send (self , request , ** kwargs )
148119
@@ -174,6 +145,8 @@ async def send(
174145
175146 if is_span_streaming_enabled :
176147 if sentry_sdk .traces .get_current_span () is None :
148+ propagate_trace_headers (client , request )
149+
177150 return await real_send (self , request , ** kwargs )
178151
179152 with sentry_sdk .traces .start_span (
@@ -197,21 +170,7 @@ async def send(
197170 if parsed_url .fragment :
198171 attributes ["url.fragment" ] = parsed_url .fragment
199172
200- if should_propagate_trace (client , str (request .url )):
201- for (
202- key ,
203- value ,
204- ) in (
205- sentry_sdk .get_current_scope ().iter_trace_propagation_headers ()
206- ):
207- logger .debug (
208- f"[Tracing] Adding `{ key } ` header { value } to outgoing request to { request .url } ."
209- )
210-
211- if key == BAGGAGE_HEADER_NAME :
212- add_sentry_baggage_to_headers (request .headers , value )
213- else :
214- request .headers [key ] = value
173+ propagate_trace_headers (client , request )
215174
216175 try :
217176 rv = await real_send (self , request , ** kwargs )
@@ -241,20 +200,7 @@ async def send(
241200 span .set_data (SPANDATA .HTTP_QUERY , parsed_url .query )
242201 span .set_data (SPANDATA .HTTP_FRAGMENT , parsed_url .fragment )
243202
244- if should_propagate_trace (client , str (request .url )):
245- for (
246- key ,
247- value ,
248- ) in (
249- sentry_sdk .get_current_scope ().iter_trace_propagation_headers ()
250- ):
251- logger .debug (
252- f"[Tracing] Adding `{ key } ` header { value } to outgoing request to { request .url } ."
253- )
254- if key == BAGGAGE_HEADER_NAME :
255- add_sentry_baggage_to_headers (request .headers , value )
256- else :
257- request .headers [key ] = value
203+ propagate_trace_headers (client , request )
258204
259205 rv = await real_send (self , request , ** kwargs )
260206
0 commit comments