|
25 | 25 | KafkaContextGetter, |
26 | 26 | KafkaContextSetter, |
27 | 27 | ) |
| 28 | +from opentelemetry.semconv._incubating.attributes.messaging_attributes import ( |
| 29 | + MESSAGING_DESTINATION_NAME, |
| 30 | + MESSAGING_KAFKA_DESTINATION_PARTITION, |
| 31 | + MESSAGING_MESSAGE_ID, |
| 32 | + MESSAGING_OPERATION, |
| 33 | + MESSAGING_SYSTEM, |
| 34 | +) |
28 | 35 | from opentelemetry.semconv.trace import ( |
29 | 36 | MessagingDestinationKindValues, |
30 | 37 | SpanAttributes, |
@@ -122,36 +129,36 @@ def test_poll(self) -> None: |
122 | 129 | { |
123 | 130 | "name": "topic-10 process", |
124 | 131 | "attributes": { |
125 | | - SpanAttributes.MESSAGING_OPERATION: "process", |
126 | | - SpanAttributes.MESSAGING_KAFKA_PARTITION: 0, |
127 | | - SpanAttributes.MESSAGING_SYSTEM: "kafka", |
128 | | - SpanAttributes.MESSAGING_DESTINATION: "topic-10", |
| 132 | + MESSAGING_OPERATION: "process", |
| 133 | + MESSAGING_KAFKA_DESTINATION_PARTITION: 0, |
| 134 | + MESSAGING_SYSTEM: "kafka", |
| 135 | + MESSAGING_DESTINATION_NAME: "topic-10", |
129 | 136 | SpanAttributes.MESSAGING_DESTINATION_KIND: MessagingDestinationKindValues.QUEUE.value, |
130 | | - SpanAttributes.MESSAGING_MESSAGE_ID: "topic-10.0.0", |
| 137 | + MESSAGING_MESSAGE_ID: "topic-10.0.0", |
131 | 138 | }, |
132 | 139 | }, |
133 | 140 | {"name": "recv", "attributes": {}}, |
134 | 141 | { |
135 | 142 | "name": "topic-20 process", |
136 | 143 | "attributes": { |
137 | | - SpanAttributes.MESSAGING_OPERATION: "process", |
138 | | - SpanAttributes.MESSAGING_KAFKA_PARTITION: 2, |
139 | | - SpanAttributes.MESSAGING_SYSTEM: "kafka", |
140 | | - SpanAttributes.MESSAGING_DESTINATION: "topic-20", |
| 144 | + MESSAGING_OPERATION: "process", |
| 145 | + MESSAGING_KAFKA_DESTINATION_PARTITION: 2, |
| 146 | + MESSAGING_SYSTEM: "kafka", |
| 147 | + MESSAGING_DESTINATION_NAME: "topic-20", |
141 | 148 | SpanAttributes.MESSAGING_DESTINATION_KIND: MessagingDestinationKindValues.QUEUE.value, |
142 | | - SpanAttributes.MESSAGING_MESSAGE_ID: "topic-20.2.4", |
| 149 | + MESSAGING_MESSAGE_ID: "topic-20.2.4", |
143 | 150 | }, |
144 | 151 | }, |
145 | 152 | {"name": "recv", "attributes": {}}, |
146 | 153 | { |
147 | 154 | "name": "topic-30 process", |
148 | 155 | "attributes": { |
149 | | - SpanAttributes.MESSAGING_OPERATION: "process", |
150 | | - SpanAttributes.MESSAGING_KAFKA_PARTITION: 1, |
151 | | - SpanAttributes.MESSAGING_SYSTEM: "kafka", |
152 | | - SpanAttributes.MESSAGING_DESTINATION: "topic-30", |
| 156 | + MESSAGING_OPERATION: "process", |
| 157 | + MESSAGING_KAFKA_DESTINATION_PARTITION: 1, |
| 158 | + MESSAGING_SYSTEM: "kafka", |
| 159 | + MESSAGING_DESTINATION_NAME: "topic-30", |
153 | 160 | SpanAttributes.MESSAGING_DESTINATION_KIND: MessagingDestinationKindValues.QUEUE.value, |
154 | | - SpanAttributes.MESSAGING_MESSAGE_ID: "topic-30.1.3", |
| 161 | + MESSAGING_MESSAGE_ID: "topic-30.1.3", |
155 | 162 | }, |
156 | 163 | }, |
157 | 164 | {"name": "recv", "attributes": {}}, |
@@ -190,29 +197,29 @@ def test_consume(self) -> None: |
190 | 197 | { |
191 | 198 | "name": "topic-1 process", |
192 | 199 | "attributes": { |
193 | | - SpanAttributes.MESSAGING_OPERATION: "process", |
194 | | - SpanAttributes.MESSAGING_SYSTEM: "kafka", |
195 | | - SpanAttributes.MESSAGING_DESTINATION: "topic-1", |
| 200 | + MESSAGING_OPERATION: "process", |
| 201 | + MESSAGING_SYSTEM: "kafka", |
| 202 | + MESSAGING_DESTINATION_NAME: "topic-1", |
196 | 203 | SpanAttributes.MESSAGING_DESTINATION_KIND: MessagingDestinationKindValues.QUEUE.value, |
197 | 204 | }, |
198 | 205 | }, |
199 | 206 | {"name": "recv", "attributes": {}}, |
200 | 207 | { |
201 | 208 | "name": "topic-2 process", |
202 | 209 | "attributes": { |
203 | | - SpanAttributes.MESSAGING_OPERATION: "process", |
204 | | - SpanAttributes.MESSAGING_SYSTEM: "kafka", |
205 | | - SpanAttributes.MESSAGING_DESTINATION: "topic-2", |
| 210 | + MESSAGING_OPERATION: "process", |
| 211 | + MESSAGING_SYSTEM: "kafka", |
| 212 | + MESSAGING_DESTINATION_NAME: "topic-2", |
206 | 213 | SpanAttributes.MESSAGING_DESTINATION_KIND: MessagingDestinationKindValues.QUEUE.value, |
207 | 214 | }, |
208 | 215 | }, |
209 | 216 | {"name": "recv", "attributes": {}}, |
210 | 217 | { |
211 | 218 | "name": "topic-3 process", |
212 | 219 | "attributes": { |
213 | | - SpanAttributes.MESSAGING_OPERATION: "process", |
214 | | - SpanAttributes.MESSAGING_SYSTEM: "kafka", |
215 | | - SpanAttributes.MESSAGING_DESTINATION: "topic-3", |
| 220 | + MESSAGING_OPERATION: "process", |
| 221 | + MESSAGING_SYSTEM: "kafka", |
| 222 | + MESSAGING_DESTINATION_NAME: "topic-3", |
216 | 223 | SpanAttributes.MESSAGING_DESTINATION_KIND: MessagingDestinationKindValues.QUEUE.value, |
217 | 224 | }, |
218 | 225 | }, |
@@ -247,12 +254,12 @@ def test_close(self) -> None: |
247 | 254 | { |
248 | 255 | "name": "topic-a process", |
249 | 256 | "attributes": { |
250 | | - SpanAttributes.MESSAGING_OPERATION: "process", |
251 | | - SpanAttributes.MESSAGING_KAFKA_PARTITION: 0, |
252 | | - SpanAttributes.MESSAGING_SYSTEM: "kafka", |
253 | | - SpanAttributes.MESSAGING_DESTINATION: "topic-a", |
| 257 | + MESSAGING_OPERATION: "process", |
| 258 | + MESSAGING_KAFKA_DESTINATION_PARTITION: 0, |
| 259 | + MESSAGING_SYSTEM: "kafka", |
| 260 | + MESSAGING_DESTINATION_NAME: "topic-a", |
254 | 261 | SpanAttributes.MESSAGING_DESTINATION_KIND: MessagingDestinationKindValues.QUEUE.value, |
255 | | - SpanAttributes.MESSAGING_MESSAGE_ID: "topic-a.0.0", |
| 262 | + MESSAGING_MESSAGE_ID: "topic-a.0.0", |
256 | 263 | }, |
257 | 264 | }, |
258 | 265 | ] |
@@ -286,7 +293,7 @@ def _compare_spans(self, spans, expected_spans): |
286 | 293 |
|
287 | 294 | def _assert_topic(self, span, expected_topic: str) -> None: |
288 | 295 | self.assertEqual( |
289 | | - span.attributes[SpanAttributes.MESSAGING_DESTINATION], |
| 296 | + span.attributes[MESSAGING_DESTINATION_NAME], |
290 | 297 | expected_topic, |
291 | 298 | ) |
292 | 299 |
|
|
0 commit comments