Converted TextMap propagator getter to a class and added keys method (#1196)

Co-authored-by: alrex <aboten@lightstep.com>
This commit is contained in:
Prajilesh N 2020-11-02 09:42:47 +05:30 committed by GitHub
parent 5f932d0aac
commit 3ce216ede9
1 changed files with 4 additions and 10 deletions

View File

@ -54,6 +54,7 @@ from opentelemetry.instrumentation.utils import (
http_status_to_status_code,
unwrap,
)
from opentelemetry.trace.propagation.textmap import DictGetter
from opentelemetry.trace.status import Status
from opentelemetry.util import ExcludeList, time_ns
@ -84,6 +85,8 @@ def get_traced_request_attrs():
_excluded_urls = get_excluded_urls()
_traced_attrs = get_traced_request_attrs()
carrier_getter = DictGetter()
class TornadoInstrumentor(BaseInstrumentor):
patched_handlers = []
@ -185,13 +188,6 @@ def _log_exception(tracer, func, handler, args, kwargs):
return func(*args, **kwargs)
def _get_header_from_request_headers(
headers: dict, header_name: str
) -> typing.List[str]:
header = headers.get(header_name)
return [header] if header else []
def _get_attributes_from_request(request):
attrs = {
"component": "tornado",
@ -218,9 +214,7 @@ def _get_operation_name(handler, request):
def _start_span(tracer, handler, start_time) -> _TraceContext:
token = context.attach(
propagators.extract(
_get_header_from_request_headers, handler.request.headers,
)
propagators.extract(carrier_getter, handler.request.headers,)
)
span = tracer.start_span(