Skip to content

Commit ccb58fc

Browse files
authored
Fix FastStream & OTEL deprecations (#126)
1 parent ead9748 commit ccb58fc

File tree

3 files changed

+8
-9
lines changed

3 files changed

+8
-9
lines changed

packages/faststream-stomp/faststream_stomp/broker.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,6 @@ def __init__(
5858
# Logging args
5959
logger: LoggerProto | None = EMPTY,
6060
log_level: int = logging.INFO,
61-
log_fmt: str | None = None,
6261
# FastDepends args
6362
apply_types: bool = True,
6463
validate: bool = True,
@@ -77,7 +76,6 @@ def __init__(
7776
graceful_timeout=graceful_timeout,
7877
logger=logger,
7978
log_level=log_level,
80-
log_fmt=log_fmt,
8179
apply_types=apply_types,
8280
validate=validate,
8381
_get_dependant=_get_dependant,

packages/faststream-stomp/faststream_stomp/opentelemetry.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from faststream.opentelemetry.middleware import TelemetryMiddleware
66
from faststream.types import AnyDict
77
from opentelemetry.metrics import Meter, MeterProvider
8+
from opentelemetry.semconv._incubating.attributes import messaging_attributes # noqa: PLC2701
89
from opentelemetry.semconv.trace import SpanAttributes
910
from opentelemetry.trace import TracerProvider
1011

@@ -16,9 +17,9 @@ class StompTelemetrySettingsProvider(TelemetrySettingsProvider[stompman.MessageF
1617

1718
def get_consume_attrs_from_message(self, msg: StreamMessage[stompman.MessageFrame]) -> "AnyDict":
1819
return {
19-
SpanAttributes.MESSAGING_SYSTEM: self.messaging_system,
20-
SpanAttributes.MESSAGING_MESSAGE_ID: msg.message_id,
21-
SpanAttributes.MESSAGING_MESSAGE_CONVERSATION_ID: msg.correlation_id,
20+
messaging_attributes.MESSAGING_SYSTEM: self.messaging_system,
21+
messaging_attributes.MESSAGING_MESSAGE_ID: msg.message_id,
22+
messaging_attributes.MESSAGING_MESSAGE_CONVERSATION_ID: msg.correlation_id,
2223
SpanAttributes.MESSAGING_MESSAGE_PAYLOAD_SIZE_BYTES: len(msg.body),
2324
MESSAGING_DESTINATION_PUBLISH_NAME: msg.raw_message.headers["destination"],
2425
}
@@ -28,11 +29,11 @@ def get_consume_destination_name(self, msg: StreamMessage[stompman.MessageFrame]
2829

2930
def get_publish_attrs_from_kwargs(self, kwargs: StompProducerPublishKwargs) -> AnyDict: # type: ignore[override]
3031
publish_attrs = {
31-
SpanAttributes.MESSAGING_SYSTEM: self.messaging_system,
32-
SpanAttributes.MESSAGING_DESTINATION_NAME: kwargs["destination"],
32+
messaging_attributes.MESSAGING_SYSTEM: self.messaging_system,
33+
messaging_attributes.MESSAGING_DESTINATION_NAME: kwargs["destination"],
3334
}
3435
if kwargs["correlation_id"]:
35-
publish_attrs[SpanAttributes.MESSAGING_MESSAGE_CONVERSATION_ID] = kwargs["correlation_id"]
36+
publish_attrs[messaging_attributes.MESSAGING_MESSAGE_CONVERSATION_ID] = kwargs["correlation_id"]
3637
return publish_attrs
3738

3839
def get_publish_destination_name(self, kwargs: StompProducerPublishKwargs) -> str: # type: ignore[override] # noqa: PLR6301

packages/faststream-stomp/test_faststream_stomp/test_integration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ async def test_subscriber_lifespan(faker: faker.Faker, broker: faststream_stomp.
112112
def _() -> None: ...
113113

114114
await broker.start()
115-
await broker.close()
115+
await broker.stop()
116116

117117

118118
class TestPing:

0 commit comments

Comments
 (0)