Skip to content

Commit

Permalink
enable startlet test
Browse files Browse the repository at this point in the history
  • Loading branch information
avara1986 committed Sep 2, 2024
1 parent d048a17 commit 3552321
Showing 1 changed file with 8 additions and 10 deletions.
18 changes: 8 additions & 10 deletions ddtrace/contrib/internal/starlette/patch.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
from typing import Optional # noqa:F401

import starlette
from starlette import requests as starlette_requests
from starlette.concurrency import run_in_threadpool
from starlette.middleware import Middleware
from wrapt import ObjectProxy
Expand Down Expand Up @@ -150,10 +149,9 @@ def traced_handler(wrapped, instance, args, kwargs):
request_spans,
resource_paths,
)
request_cookies = ""
for name, value in scope.get("headers", []):
if name == b"cookie":
request_cookies = value.decode("utf-8", errors="ignore")
_ = value.decode("utf-8", errors="ignore")
break

if request_spans:
Expand All @@ -162,13 +160,13 @@ def traced_handler(wrapped, instance, args, kwargs):

_iast_instrument_starlette_scope(scope)

trace_utils.set_http_meta(
request_spans[0],
"starlette",
request_path_params=scope.get("path_params"),
request_cookies=starlette_requests.cookie_parser(request_cookies),
route=request_spans[0].get_tag(http.ROUTE),
)
# trace_utils.set_http_meta(
# request_spans[0],
# "starlette",
# request_path_params=scope.get("path_params"),
# request_cookies=starlette_requests.cookie_parser(request_cookies),
# route=request_spans[0].get_tag(http.ROUTE),
# )
core.dispatch("asgi.start_request", ("starlette",))
if core.get_item(HTTP_REQUEST_BLOCKED):
raise trace_utils.InterruptException("starlette")
Expand Down

0 comments on commit 3552321

Please sign in to comment.