Change status codes from grpc status codes, remove setting status in instrumentations except on ERROR (#1282)
This commit is contained in:
parent
0586264ec6
commit
fd5ffe84b9
|
|
@ -8,7 +8,7 @@ from opentelemetry.instrumentation.dbapi import (
|
||||||
TracedCursor,
|
TracedCursor,
|
||||||
)
|
)
|
||||||
from opentelemetry.trace import SpanKind
|
from opentelemetry.trace import SpanKind
|
||||||
from opentelemetry.trace.status import Status, StatusCanonicalCode
|
from opentelemetry.trace.status import Status, StatusCode
|
||||||
|
|
||||||
|
|
||||||
# pylint: disable=abstract-method
|
# pylint: disable=abstract-method
|
||||||
|
|
@ -108,14 +108,10 @@ class AsyncTracedCursor(TracedCursor):
|
||||||
self._populate_span(span, *args)
|
self._populate_span(span, *args)
|
||||||
try:
|
try:
|
||||||
result = await query_method(*args, **kwargs)
|
result = await query_method(*args, **kwargs)
|
||||||
if span.is_recording():
|
|
||||||
span.set_status(Status(StatusCanonicalCode.OK))
|
|
||||||
return result
|
return result
|
||||||
except Exception as ex: # pylint: disable=broad-except
|
except Exception as ex: # pylint: disable=broad-except
|
||||||
if span.is_recording():
|
if span.is_recording():
|
||||||
span.set_status(
|
span.set_status(Status(StatusCode.ERROR, str(ex)))
|
||||||
Status(StatusCanonicalCode.UNKNOWN, str(ex))
|
|
||||||
)
|
|
||||||
raise ex
|
raise ex
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -226,8 +226,7 @@ class TestAiopgIntegration(TestBase):
|
||||||
self.assertEqual(span.attributes["net.peer.name"], "testhost")
|
self.assertEqual(span.attributes["net.peer.name"], "testhost")
|
||||||
self.assertEqual(span.attributes["net.peer.port"], 123)
|
self.assertEqual(span.attributes["net.peer.port"], 123)
|
||||||
self.assertIs(
|
self.assertIs(
|
||||||
span.status.canonical_code,
|
span.status.status_code, trace_api.status.StatusCode.UNSET,
|
||||||
trace_api.status.StatusCanonicalCode.OK,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
def test_span_not_recording(self):
|
def test_span_not_recording(self):
|
||||||
|
|
@ -278,8 +277,7 @@ class TestAiopgIntegration(TestBase):
|
||||||
span = spans_list[0]
|
span = spans_list[0]
|
||||||
self.assertEqual(span.attributes["db.statement"], "Test query")
|
self.assertEqual(span.attributes["db.statement"], "Test query")
|
||||||
self.assertIs(
|
self.assertIs(
|
||||||
span.status.canonical_code,
|
span.status.status_code, trace_api.status.StatusCode.ERROR,
|
||||||
trace_api.status.StatusCanonicalCode.UNKNOWN,
|
|
||||||
)
|
)
|
||||||
self.assertEqual(span.status.description, "Test Exception")
|
self.assertEqual(span.status.description, "Test Exception")
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue