Skip to content

Commit fbb2763

Browse files
committed
Replaced usage of opentelemetry.semconv.trace with opentelemetry.semconv._incubating.attributes.
1 parent 0db4d0b commit fbb2763

File tree

2 files changed

+28
-21
lines changed

2 files changed

+28
-21
lines changed

instrumentation/opentelemetry-instrumentation-redis/src/opentelemetry/instrumentation/redis/__init__.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def response_hook(span, instance, response):
135135
)
136136
from opentelemetry.instrumentation.redis.version import __version__
137137
from opentelemetry.instrumentation.utils import unwrap
138-
from opentelemetry.semconv.trace import SpanAttributes
138+
from opentelemetry.semconv._incubating.attributes.db_attributes import DB_STATEMENT
139139
from opentelemetry.trace import (
140140
StatusCode,
141141
Tracer,
@@ -200,7 +200,7 @@ def _traced_execute_command(
200200
name, kind=trace.SpanKind.CLIENT
201201
) as span:
202202
if span.is_recording():
203-
span.set_attribute(SpanAttributes.DB_STATEMENT, query)
203+
span.set_attribute(DB_STATEMENT, query)
204204
_set_connection_attributes(span, instance)
205205
span.set_attribute("db.redis.args_length", len(args))
206206
if span.name == "redis.create_index":
@@ -239,7 +239,7 @@ def _traced_execute_pipeline(
239239
span_name, kind=trace.SpanKind.CLIENT
240240
) as span:
241241
if span.is_recording():
242-
span.set_attribute(SpanAttributes.DB_STATEMENT, resource)
242+
span.set_attribute(DB_STATEMENT, resource)
243243
_set_connection_attributes(span, instance)
244244
span.set_attribute(
245245
"db.redis.pipeline_length", len(command_stack)
@@ -281,7 +281,7 @@ async def _async_traced_execute_command(
281281
name, kind=trace.SpanKind.CLIENT
282282
) as span:
283283
if span.is_recording():
284-
span.set_attribute(SpanAttributes.DB_STATEMENT, query)
284+
span.set_attribute(DB_STATEMENT, query)
285285
_set_connection_attributes(span, instance)
286286
span.set_attribute("db.redis.args_length", len(args))
287287
if callable(request_hook):
@@ -317,7 +317,7 @@ async def _async_traced_execute_pipeline(
317317
span_name, kind=trace.SpanKind.CLIENT
318318
) as span:
319319
if span.is_recording():
320-
span.set_attribute(SpanAttributes.DB_STATEMENT, resource)
320+
span.set_attribute(DB_STATEMENT, resource)
321321
_set_connection_attributes(span, instance)
322322
span.set_attribute(
323323
"db.redis.pipeline_length", len(command_stack)

instrumentation/opentelemetry-instrumentation-redis/tests/test_redis.py

Lines changed: 23 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,18 @@
2525

2626
from opentelemetry import trace
2727
from opentelemetry.instrumentation.redis import RedisInstrumentor
28-
from opentelemetry.semconv.trace import (
28+
from opentelemetry.semconv._incubating.attributes.db_attributes import (
29+
DB_SYSTEM,
30+
DB_REDIS_DATABASE_INDEX,
2931
DbSystemValues,
32+
)
33+
from opentelemetry.semconv._incubating.attributes.net_attributes import (
34+
NET_PEER_NAME,
35+
NET_PEER_PORT,
36+
NET_TRANSPORT,
3037
NetTransportValues,
31-
SpanAttributes,
3238
)
39+
3340
from opentelemetry.test.test_base import TestBase
3441
from opentelemetry.trace import SpanKind
3542

@@ -257,18 +264,18 @@ def test_attributes_default(self):
257264

258265
span = spans[0]
259266
self.assertEqual(
260-
span.attributes[SpanAttributes.DB_SYSTEM],
267+
span.attributes[DB_SYSTEM],
261268
DbSystemValues.REDIS.value,
262269
)
263270
self.assertEqual(
264-
span.attributes[SpanAttributes.DB_REDIS_DATABASE_INDEX], 0
271+
span.attributes[DB_REDIS_DATABASE_INDEX], 0
265272
)
266273
self.assertEqual(
267-
span.attributes[SpanAttributes.NET_PEER_NAME], "localhost"
274+
span.attributes[NET_PEER_NAME], "localhost"
268275
)
269-
self.assertEqual(span.attributes[SpanAttributes.NET_PEER_PORT], 6379)
276+
self.assertEqual(span.attributes[NET_PEER_PORT], 6379)
270277
self.assertEqual(
271-
span.attributes[SpanAttributes.NET_TRANSPORT],
278+
span.attributes[NET_TRANSPORT],
272279
NetTransportValues.IP_TCP.value,
273280
)
274281

@@ -283,18 +290,18 @@ def test_attributes_tcp(self):
283290

284291
span = spans[0]
285292
self.assertEqual(
286-
span.attributes[SpanAttributes.DB_SYSTEM],
293+
span.attributes[DB_SYSTEM],
287294
DbSystemValues.REDIS.value,
288295
)
289296
self.assertEqual(
290-
span.attributes[SpanAttributes.DB_REDIS_DATABASE_INDEX], 1
297+
span.attributes[DB_REDIS_DATABASE_INDEX], 1
291298
)
292299
self.assertEqual(
293-
span.attributes[SpanAttributes.NET_PEER_NAME], "1.1.1.1"
300+
span.attributes[NET_PEER_NAME], "1.1.1.1"
294301
)
295-
self.assertEqual(span.attributes[SpanAttributes.NET_PEER_PORT], 6380)
302+
self.assertEqual(span.attributes[NET_PEER_PORT], 6380)
296303
self.assertEqual(
297-
span.attributes[SpanAttributes.NET_TRANSPORT],
304+
span.attributes[NET_TRANSPORT],
298305
NetTransportValues.IP_TCP.value,
299306
)
300307

@@ -311,18 +318,18 @@ def test_attributes_unix_socket(self):
311318

312319
span = spans[0]
313320
self.assertEqual(
314-
span.attributes[SpanAttributes.DB_SYSTEM],
321+
span.attributes[DB_SYSTEM],
315322
DbSystemValues.REDIS.value,
316323
)
317324
self.assertEqual(
318-
span.attributes[SpanAttributes.DB_REDIS_DATABASE_INDEX], 3
325+
span.attributes[DB_REDIS_DATABASE_INDEX], 3
319326
)
320327
self.assertEqual(
321-
span.attributes[SpanAttributes.NET_PEER_NAME],
328+
span.attributes[NET_PEER_NAME],
322329
"/path/to/socket.sock",
323330
)
324331
self.assertEqual(
325-
span.attributes[SpanAttributes.NET_TRANSPORT],
332+
span.attributes[NET_TRANSPORT],
326333
NetTransportValues.OTHER.value,
327334
)
328335

0 commit comments

Comments
 (0)