add logging

Author: ionmincu

src/uipath/core/tracing/manager.py

@@ -4,6 +4,7 @@
 from typing import Any, Callable, List, Optional
 
 from opentelemetry import context, trace
+from opentelemetry.sdk.trace import ReadableSpan, TracerProvider
 from opentelemetry.trace import set_span_in_context
 
 logger = logging.getLogger(__name__)
@@ -37,14 +38,20 @@ def get_parent_context() -> context.Context:
         - Current context as fallback
         """
         current_span = trace.get_current_span()
-        has_current_span = current_span is not None and current_span.get_span_context().is_valid
+        has_current_span = (
+            current_span is not None and current_span.get_span_context().is_valid
+        )
 
         external_span = UiPathTracingManager.get_external_current_span()
         has_external_span = external_span is not None
 
         # Only one or no spans available
         if not has_current_span:
-            return set_span_in_context(external_span) if has_external_span else context.get_current()
+            return (
+                set_span_in_context(external_span)
+                if has_external_span
+                else context.get_current()
+            )
         if not has_external_span:
             return set_span_in_context(current_span)
 
@@ -55,7 +62,9 @@ def get_parent_context() -> context.Context:
         return set_span_in_context(bottom_span)
 
     @staticmethod
-    def _get_bottom_most_span(current_span: Any, external_span: Any) -> Any:
+    def _get_bottom_most_span(
+        current_span: ReadableSpan, external_span: ReadableSpan
+    ) -> ReadableSpan:
         """Determine which span is deeper in the ancestor tree.
 
         Args:
@@ -66,33 +75,85 @@ def _get_bottom_most_span(current_span: Any, external_span: Any) -> Any:
             The span that is deeper (closer to the bottom) in the call hierarchy
         """
         ancestor_spans = UiPathTracingManager.get_ancestor_spans()
+
+        logger.info("=" * 80)
+        logger.info("Determining bottom-most span...")
+
         if not ancestor_spans:
+            logger.info("No ancestor spans available -> returning external_span")
+            logger.info("=" * 80)
             return external_span
 
         current_span_id = current_span.get_span_context().span_id
         external_span_id = external_span.get_span_context().span_id
 
+        current_parent_id = (
+            f"{current_span.parent.span_id:016x}" if current_span.parent else "None"
+        )
+        external_parent_id = (
+            f"{external_span.parent.span_id:016x}" if external_span.parent else "None"
+        )
+
+        logger.info(
+            f"current_span: '{current_span.name}' (id: {current_span_id:016x}, parent: {current_parent_id})"
+        )
+        logger.info(
+            f"external_span: '{external_span.name}' (id: {external_span_id:016x}, parent: {external_parent_id})"
+        )
+        logger.info(f"Total ancestors: {len(ancestor_spans)}")
+
         current_index = None
         external_index = None
 
+        # Log all ancestors in a single line
+        ancestor_info = " -> ".join(
+            [f"{ancestor.name}[{i}]" for i, ancestor in enumerate(ancestor_spans)]
+        )
+        logger.info(f"Ancestor chain (top to bottom): {ancestor_info}")
+
         for i, ancestor in enumerate(ancestor_spans):
             ancestor_id = ancestor.get_span_context().span_id
             if ancestor_id == current_span_id:
                 current_index = i
+                logger.info(f"  Found current_span at index {i}: '{ancestor.name}'")
             if ancestor_id == external_span_id:
                 external_index = i
+                logger.info(f"  Found external_span at index {i}: '{ancestor.name}'")
 
         # Both in tree: higher index = deeper
         if current_index is not None and external_index is not None:
-            return current_span if current_index > external_index else external_span
+            if current_index > external_index:
+                logger.info(
+                    f"RESULT: current_span '{current_span.name}' is deeper (index {current_index} > {external_index})"
+                )
+                logger.info("=" * 80)
+                return current_span
+            else:
+                logger.info(
+                    f"RESULT: external_span '{external_span.name}' is deeper (index {external_index} >= {current_index})"
+                )
+                logger.info("=" * 80)
+                return external_span
 
         # Only one in tree: that one is deeper
         if current_index is not None:
+            logger.info(
+                f"RESULT: Only current_span '{current_span.name}' in tree -> returning it"
+            )
+            logger.info("=" * 80)
             return current_span
         if external_index is not None:
+            logger.info(
+                f"RESULT: Only external_span '{external_span.name}' in tree -> returning it"
+            )
+            logger.info("=" * 80)
             return external_span
 
         # Neither in tree: default to external
+        logger.info(
+            f"RESULT: Neither span in tree -> defaulting to external_span '{external_span.name}'"
+        )
+        logger.info("=" * 80)
         return external_span
 
     @staticmethod