Skip to content

Commit 7fc47f8

Browse files
committed
Remove setters and getters
Fixes open-telemetry#1644
1 parent c81fd5e commit 7fc47f8

File tree

14 files changed

+546
-740
lines changed

14 files changed

+546
-740
lines changed

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

+38-54
Original file line numberDiff line numberDiff line change
@@ -11,29 +11,26 @@
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"
3131

3232
def extract(
33-
self,
34-
getter: textmap.Getter[textmap.TextMapPropagatorT],
35-
carrier: textmap.TextMapPropagatorT,
36-
context: typing.Optional[Context] = None,
33+
self, carrier: TextMapPropagatorT, context: Optional[Context] = None,
3734
) -> Context:
3835
"""Extract Baggage from the carrier.
3936
@@ -44,67 +41,54 @@ def extract(
4441
if context is None:
4542
context = get_current()
4643

47-
header = _extract_first_element(
48-
getter.get(carrier, self._BAGGAGE_HEADER_NAME)
49-
)
44+
value = carrier.get(self._baggage_header_name)
45+
46+
if value is None:
47+
header = None
48+
else:
49+
header = next(iter(value), None)
5050

51-
if not header or len(header) > self._MAX_HEADER_LENGTH:
51+
# 8192 is the maximum header length
52+
if header is None or len(header) > 8192:
5253
return context
5354

54-
baggage_entries = header.split(",")
55-
total_baggage_entries = self._MAX_PAIRS
56-
for entry in baggage_entries:
55+
# 180 is the maximum amount of pairs
56+
total_baggage_entries = 180
57+
58+
for entry in header.split(","):
5759
if total_baggage_entries <= 0:
5860
return context
5961
total_baggage_entries -= 1
60-
if len(entry) > self._MAX_PAIR_LENGTH:
62+
# 4096 is the maximum pair length
63+
if len(entry) > 4096:
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)