Skip to content

Commit 54eb5c2

Browse files
committed
Remove setters and getters
Fixes open-telemetry#1644
1 parent a46b4c2 commit 54eb5c2

File tree

24 files changed

+587
-842
lines changed

24 files changed

+587
-842
lines changed

docs/conf.py

-1
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,6 @@
104104
("py:class", "opentelemetry.trace._LinkBase",),
105105
# TODO: Understand why sphinx is not able to find this local class
106106
("py:class", "opentelemetry.propagators.textmap.TextMapPropagator",),
107-
("py:class", "opentelemetry.propagators.textmap.DictGetter",),
108107
("any", "opentelemetry.propagators.textmap.TextMapPropagator.extract",),
109108
("any", "opentelemetry.propagators.textmap.TextMapPropagator.inject",),
110109
]

docs/examples/auto-instrumentation/README.rst

+1-1
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ Manually instrumented server
3737
def server_request():
3838
with tracer.start_as_current_span(
3939
"server_request",
40-
context=propagators.extract(DictGetter(), request.headers
40+
context=propagators.extract(request.headers
4141
),
4242
):
4343
print(request.args.get("param"))

docs/examples/auto-instrumentation/client.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737

3838
with tracer.start_as_current_span("client-server"):
3939
headers = {}
40-
propagators.inject(dict.__setitem__, headers)
40+
propagators.inject(headers)
4141
requested = get(
4242
"http://localhost:8082/server_request",
4343
params={"param": argv[1]},

docs/examples/auto-instrumentation/server_instrumented.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717
from opentelemetry import trace
1818
from opentelemetry.instrumentation.wsgi import collect_request_attributes
1919
from opentelemetry.propagate import extract
20-
from opentelemetry.propagators.textmap import DictGetter
2120
from opentelemetry.sdk.trace import TracerProvider
2221
from opentelemetry.sdk.trace.export import (
2322
ConsoleSpanExporter,
@@ -38,7 +37,7 @@
3837
def server_request():
3938
with tracer.start_as_current_span(
4039
"server_request",
41-
context=extract(DictGetter(), request.headers),
40+
context=extract(request.headers),
4241
kind=trace.SpanKind.SERVER,
4342
attributes=collect_request_attributes(request.environ),
4443
):

docs/examples/datadog_exporter/client.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@
4747

4848
with tracer.start_as_current_span("client-server"):
4949
headers = {}
50-
inject(dict.__setitem__, headers)
50+
inject(headers)
5151
requested = get(
5252
"http://localhost:8082/server_request",
5353
params={"param": argv[1]},

docs/examples/django/client.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636

3737
with tracer.start_as_current_span("client-server"):
3838
headers = {}
39-
inject(dict.__setitem__, headers)
39+
inject(headers)
4040
requested = get(
4141
"http://localhost:8000",
4242
params={"param": argv[1]},

opentelemetry-api/src/opentelemetry/baggage/propagation/__init__.py

+38-54
Original file line numberDiff line numberDiff line change
@@ -11,29 +11,29 @@
1111
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
14-
#
15-
import typing
16-
import urllib.parse
14+
15+
from typing import Optional, Set
16+
from urllib.parse import quote_plus, unquote
1717

1818
from opentelemetry import baggage
1919
from opentelemetry.context import get_current
2020
from opentelemetry.context.context import Context
21-
from opentelemetry.propagators import textmap
21+
from opentelemetry.propagators.textmap import (
22+
TextMapPropagator,
23+
TextMapPropagatorT,
24+
)
2225

2326

24-
class W3CBaggagePropagator(textmap.TextMapPropagator):
27+
class W3CBaggagePropagator(TextMapPropagator):
2528
"""Extracts and injects Baggage which is used to annotate telemetry."""
2629

27-
_MAX_HEADER_LENGTH = 8192
28-
_MAX_PAIR_LENGTH = 4096
29-
_MAX_PAIRS = 180
30-
_BAGGAGE_HEADER_NAME = "baggage"
30+
_baggage_header_name = "baggage"
31+
_max_header_length = 9182
32+
_max_pairs = 180
33+
_max_pair_length = 4096
3134

3235
def extract(
33-
self,
34-
getter: textmap.Getter[textmap.TextMapPropagatorT],
35-
carrier: textmap.TextMapPropagatorT,
36-
context: typing.Optional[Context] = None,
36+
self, carrier: TextMapPropagatorT, context: Optional[Context] = None,
3737
) -> Context:
3838
"""Extract Baggage from the carrier.
3939
@@ -44,67 +44,51 @@ def extract(
4444
if context is None:
4545
context = get_current()
4646

47-
header = _extract_first_element(
48-
getter.get(carrier, self._BAGGAGE_HEADER_NAME)
49-
)
47+
value = carrier.get(self._baggage_header_name)
48+
49+
if value is None:
50+
header = None
51+
else:
52+
header = next(iter(value), None)
5053

51-
if not header or len(header) > self._MAX_HEADER_LENGTH:
54+
if header is None or len(header) > self._max_header_length:
5255
return context
5356

54-
baggage_entries = header.split(",")
55-
total_baggage_entries = self._MAX_PAIRS
56-
for entry in baggage_entries:
57+
total_baggage_entries = self._max_pairs
58+
59+
for entry in header.split(","):
5760
if total_baggage_entries <= 0:
5861
return context
5962
total_baggage_entries -= 1
60-
if len(entry) > self._MAX_PAIR_LENGTH:
63+
if len(entry) > self._max_pair_length:
6164
continue
62-
try:
65+
if "=" in entry:
6366
name, value = entry.split("=", 1)
64-
except Exception: # pylint: disable=broad-except
65-
continue
66-
context = baggage.set_baggage(
67-
urllib.parse.unquote(name).strip(),
68-
urllib.parse.unquote(value).strip(),
69-
context=context,
70-
)
67+
context = baggage.set_baggage(
68+
unquote(name).strip(),
69+
unquote(value).strip(),
70+
context=context,
71+
)
7172

7273
return context
7374

7475
def inject(
75-
self,
76-
set_in_carrier: textmap.Setter[textmap.TextMapPropagatorT],
77-
carrier: textmap.TextMapPropagatorT,
78-
context: typing.Optional[Context] = None,
76+
self, carrier: TextMapPropagatorT, context: Optional[Context] = None,
7977
) -> None:
8078
"""Injects Baggage into the carrier.
8179
8280
See
8381
`opentelemetry.propagators.textmap.TextMapPropagator.inject`
8482
"""
8583
baggage_entries = baggage.get_all(context=context)
86-
if not baggage_entries:
87-
return
8884

89-
baggage_string = _format_baggage(baggage_entries)
90-
set_in_carrier(carrier, self._BAGGAGE_HEADER_NAME, baggage_string)
85+
if baggage_entries:
86+
carrier[self._baggage_header_name] = ",".join(
87+
key + "=" + quote_plus(str(value))
88+
for key, value in baggage_entries.items()
89+
)
9190

9291
@property
93-
def fields(self) -> typing.Set[str]:
92+
def fields(self) -> Set[str]:
9493
"""Returns a set with the fields set in `inject`."""
95-
return {self._BAGGAGE_HEADER_NAME}
96-
97-
98-
def _format_baggage(baggage_entries: typing.Mapping[str, object]) -> str:
99-
return ",".join(
100-
key + "=" + urllib.parse.quote_plus(str(value))
101-
for key, value in baggage_entries.items()
102-
)
103-
104-
105-
def _extract_first_element(
106-
items: typing.Optional[typing.Iterable[textmap.TextMapPropagatorT]],
107-
) -> typing.Optional[textmap.TextMapPropagatorT]:
108-
if items is None:
109-
return None
110-
return next(iter(items), None)
94+
return {self._baggage_header_name}

opentelemetry-api/src/opentelemetry/propagate/__init__.py

+21-35
Original file line numberDiff line numberDiff line change
@@ -40,23 +40,12 @@
4040
PROPAGATOR = propagators.get_global_textmap()
4141
4242
43-
def get_header_from_flask_request(request, key):
44-
return request.headers.get_all(key)
45-
46-
def set_header_into_requests_request(request: requests.Request,
47-
key: str, value: str):
48-
request.headers[key] = value
49-
5043
def example_route():
51-
context = PROPAGATOR.extract(
52-
get_header_from_flask_request,
53-
flask.request
54-
)
44+
context = PROPAGATOR.extract(flask.request)
5545
request_to_downstream = requests.Request(
5646
"GET", "http://httpbin.org/get"
5747
)
5848
PROPAGATOR.inject(
59-
set_header_into_requests_request,
6049
request_to_downstream,
6150
context=context
6251
)
@@ -68,23 +57,25 @@ def example_route():
6857
https://github.com/open-telemetry/opentelemetry-specification/blob/main/specification/context/api-propagators.md
6958
"""
7059

71-
import typing
7260
from logging import getLogger
7361
from os import environ
62+
from typing import Optional
7463

7564
from pkg_resources import iter_entry_points
7665

7766
from opentelemetry.context.context import Context
7867
from opentelemetry.environment_variables import OTEL_PROPAGATORS
79-
from opentelemetry.propagators import composite, textmap
68+
from opentelemetry.propagators import composite
69+
from opentelemetry.propagators.textmap import (
70+
TextMapPropagator,
71+
TextMapPropagatorT,
72+
)
8073

81-
logger = getLogger(__name__)
74+
_logger = getLogger(__name__)
8275

8376

8477
def extract(
85-
getter: textmap.Getter[textmap.TextMapPropagatorT],
86-
carrier: textmap.TextMapPropagatorT,
87-
context: typing.Optional[Context] = None,
78+
carrier: TextMapPropagatorT, context: Optional[Context] = None,
8879
) -> Context:
8980
"""Uses the configured propagator to extract a Context from the carrier.
9081
@@ -99,26 +90,21 @@ def extract(
9990
context: an optional Context to use. Defaults to current
10091
context if not set.
10192
"""
102-
return get_global_textmap().extract(getter, carrier, context)
93+
return get_global_textmap().extract(carrier, context)
10394

10495

10596
def inject(
106-
set_in_carrier: textmap.Setter[textmap.TextMapPropagatorT],
107-
carrier: textmap.TextMapPropagatorT,
108-
context: typing.Optional[Context] = None,
97+
carrier: TextMapPropagatorT, context: Optional[Context] = None,
10998
) -> None:
11099
"""Uses the configured propagator to inject a Context into the carrier.
111100
112101
Args:
113-
set_in_carrier: A setter function that can set values
114-
on the carrier.
115-
carrier: An object that contains a representation of HTTP
116-
headers. Should be paired with set_in_carrier, which
117-
should know how to set header values on the carrier.
102+
carrier: A dict-like object that contains a representation of HTTP
103+
headers.
118104
context: an optional Context to use. Defaults to current
119105
context if not set.
120106
"""
121-
get_global_textmap().inject(set_in_carrier, carrier, context)
107+
get_global_textmap().inject(carrier, context)
122108

123109

124110
try:
@@ -138,16 +124,16 @@ def inject(
138124
)
139125

140126
except Exception: # pylint: disable=broad-except
141-
logger.exception("Failed to load configured propagators")
127+
_logger.error("Failed to load configured propagators")
142128
raise
143129

144-
_HTTP_TEXT_FORMAT = composite.CompositeHTTPPropagator(propagators) # type: ignore
130+
_textmap_propagator = composite.CompositeHTTPPropagator(propagators) # type: ignore
145131

146132

147-
def get_global_textmap() -> textmap.TextMapPropagator:
148-
return _HTTP_TEXT_FORMAT
133+
def get_global_textmap() -> TextMapPropagator:
134+
return _textmap_propagator
149135

150136

151-
def set_global_textmap(http_text_format: textmap.TextMapPropagator,) -> None:
152-
global _HTTP_TEXT_FORMAT # pylint:disable=global-statement
153-
_HTTP_TEXT_FORMAT = http_text_format # type: ignore
137+
def set_global_textmap(http_text_format: TextMapPropagator,) -> None:
138+
global _textmap_propagator # pylint:disable=global-statement
139+
_textmap_propagator = http_text_format # type: ignore

opentelemetry-api/src/opentelemetry/propagators/composite.py

+23-26
Original file line numberDiff line numberDiff line change
@@ -11,63 +11,60 @@
1111
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
14-
import logging
15-
import typing
14+
15+
from logging import getLogger
16+
from typing import Optional, Sequence, Set
1617

1718
from opentelemetry.context.context import Context
18-
from opentelemetry.propagators import textmap
19+
from opentelemetry.propagators.textmap import (
20+
TextMapPropagator,
21+
TextMapPropagatorT,
22+
)
1923

20-
logger = logging.getLogger(__name__)
24+
_logger = getLogger(__name__)
2125

2226

23-
class CompositeHTTPPropagator(textmap.TextMapPropagator):
27+
class CompositeHTTPPropagator(TextMapPropagator):
2428
"""CompositeHTTPPropagator provides a mechanism for combining multiple
2529
propagators into a single one.
2630
2731
Args:
2832
propagators: the list of propagators to use
2933
"""
3034

31-
def __init__(
32-
self, propagators: typing.Sequence[textmap.TextMapPropagator]
33-
) -> None:
35+
def __init__(self, propagators: Sequence[TextMapPropagator]) -> None:
3436
self._propagators = propagators
3537

3638
def extract(
37-
self,
38-
getter: textmap.Getter[textmap.TextMapPropagatorT],
39-
carrier: textmap.TextMapPropagatorT,
40-
context: typing.Optional[Context] = None,
39+
self, carrier: TextMapPropagatorT, context: Optional[Context] = None,
4140
) -> Context:
42-
"""Run each of the configured propagators with the given context and carrier.
41+
"""Run each of the configured propagators with the given context and
42+
carrier.
4343
Propagators are run in the order they are configured, if multiple
44-
propagators write the same context key, the propagator later in the list
45-
will override previous propagators.
44+
propagators write the same context key, the last propagator that writes
45+
the context key will override previous propagators.
4646
4747
See `opentelemetry.propagators.textmap.TextMapPropagator.extract`
4848
"""
4949
for propagator in self._propagators:
50-
context = propagator.extract(getter, carrier, context)
50+
context = propagator.extract(carrier, context)
5151
return context # type: ignore
5252

5353
def inject(
54-
self,
55-
set_in_carrier: textmap.Setter[textmap.TextMapPropagatorT],
56-
carrier: textmap.TextMapPropagatorT,
57-
context: typing.Optional[Context] = None,
54+
self, carrier: TextMapPropagatorT, context: Optional[Context] = None,
5855
) -> None:
59-
"""Run each of the configured propagators with the given context and carrier.
60-
Propagators are run in the order they are configured, if multiple
61-
propagators write the same carrier key, the propagator later in the list
62-
will override previous propagators.
56+
"""Run each of the configured propagators with the given context and
57+
carrier. Propagators are run in the order they are configured, if
58+
multiple propagators write the same carrier key, the last propagator
59+
that writes the carrier key will override previous propagators.
6360
6461
See `opentelemetry.propagators.textmap.TextMapPropagator.inject`
6562
"""
6663
for propagator in self._propagators:
67-
propagator.inject(set_in_carrier, carrier, context)
64+
propagator.inject(carrier, context)
6865

6966
@property
70-
def fields(self) -> typing.Set[str]:
67+
def fields(self) -> Set[str]:
7168
"""Returns a set with the fields set in `inject`.
7269
7370
See

0 commit comments

Comments
 (0)