Merge pull request #100 from NathanielRN/move-instrumentation-dbapi

Move instrumentation dbapi
This commit is contained in:
alrex
2020-11-02 13:08:23 -08:00
committed by GitHub
4 changed files with 8 additions and 14 deletions

View File

@ -39,13 +39,13 @@ package_dir=
=src
packages=find_namespace:
install_requires =
opentelemetry-api == 0.15.dev0
opentelemetry-instrumentation == 0.15.dev0
opentelemetry-api == 0.15b0
opentelemetry-instrumentation == 0.15b0
wrapt >= 1.0.0, < 2.0.0
[options.extras_require]
test =
opentelemetry-test == 0.15.dev0
opentelemetry-test == 0.15b0
[options.packages.find]
where = src

View File

@ -50,7 +50,7 @@ from opentelemetry import trace as trace_api
from opentelemetry.instrumentation.dbapi.version import __version__
from opentelemetry.instrumentation.utils import unwrap
from opentelemetry.trace import SpanKind, TracerProvider, get_tracer
from opentelemetry.trace.status import Status, StatusCanonicalCode
from opentelemetry.trace.status import Status, StatusCode
logger = logging.getLogger(__name__)
@ -343,14 +343,10 @@ class TracedCursor:
self._populate_span(span, *args)
try:
result = query_method(*args, **kwargs)
if span.is_recording():
span.set_status(Status(StatusCanonicalCode.OK))
return result
except Exception as ex: # pylint: disable=broad-except
if span.is_recording():
span.set_status(
Status(StatusCanonicalCode.UNKNOWN, str(ex))
)
span.set_status(Status(StatusCode.ERROR, str(ex)))
raise ex

View File

@ -12,4 +12,4 @@
# See the License for the specific language governing permissions and
# limitations under the License.
__version__ = "0.15.dev0"
__version__ = "0.15b0"

View File

@ -65,8 +65,7 @@ class TestDBApiIntegration(TestBase):
self.assertEqual(span.attributes["net.peer.name"], "testhost")
self.assertEqual(span.attributes["net.peer.port"], 123)
self.assertIs(
span.status.canonical_code,
trace_api.status.StatusCanonicalCode.OK,
span.status.status_code, trace_api.status.StatusCode.UNSET,
)
def test_span_not_recording(self):
@ -117,8 +116,7 @@ class TestDBApiIntegration(TestBase):
span = spans_list[0]
self.assertEqual(span.attributes["db.statement"], "Test query")
self.assertIs(
span.status.canonical_code,
trace_api.status.StatusCanonicalCode.UNKNOWN,
span.status.status_code, trace_api.status.StatusCode.ERROR,
)
self.assertEqual(span.status.description, "Test Exception")