datadog: use self.config.get_current_span

TheShellLand · Apr 13, 2024 · 2346e4f · 2346e4f
1 parent d053b78
commit 2346e4f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/automon/integrations/openTelemetryWrapper/client.py b/automon/integrations/openTelemetryWrapper/client.py
@@ -18,7 +18,7 @@ def __init__(self):
 
     def add_event(self, name: str, attributes: types.Attributes = None, **kwargs):
         logger.debug(dict(name=name, attributes=attributes, kwargs=kwargs))
-        return self.config.opentelemetry.trace.get_current_span().add_event(
+        return self.config.get_current_span().add_event(
             name=name,
             attributes=attributes,
             **kwargs
@@ -28,9 +28,9 @@ async def clear(self):
         logger.debug('clear')
         return await self.config.clear()
 
-    def current_span(self):
+    def get_current_span(self):
         logger.debug('current_span')
-        return self.config.current_span()
+        return self.config.get_current_span()
 
     async def is_ready(self):
         if await self.config.is_ready():
@@ -46,7 +46,7 @@ async def pop_finished_spans(self):
 
     async def record_exception(self, exception: Exception):
         logger.error(f'{exception}')
-        span = await self.current_span()
+        span = await self.get_current_span()
         span.set_status(Status(StatusCode.ERROR))
         return span.record_exception(exception=exception)
 

diff --git a/automon/integrations/openTelemetryWrapper/config.py b/automon/integrations/openTelemetryWrapper/config.py
@@ -30,7 +30,7 @@ async def clear(self):
         logger.debug('clear')
         return self.memory_processor.clear()
 
-    def current_span(self):
+    def get_current_span(self):
         logger.debug('get_current_span')
         return opentelemetry.trace.get_current_span()